Coverage for polar/models/user_session.py: 96%

1from datetime import datetime 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import CHAR, TIMESTAMP, ForeignKey, Text, Uuid 2 ctx1ab

5from sqlalchemy.dialects.postgresql import ARRAY 2 ctx1ab

6from sqlalchemy.orm import Mapped, declared_attr, mapped_column, relationship 2 ctx1ab

7 

8from polar.auth.scope import Scope 2 ctx1ab

9from polar.config import settings 2 ctx1ab

10from polar.kit.db.models.base import RecordModel 2 ctx1ab

11from polar.kit.extensions.sqlalchemy import StringEnum 2 ctx1ab

12from polar.kit.utils import utc_now 2 ctx1ab

13from polar.models.user import User 2 ctx1ab

14 

15 

16def get_expires_at() -> datetime: 2 ctx1ab

17 return utc_now() + settings.USER_SESSION_TTL 

18 

19 

20class UserSession(RecordModel): 2 ctx1ab

21 __tablename__ = "user_sessions" 2 ctx1ab

22 

23 token: Mapped[str] = mapped_column(CHAR(64), unique=True, nullable=False) 2 ctx1ab

24 expires_at: Mapped[datetime] = mapped_column( 2 ctx1ab

25 TIMESTAMP(timezone=True), nullable=False, index=True, default=get_expires_at 

26 ) 

27 user_agent: Mapped[str] = mapped_column(Text, nullable=False) 2 ctx1ab

28 scopes: Mapped[list[Scope]] = mapped_column( 2 ctx1ab

29 ARRAY(StringEnum(Scope)), nullable=False, default=list 

30 ) 

31 

32 user_id: Mapped[UUID] = mapped_column( 2 ctx1ab

33 Uuid, ForeignKey("users.id", ondelete="cascade"), nullable=False 

34 ) 

35 

36 @declared_attr 2 ctx1ab

37 def user(cls) -> Mapped[User]: 2 ctx1ab

38 return relationship(User, lazy="joined") 2 ctx1ab