Coverage for polar/models/subscription_meter.py: 80%

1from decimal import Decimal 2 ctx1ab

2from typing import TYPE_CHECKING 2 ctx1ab

3from uuid import UUID 2 ctx1ab

4 

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

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

7from sqlalchemy.sql.sqltypes import Integer 2 ctx1ab

8 

9from polar.kit.db.models.base import RecordModel 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 .meter import Meter 

13 from .subscription import Subscription 

14 

15 

16class SubscriptionMeter(RecordModel): 2 ctx1ab

17 __tablename__ = "subscription_meters" 2 ctx1ab

18 __table_args__ = (UniqueConstraint("subscription_id", "meter_id"),) 2 ctx1ab

19 

20 consumed_units: Mapped[Decimal] = mapped_column(Numeric, nullable=False, default=0) 2 ctx1ab

21 credited_units: Mapped[int] = mapped_column(Integer, nullable=False, default=0) 2 ctx1ab

22 amount: Mapped[int] = mapped_column(Integer, nullable=False, default=0) 2 ctx1ab

23 

24 subscription_id: Mapped[UUID] = mapped_column( 2 ctx1ab

25 Uuid, ForeignKey("subscriptions.id", ondelete="cascade"), nullable=False 

26 ) 

27 meter_id: Mapped[UUID] = mapped_column( 2 ctx1ab

28 Uuid, ForeignKey("meters.id", ondelete="cascade"), nullable=False 

29 ) 

30 

31 @declared_attr 2 ctx1ab

32 def subscription(cls) -> Mapped["Subscription"]: 2 ctx1ab

33 return relationship( 2 ctx1ab

34 "Subscription", 

35 lazy="raise_on_sql", 

36 back_populates="meters", 

37 # cascade="all, delete-orphan", 

38 ) 

39 

40 @declared_attr 2 ctx1ab

41 def meter(cls) -> Mapped["Meter"]: 2 ctx1ab

42 return relationship("Meter", lazy="raise_on_sql") 2 ctx1ab

43 

44 def reset(self) -> None: 2 ctx1ab

45 self.consumed_units = Decimal(0) 

46 self.credited_units = 0 

47 self.amount = 0