Coverage for polar/models/wallet_transaction.py: 88%

1from datetime import datetime 2 ctx1ab

2from typing import TYPE_CHECKING 2 ctx1ab

3from uuid import UUID 2 ctx1ab

4 

5from sqlalchemy import TIMESTAMP, ForeignKey, String, Uuid 2 ctx1ab

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

7from sqlalchemy.sql.sqltypes import BigInteger 2 ctx1ab

8 

9from polar.kit.db.models import IDModel 2 ctx1ab

10from polar.kit.utils import utc_now 2 ctx1ab

11 

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

13 from .order import Order 

14 from .refund import Refund 

15 from .wallet import Wallet 

16 

17 

18class WalletTransaction(IDModel): 2 ctx1ab

19 __tablename__ = "wallet_transactions" 2 ctx1ab

20 

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

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

23 ) 

24 currency: Mapped[str] = mapped_column(String(3)) 2 ctx1ab

25 amount: Mapped[int] = mapped_column(BigInteger) 2 ctx1ab

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

27 Uuid, ForeignKey("wallets.id", ondelete="restrict"), index=True 

28 ) 

29 

30 tax_amount: Mapped[int | None] = mapped_column( 2 ctx1ab

31 BigInteger, nullable=True, default=None 

32 ) 

33 tax_calculation_processor_id: Mapped[str | None] = mapped_column( 2 ctx1ab

34 String, nullable=True, default=None 

35 ) 

36 

37 order_id: Mapped[UUID | None] = mapped_column( 2 ctx1ab

38 "order_id", 

39 Uuid, 

40 ForeignKey("orders.id", ondelete="restrict"), 

41 index=True, 

42 nullable=True, 

43 ) 

44 

45 refund_id: Mapped[UUID | None] = mapped_column( 2 ctx1ab

46 "refund_id", 

47 Uuid, 

48 ForeignKey("refunds.id", ondelete="restrict"), 

49 index=True, 

50 nullable=True, 

51 ) 

52 

53 @declared_attr 2 ctx1ab

54 def wallet(cls) -> Mapped["Wallet"]: 2 ctx1ab

55 return relationship("Wallet", lazy="raise_on_sql") 2 ctx1ab

56 

57 @declared_attr 2 ctx1ab

58 def order(cls) -> Mapped["Order | None"]: 2 ctx1ab

59 return relationship("Order", lazy="raise_on_sql") 2 ctx1ab

60 

61 @declared_attr 2 ctx1ab

62 def refund(cls) -> Mapped["Refund | None"]: 2 ctx1ab

63 return relationship("Refund", lazy="raise_on_sql") 2 ctx1ab