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

1from uuid import UUID 2 ctx1ab

2 

3from sqlalchemy import ForeignKey, Uuid 2 ctx1ab

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

5 

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

7from polar.models.organization import Organization 2 ctx1ab

8from polar.models.user import User 2 ctx1ab

9 

10 

11class UserOrganization(TimestampedModel): 2 ctx1ab

12 __tablename__ = "user_organizations" 2 ctx1ab

13 

14 user_id: Mapped[UUID] = mapped_column( 2 ctx1ab

15 Uuid, 

16 ForeignKey("users.id"), 

17 nullable=False, 

18 primary_key=True, 

19 ) 

20 

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

22 Uuid, 

23 ForeignKey("organizations.id", ondelete="CASCADE"), 

24 nullable=False, 

25 primary_key=True, 

26 ) 

27 

28 @declared_attr 2 ctx1ab

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

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

31 

32 @declared_attr 2 ctx1ab

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

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