Coverage for polar/models/trial_redemption.py: 91%

1from typing import TYPE_CHECKING 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import ForeignKey, String, Uuid 2 ctx1ab

5from sqlalchemy.ext.associationproxy import AssociationProxy, association_proxy 2 ctx1ab

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

7 

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

9 

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

11 from polar.models import Customer, Organization, Product 

12 

13 

14class TrialRedemption(RecordModel): 2 ctx1ab

15 __tablename__ = "trial_redemptions" 2 ctx1ab

16 

17 customer_email: Mapped[str] = mapped_column(String, nullable=False, index=True) 2 ctx1ab

18 payment_method_fingerprint: Mapped[str | None] = mapped_column( 2 ctx1ab

19 String, nullable=True, index=True 

20 ) 

21 

22 customer_id: Mapped[UUID] = mapped_column( 2 ctx1ab

23 Uuid, ForeignKey("customers.id", ondelete="cascade"), nullable=False, index=True 

24 ) 

25 

26 @declared_attr 2 ctx1ab

27 def customer(cls) -> Mapped["Customer"]: 2 ctx1ab

28 return relationship("Customer", lazy="raise") 2 ctx1ab

29 

30 organization: AssociationProxy["Organization"] = association_proxy( 2 ctx1ab

31 "customer", "organization" 

32 ) 

33 

34 product_id: Mapped[UUID | None] = mapped_column( 2 ctx1ab

35 Uuid, ForeignKey("products.id", ondelete="cascade"), nullable=True, index=True 

36 ) 

37 

38 @declared_attr 2 ctx1ab

39 def product(cls) -> Mapped["Product | None"]: 2 ctx1ab

40 return relationship("Product", lazy="raise") 2 ctx1ab