Mobilizon-Reshare-condividi-sui-social/mobilizon_reshare/models/event.py

from tortoise import fields
from tortoise.models import Model

from mobilizon_reshare.models import WithPydantic
from mobilizon_reshare.models.publication import PublicationStatus, Publication
from mobilizon_reshare.models.publisher import Publisher


class Event(Model, WithPydantic):
    id = fields.UUIDField(pk=True)
    name = fields.TextField()
    description = fields.TextField(null=True)

    mobilizon_id = fields.UUIDField()
    mobilizon_link = fields.TextField()
    thumbnail_link = fields.TextField(null=True)

    location = fields.TextField(null=True)

    begin_datetime = fields.DatetimeField()
    end_datetime = fields.DatetimeField()
    last_update_time = fields.DatetimeField()

    publications: fields.ReverseRelation["Publication"]

    def __str__(self):
        return str(self.name)

    def __repr__(self):
        return f"{self.id} - {self.name}"

    class Meta:
        table = "event"

    async def build_publication_by_publisher_name(
        self, publisher_name: str, status: PublicationStatus = PublicationStatus.FAILED
    ) -> Publication:
        publisher = await Publisher.filter(name=publisher_name).first()
        return Publication(
            status=status,
            event_id=self.id,
            publisher_id=publisher.id,
            publisher=publisher,
        )