62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from typing import Iterator
|
|
|
|
from tortoise import fields
|
|
from tortoise.models import Model
|
|
from tortoise.transactions import atomic
|
|
|
|
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,
|
|
)
|
|
|
|
async def build_publications(self, publishers: Iterator[str]):
|
|
return [
|
|
await self.build_publication_by_publisher_name(name) for name in publishers
|
|
]
|
|
|
|
@atomic()
|
|
async def get_failed_publications(self,) -> list[Publication]:
|
|
return list(
|
|
filter(
|
|
lambda publications: publications.status == PublicationStatus.FAILED,
|
|
self.publications,
|
|
)
|
|
)
|