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

1from enum import StrEnum 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import BigInteger, ForeignKey, String, Uuid 2 ctx1ab

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

6 

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

8from polar.models.issue_reward import IssueReward 2 ctx1ab

9from polar.models.pledge import Pledge 2 ctx1ab

10 

11 

12class PledgeTransactionType(StrEnum): 2 ctx1ab

13 pledge = "pledge" 2 ctx1ab

14 transfer = "transfer" 2 ctx1ab

15 refund = "refund" 2 ctx1ab

16 disputed = "disputed" 2 ctx1ab

17 

18 

19class PledgeTransaction(RecordModel): 2 ctx1ab

20 __tablename__ = "pledge_transactions" 2 ctx1ab

21 

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

23 Uuid, ForeignKey("pledges.id"), nullable=False 

24 ) 

25 type: Mapped[PledgeTransactionType] = mapped_column(String, nullable=False) 2 ctx1ab

26 amount: Mapped[int] = mapped_column(BigInteger, nullable=False) 2 ctx1ab

27 transaction_id: Mapped[str | None] = mapped_column( 2 ctx1ab

28 String, nullable=True, default=None 

29 ) 

30 issue_reward_id: Mapped[UUID | None] = mapped_column( 2 ctx1ab

31 Uuid, ForeignKey("issue_rewards.id"), nullable=True, default=None 

32 ) 

33 

34 @declared_attr 2 ctx1ab

35 def pledge(cls) -> Mapped[Pledge]: 2 ctx1ab

36 return relationship( 2 ctx1ab

37 Pledge, 

38 lazy="raise", 

39 ) 

40 

41 @declared_attr 2 ctx1ab

42 def issue_reward(cls) -> Mapped[IssueReward]: 2 ctx1ab

43 return relationship( 2 ctx1ab

44 IssueReward, 

45 lazy="raise", 

46 )