Coverage for polar/models/oauth2_client.py: 78%

1from typing import TYPE_CHECKING 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from authlib.integrations.sqla_oauth2 import OAuth2ClientMixin 2 ctx1ab

5from sqlalchemy import ForeignKey, String, UniqueConstraint, Uuid 2 ctx1ab

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

7 

8from polar.kit.db.models import RateLimitGroupMixin, RecordModel 2 ctx1ab

9from polar.oauth2.sub_type import SubType 2 ctx1ab

10 

11if TYPE_CHECKING: 11 ↛ 12line 11 didn't jump to line 12 because the condition on line 11 was never true2 ctx1ab

12 from polar.models import User 

13 

14 

15class OAuth2Client(RateLimitGroupMixin, RecordModel, OAuth2ClientMixin): 2 ctx1ab

16 __tablename__ = "oauth2_clients" 2 ctx1ab

17 __table_args__ = (UniqueConstraint("client_id"),) 2 ctx1ab

18 

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

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

21 registration_access_token: Mapped[str] = mapped_column( 2 ctx1ab

22 String, index=True, nullable=False 

23 ) 

24 

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

26 Uuid, ForeignKey("users.id"), nullable=True, index=True 

27 ) 

28 

29 @declared_attr 2 ctx1ab

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

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

32 

33 @property 2 ctx1ab

34 def default_sub_type(self) -> SubType: 2 ctx1ab

35 try: 

36 return SubType(self.client_metadata["default_sub_type"]) 

37 except KeyError: 

38 return SubType.organization