Coverage for polar/models/license_key_activation.py: 86%

1from typing import TYPE_CHECKING, Any 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import ( 2 ctx1ab

5 ForeignKey, 

6 String, 

7 Uuid, 

8) 

9from sqlalchemy.dialects.postgresql import JSONB 2 ctx1ab

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

11 

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

13from polar.kit.utils import utc_now 2 ctx1ab

14 

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

16 from .license_key import LicenseKey 

17 

18 

19class LicenseKeyActivation(RecordModel): 2 ctx1ab

20 __tablename__ = "license_key_activations" 2 ctx1ab

21 

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

23 Uuid, 

24 ForeignKey("license_keys.id", ondelete="cascade"), 

25 nullable=False, 

26 index=True, 

27 ) 

28 

29 @declared_attr 2 ctx1ab

30 def license_key(cls) -> Mapped["LicenseKey"]: 2 ctx1ab

31 return relationship("LicenseKey", lazy="raise", back_populates="activations") 2 ctx1ab

32 

33 label: Mapped[str] = mapped_column(String, nullable=False) 2 ctx1ab

34 

35 conditions: Mapped[dict[str, Any]] = mapped_column( 2 ctx1ab

36 JSONB, nullable=False, default=dict 

37 ) 

38 

39 meta: Mapped[dict[str, Any]] = mapped_column(JSONB, nullable=False, default=dict) 2 ctx1ab

40 

41 def mark_deleted(self) -> None: 2 ctx1ab

42 self.deleted_at = utc_now()