mirror of
https://github.com/maxdorninger/MediaManager.git
synced 2026-04-19 14:54:13 +02:00
a39e0d204a
This PR enables the ruff rule for return type annotations (ANN), and adds the ty package for type checking.
43 lines
733 B
Python
43 lines
733 B
Python
import typing
|
|
import uuid
|
|
from enum import Enum
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
TorrentId = typing.NewType("TorrentId", uuid.UUID)
|
|
|
|
|
|
class Quality(Enum):
|
|
uhd = 1
|
|
fullhd = 2
|
|
hd = 3
|
|
sd = 4
|
|
unknown = 5
|
|
|
|
|
|
class QualityStrings(Enum):
|
|
uhd = "4K"
|
|
fullhd = "1080p"
|
|
hd = "720p"
|
|
sd = "400p"
|
|
unknown = "unknown"
|
|
|
|
|
|
class TorrentStatus(Enum):
|
|
finished = 1
|
|
downloading = 2
|
|
error = 3
|
|
unknown = 4
|
|
|
|
|
|
class Torrent(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: TorrentId = Field(default_factory=lambda: TorrentId(uuid.uuid4()))
|
|
status: TorrentStatus
|
|
title: str
|
|
quality: Quality
|
|
imported: bool
|
|
hash: str
|
|
usenet: bool = False
|