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

1from typing import TYPE_CHECKING 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import ForeignKey, UniqueConstraint, Uuid 2 ctx1ab

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

6 

7from polar.custom_field.attachment import AttachedCustomFieldMixin 2 ctx1ab

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

9 

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

11 from polar.models import Product 

12 

13 

14class ProductCustomField(AttachedCustomFieldMixin, RecordModel): 2 ctx1ab

15 __tablename__ = "product_custom_fields" 2 ctx1ab

16 __table_args__ = (UniqueConstraint("product_id", "order"),) 2 ctx1ab

17 

18 product_id: Mapped[UUID] = mapped_column( 2 ctx1ab

19 Uuid, 

20 ForeignKey("products.id", ondelete="cascade"), 

21 primary_key=True, 

22 ) 

23 

24 @declared_attr 2 ctx1ab

25 def product(cls) -> Mapped["Product"]: 2 ctx1ab

26 return relationship( 2 ctx1ab

27 "Product", lazy="raise", back_populates="attached_custom_fields" 

28 )