change domen

src/domain/purchase.py

@@ -0,0 +1,74 @@
+import enum
+import uuid
+from typing import TYPE_CHECKING
+
+from tortoise import fields
+
+from .base import TimestampedModel
+
+if TYPE_CHECKING:
+    from .channel import Channel
+    from .creative import Creative
+    from .project import Project
+    from .workspace import Workspace
+
+
+class PurchaseStatus(str, enum.Enum):
+    ACTIVE = 'active'
+    ARCHIVED = 'archived'
+
+
+class PurchaseChannelStatus(str, enum.Enum):
+    PLANNED = 'planned'
+    APPROVED = 'approved'
+    REJECTED = 'rejected'
+    IN_PROGRESS = 'in_progress'
+    COMPLETED = 'completed'
+
+
+class Purchase(TimestampedModel):
+    status = fields.CharEnumField(PurchaseStatus, default=PurchaseStatus.ACTIVE)
+
+    project: fields.ForeignKeyRelation['Project'] = fields.ForeignKeyField(
+        'models.Project', related_name='purchases', on_delete=fields.CASCADE, index=True
+    )
+    creative: fields.ForeignKeyRelation['Creative'] = fields.ForeignKeyField(
+        'models.Creative', related_name='purchases', on_delete=fields.CASCADE, index=True
+    )
+
+    workspace: fields.ForeignKeyRelation['Workspace'] = fields.ForeignKeyField(
+        'models.Workspace', related_name='purchases', on_delete=fields.CASCADE, index=True
+    )
+
+    class Meta:
+        table = 'purchase'
+        indexes = (('project', 'status'),)
+
+
+class InviteLinkType(str, enum.Enum):
+    PUBLIC = 'public'  # открытая ссылка
+    APPROVAL = 'approval'  # с одобрением ботом
+
+
+class PurchaseChannel(TimestampedModel):
+    status = fields.CharEnumField(PurchaseChannelStatus, default=PurchaseChannelStatus.PLANNED)
+    planned_cost = fields.FloatField(null=True)
+    comment = fields.TextField(null=True)
+
+    invite_link = fields.CharField(max_length=512)
+    invite_link_type = fields.CharEnumField(InviteLinkType)
+
+    purchase: fields.ForeignKeyRelation['Purchase'] = fields.ForeignKeyField(
+        'models.Purchase', related_name='channels', on_delete=fields.CASCADE, index=True
+    )
+    channel: fields.ForeignKeyRelation['Channel'] = fields.ForeignKeyField(
+        'models.Channel', related_name='purchase_channels', on_delete=fields.CASCADE, index=True
+    )
+
+    if TYPE_CHECKING:
+        purchase_id: uuid.UUID
+        channel_id: uuid.UUID
+
+    class Meta:
+        table = 'purchase_channel'
+        unique_together = (('purchase_id', 'channel_id'),)