Create python types

This commit is contained in:
Zoe Roux 2023-03-18 17:42:55 +09:00
parent 424390417a
commit edff93917e
3 changed files with 59 additions and 0 deletions

View File

@ -0,0 +1,22 @@
from enum import Enum
class Genre(Enum):
ACTION = "action"
ADVENTURE = "adventure"
ANIMATION = "animation"
COMEDY = "comedy"
CRIME = "crime"
DOCUMENTARY = "documentary"
DRAMA = "drama"
FAMILY = "family"
FANTASY = "fantasy"
HISTORY = "history"
HORROR = "horror"
MUSIC = "music"
MYSTERY = "mystery"
ROMANCE = "romance"
SCIENCE_FICTION = "scienceFiction"
THRILLER = "thriller"
WAR = "war"
WESTERN = "western"

View File

@ -0,0 +1,30 @@
from dataclasses import dataclass, field
from datetime import datetime
from typing import Optional
from .genre import Genre
from .status import Status
@dataclass
class MovieTranslation:
name: str
keywords: list[str] = field(default_factory=list)
overview: Optional[str] = None
@dataclass
class Movie:
aliases: list[str] = field(default_factory=list)
release_date: Optional[datetime | int] = None
status: Status = Status.UNKNOWN
studio: Optional[int | str] = None
genres: list[Genre] = field(default_factory=list)
poster: list[str] = field(default_factory=list)
thumbnails: list[str] = field(default_factory=list)
logo: list[str] = field(default_factory=list)
path: Optional[str] = None
# TODO: handle staff
# staff: list[Staff]
translations: dict[str, MovieTranslation] = field(default_factory=dict)

View File

@ -0,0 +1,7 @@
from enum import Enum
class Status(Enum):
UNKNOWN = "unknown"
FINISHED = "finished"
AIRING = "airing"
PLANNED = "planned"