Coverage for polar/models/organization_access_token.py: 95%

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

9 

10from .organization import Organization 2 ctx1ab

11 

12 

13class OrganizationAccessToken(RecordModel): 2 ctx1ab

14 __tablename__ = "organization_access_tokens" 2 ctx1ab

15 

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

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

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

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

20 ) 

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

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

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

24 ) 

25 

26 organization_id: Mapped[UUID] = mapped_column( 2 ctx1ab

27 Uuid, ForeignKey("organizations.id"), nullable=False, index=True 

28 ) 

29 

30 @declared_attr 2 ctx1ab

31 def organization(cls) -> Mapped["Organization"]: 2 ctx1ab

32 return relationship(Organization, lazy="raise") 2 ctx1ab

33 

34 @property 2 ctx1ab

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

36 return scope_to_set(self.scope)