Coverage for polar/models/personal_access_token.py: 100%

1from datetime import datetime 

2from uuid import UUID 

3 

4from sqlalchemy import CHAR, TIMESTAMP, ForeignKey, String, Text, Uuid 

5from sqlalchemy.orm import Mapped, declared_attr, mapped_column, relationship 

6 

7from polar.auth.scope import Scope, scope_to_set 

8from polar.kit.db.models.base import RecordModel 

9from polar.models.user import User 

10 

11 

12class PersonalAccessToken(RecordModel): 

13 __tablename__ = "personal_access_tokens" 

14 

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

16 scope: Mapped[str] = mapped_column(Text, nullable=False) 

17 expires_at: Mapped[datetime | None] = mapped_column( 

18 TIMESTAMP(timezone=True), nullable=True, index=True 

19 ) 

20 comment: Mapped[str] = mapped_column(String, nullable=False) 

21 last_used_at: Mapped[datetime | None] = mapped_column( 

22 TIMESTAMP(timezone=True), nullable=True, default=None 

23 ) 

24 

25 user_id: Mapped[UUID] = mapped_column(Uuid, ForeignKey("users.id"), nullable=False) 

26 

27 @declared_attr 

28 def user(cls) -> Mapped[User]: 

29 return relationship(User, lazy="raise") 

30 

31 @property 

32 def scopes(self) -> set[Scope]: 

33 return scope_to_set(self.scope)