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

1from uuid import UUID 2 ctx1ab

2 

3from sqlalchemy import ForeignKey, String, Text, UniqueConstraint, Uuid 2 ctx1ab

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

5 

6from polar.kit.db.models import RecordModel 2 ctx1ab

7from polar.models.organization import Organization 2 ctx1ab

8from polar.models.user import User 2 ctx1ab

9 

10 

11class OAuth2Grant(RecordModel): 2 ctx1ab

12 __tablename__ = "oauth2_grants" 2 ctx1ab

13 __table_args__ = ( 2 ctx1ab

14 UniqueConstraint("client_id", "user_id"), 

15 UniqueConstraint("client_id", "organization_id"), 

16 ) 

17 

18 client_id: Mapped[str] = mapped_column(String(52), nullable=False, index=True) 2 ctx1ab

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

20 user_id: Mapped[UUID | None] = mapped_column( 2 ctx1ab

21 Uuid, 

22 ForeignKey("users.id", ondelete="cascade"), 

23 nullable=True, 

24 index=True, 

25 ) 

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

27 Uuid, 

28 ForeignKey("organizations.id", ondelete="cascade"), 

29 nullable=True, 

30 index=True, 

31 ) 

32 

33 @declared_attr 2 ctx1ab

34 def user(cls) -> Mapped[User | None]: 2 ctx1ab

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

36 

37 @declared_attr 2 ctx1ab

38 def organization(cls) -> Mapped[Organization | None]: 2 ctx1ab

39 return relationship(Organization, lazy="joined") 2 ctx1ab

40 

41 @property 2 ctx1ab

42 def scopes(self) -> list[str]: 2 ctx1ab

43 return self.scope.strip().split()