43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import enum
|
|
from typing import TYPE_CHECKING
|
|
from uuid import UUID
|
|
|
|
from tortoise import fields
|
|
|
|
from .base import TimestampedModel
|
|
|
|
if TYPE_CHECKING:
|
|
from .placement import Placement
|
|
from .subscriber import Subscriber
|
|
from .target_channel import TargetChannel
|
|
|
|
|
|
class SubscriptionStatus(str, enum.Enum):
|
|
ACTIVE = 'active'
|
|
UNSUBSCRIBED = 'unsubscribed'
|
|
|
|
|
|
class Subscription(TimestampedModel):
|
|
id = fields.UUIDField(pk=True)
|
|
invite_link = fields.CharField(max_length=512, index=True)
|
|
status = fields.CharEnumField(SubscriptionStatus, default=SubscriptionStatus.ACTIVE)
|
|
unsubscribed_at = fields.DatetimeField(null=True)
|
|
|
|
placement: fields.ForeignKeyRelation['Placement'] = fields.ForeignKeyField(
|
|
'models.Placement', related_name='subscriptions', on_delete=fields.CASCADE, index=True
|
|
)
|
|
subscriber: fields.ForeignKeyRelation['Subscriber'] = fields.ForeignKeyField(
|
|
'models.Subscriber', related_name='subscriptions', on_delete=fields.CASCADE, index=True
|
|
)
|
|
target_channel: fields.ForeignKeyRelation['TargetChannel'] = fields.ForeignKeyField(
|
|
'models.TargetChannel', related_name='subscriptions', on_delete=fields.CASCADE, index=True
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
placement_id: UUID
|
|
subscriber_id: UUID
|
|
target_channel_id: UUID
|
|
|
|
class Meta:
|
|
table = 'subscription'
|