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

1from datetime import datetime 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import CHAR, TIMESTAMP, ForeignKey, String, Uuid 2 ctx1ab

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

6 

7from polar.config import settings 2 ctx1ab

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

9from polar.kit.utils import utc_now 2 ctx1ab

10from polar.models.customer import Customer 2 ctx1ab

11 

12 

13def get_expires_at() -> datetime: 2 ctx1ab

14 return utc_now() + settings.CUSTOMER_SESSION_CODE_TTL 

15 

16 

17class CustomerSessionCode(RecordModel): 2 ctx1ab

18 __tablename__ = "customer_session_codes" 2 ctx1ab

19 

20 code: Mapped[str] = mapped_column(CHAR(64), unique=True, nullable=False) 2 ctx1ab

21 expires_at: Mapped[datetime] = mapped_column( 2 ctx1ab

22 TIMESTAMP(timezone=True), nullable=False, index=True, default=get_expires_at 

23 ) 

24 email: Mapped[str] = mapped_column(String(320), nullable=False) 2 ctx1ab

25 

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

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

28 ) 

29 

30 @declared_attr 2 ctx1ab

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

32 return relationship(Customer, lazy="joined") 2 ctx1ab