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

1from datetime import datetime 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

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

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

6 

7from polar.auth.scope import Scope, scope_to_set 2 ctx1ab

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

9from polar.models.user import User 2 ctx1ab

10 

11 

12class PersonalAccessToken(RecordModel): 2 ctx1ab

13 __tablename__ = "personal_access_tokens" 2 ctx1ab

14 

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

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

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

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

19 ) 

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

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

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

23 ) 

24 

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

26 

27 @declared_attr 2 ctx1ab

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

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

30 

31 @property 2 ctx1ab

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

33 return scope_to_set(self.scope) 1 ctx1c