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

1from uuid import UUID 

2 

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

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

5 

6from polar.kit.db.models import RecordModel 

7from polar.models.organization import Organization 

8from polar.models.user import User 

9 

10 

11class OAuth2Grant(RecordModel): 

12 __tablename__ = "oauth2_grants" 

13 __table_args__ = ( 

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) 

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

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

21 Uuid, 

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

23 nullable=True, 

24 index=True, 

25 ) 

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

27 Uuid, 

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

29 nullable=True, 

30 index=True, 

31 ) 

32 

33 @declared_attr 

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

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

36 

37 @declared_attr 

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

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

40 

41 @property 

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

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