Skip to content

Commit

Permalink
Don't use Dataclass
Browse files Browse the repository at this point in the history
  • Loading branch information
@Teagan42
Teagan42 authored Sep 8, 2024
1 parent 2142516 commit 41793fd
Showing 1 changed file with 50 additions and 51 deletions.
101 changes: 50 additions & 51 deletions varken/structures.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
exit(1)


@dataclass
class BaseModel:
"""Base class to handle dynamic fields from APIs."""

Expand Down Expand Up @@ -135,7 +134,7 @@ class QueuePages(BaseModel):
sortKey: str = None
sortDirection: str = None
totalRecords: str = None
records: Optional[list] = field(default_factory=list, default=None, init=None)
records: Optional[list] = None


# Ombi Structures
Expand All @@ -154,7 +153,7 @@ class OmbiIssuesCounts(BaseModel):
@dataclass
class OmbiTVRequest(BaseModel):
background: str = None
childRequests: Optional[list] = field(default_factory=list, default=None, init=None)
childRequests: Optional[list] = None
denied: bool = None
deniedReason: None = None
externalProviderId: str = None
Expand Down Expand Up @@ -211,7 +210,7 @@ class OmbiMovieRequest(BaseModel):
requestedByAlias: str = None
requestedDate: str = None
requestedDate4k: str = None
requestedUser: Optional[dict] = field(default_factory=dict, default=None, init=None)
requestedUser: Optional[dict] = None
requestedUserId: str = None
requestStatus: str = None
requestType: int = None
Expand Down Expand Up @@ -243,14 +242,14 @@ class OverseerrRequestCounts(BaseModel):
class SonarrTVShow(BaseModel):
added: str = None
airTime: str = None
alternateTitles: Optional[list] = field(default_factory=list, default=None, init=None)
alternateTitles: Optional[list] = None
certification: str = None
cleanTitle: str = None
ended: bool = None
firstAired: str = None
genres: Optional[list] = field(default_factory=list, default=None, init=None)
genres: Optional[list] = None
id: int = None
images: Optional[list] = field(default_factory=list, default=None, init=None)
images: Optional[list] = None
imdbId: str = None
languageProfileId: int = None
monitored: bool = None
Expand All @@ -260,16 +259,16 @@ class SonarrTVShow(BaseModel):
path: str = None
previousAiring: str = None
qualityProfileId: int = None
ratings: Optional[dict] = field(default_factory=dict, default=None, init=None)
ratings: Optional[dict] = None
rootFolderPath: str = None
runtime: int = None
seasonFolder: bool = None
seasons: Optional[list] = field(default_factory=list, default=None, init=None)
seasons: Optional[list] = None
seriesType: str = None
sortTitle: str = None
statistics: Optional[dict] = field(default_factory=dict, default=None, init=None)
statistics: Optional[dict] = None
status: str = None
tags: Optional[list] = field(default_factory=list, default=None, init=None)
tags: Optional[list] = None
title: str = None
titleSlug: str = None
tvdbId: int = None
Expand Down Expand Up @@ -298,14 +297,14 @@ class SonarrEpisode(BaseModel):
sceneAbsoluteEpisodeNumber: int = None
sceneEpisodeNumber: int = None
sceneSeasonNumber: int = None
series: Optional[SonarrTVShow] = field(default_factory=SonarrTVShow, default=None, init=None)
series: Optional[SonarrTVShow] = None
tvdbId: int = None
finaleType: str = None
episodeFile: Optional[dict] = field(default_factory=dict, default=None, init=None)
episodeFile: Optional[dict] = None
endTime: str = None
grabTime: str = None
seriesTitle: str = None
images: Optional[list] = field(default_factory=list, default=None, init=None)
images: Optional[list] = None

@dataclass
class SonarrQueue(BaseModel):
Expand All @@ -314,21 +313,21 @@ class SonarrQueue(BaseModel):
episodeId: int = None
id: int = None
indexer: str = None
language: Optional[dict] = field(default_factory=dict, default=None, init=None)
language: Optional[dict] = None
protocol: str = None
quality: Optional[dict] = field(default_factory=dict, default=None, init=None)
quality: Optional[dict] = None
size: float = None
sizeleft: float = None
status: str = None
statusMessages: Optional[list] = field(default_factory=list, default=None, init=None)
statusMessages: Optional[list] = None
title: str = None
trackedDownloadState: str = None
trackedDownloadStatus: str = None
seriesId: int = None
errorMessage: str = None
outputPath: str = None
series: Optional[SonarrTVShow] = field(default_factory=SonarrTVShow, default=None, init=None)
episode: Optional[SonarrEpisode] = field(default_factory=SonarrEpisode, default=None, init=None)
series: Optional[SonarrTVShow] = None
episode: Optional[SonarrEpisode] = None
timeleft: str = None
estimatedCompletionTime: str = None

Expand All @@ -337,36 +336,36 @@ class SonarrQueue(BaseModel):
@dataclass
class RadarrMovie(BaseModel):
added: str = None
alternateTitles: Optional[list] = field(default_factory=list, default=None, init=None)
alternateTitles: Optional[list] = None
certification: str = None
cleanTitle: str = None
collection: Optional[dict] = field(default_factory=dict, default=None, init=None)
collection: Optional[dict] = None
digitalRelease: str = None
folderName: str = None
genres: Optional[list] = field(default_factory=list, default=None, init=None)
genres: Optional[list] = None
hasFile: bool = None
id: int = None
images: Optional[list] = field(default_factory=list, default=None, init=None)
images: Optional[list] = None
imdbId: str = None
inCinemas: str = None
isAvailable: bool = None
minimumAvailability: str = None
monitored: bool = None
movieFile: Optional[dict] = field(default_factory=dict, default=None, init=None)
movieFile: Optional[dict] = None
originalTitle: str = None
overview: str = None
path: str = None
physicalRelease: str = None
qualityProfileId: int = None
ratings: Optional[dict] = field(default_factory=dict, default=None, init=None)
ratings: Optional[dict] = None
runtime: int = None
secondaryYear: int = None
secondaryYearSourceId: int = None
sizeOnDisk: float = None
sortTitle: str = None
status: str = None
studio: str = None
tags: Optional[list] = field(default_factory=list, default=None, init=None)
tags: Optional[list] = None
titleSlug: str = None
tmdbId: int = None
website: str = None
Expand All @@ -377,34 +376,34 @@ class RadarrMovie(BaseModel):
addOptions: str = None
popularity: str = None
rootFolderPath: str = None
statistics: Optional[dict] = field(default_factory=dict, default=None, init=None)
statistics: Optional[dict] = None
movieFileId: int = None


# Radarr Queue
@dataclass
class RadarrQueue(BaseModel):
customFormats: Optional[list] = field(default_factory=list, default=None, init=None)
customFormats: Optional[list] = None
downloadClient: str = None
downloadId: str = None
id: int = None
indexer: str = None
languages: Optional[list] = field(default_factory=list, default=None, init=None)
languages: Optional[list] = None
movieId: int = None
protocol: str = None
quality: Optional[dict] = field(default_factory=dict, default=None, init=None)
quality: Optional[dict] = None
size: float = None
sizeleft: float = None
status: str = None
statusMessages: Optional[list] = field(default_factory=list, default=None, init=None)
statusMessages: Optional[list] = None
title: str = None
trackedDownloadState: str = None
trackedDownloadStatus: str = None
timeleft: str = None
estimatedCompletionTime: str = None
errorMessage: str = None
outputPath: str = None
movie: Optional[RadarrMovie] = field(default_factory=RadarrMovie, default=None, init=None)
movie: Optional[RadarrMovie] = None
timeleft: str = None
customFormatScore: int = None
indexerFlags: int = None
Expand Down Expand Up @@ -434,7 +433,7 @@ class SickChillTVShow(BaseModel):
# Tautulli
@dataclass
class TautulliStream(BaseModel):
actors: Optional[list] = field(default_factory=list, default=None, init=None)
actors: Optional[list] = None
added_at: str = None
allow_guest: int = None
art: str = None
Expand All @@ -459,14 +458,14 @@ class TautulliStream(BaseModel):
channel_stream: int = None
channel_title: str = None
children_count: str = None
collections: Optional[list] = field(default_factory=list, default=None, init=None)
collections: Optional[list] = None
container: str = None
content_rating: str = None
current_session: str = None
date: str = None
deleted_user: int = None
device: str = None
directors: Optional[list] = field(default_factory=list, default=None, init=None)
directors: Optional[list] = None
do_notify: int = None
duration: str = None
email: str = None
Expand All @@ -475,7 +474,7 @@ class TautulliStream(BaseModel):
file_size: str = None
friendly_name: str = None
full_title: str = None
genres: Optional[list] = field(default_factory=list, default=None, init=None)
genres: Optional[list] = None
grandparent_guid: str = None
grandparent_rating_key: str = None
grandparent_thumb: str = None
Expand All @@ -493,7 +492,7 @@ class TautulliStream(BaseModel):
is_home_user: int = None
is_restricted: int = None
keep_history: int = None
labels: Optional[list] = field(default_factory=list, default=None, init=None)
labels: Optional[list] = None
last_viewed_at: str = None
library_name: str = None
live: int = None
Expand Down Expand Up @@ -537,7 +536,7 @@ class TautulliStream(BaseModel):
selected: int = None
session_id: str = None
session_key: str = None
shared_libraries: Optional[list] = field(default_factory=list, default=None, init=None)
shared_libraries: Optional[list] = None
sort_title: str = None
started: int = None
state: str = None
Expand Down Expand Up @@ -645,7 +644,7 @@ class TautulliStream(BaseModel):
view_offset: str = None
watched_status: int = None
width: str = None
writers: Optional[list] = field(default_factory=list, default=None, init=None)
writers: Optional[list] = None
year: str = None


Expand All @@ -654,16 +653,16 @@ class TautulliStream(BaseModel):
class LidarrQueue(BaseModel):
artistId: int = None
albumId: int = None
language: Optional[dict] = field(default_factory=dict, default=None, init=None)
quality: Optional[dict] = field(default_factory=dict, default=None, init=None)
language: Optional[dict] = None
quality: Optional[dict] = None
size: float = None
title: str = None
timeleft: str = None
sizeleft: float = None
status: str = None
trackedDownloadStatus: str = None
trackedDownloadState: str = None
statusMessages: Optional[list] = field(default_factory=list, default=None, init=None)
statusMessages: Optional[list] = None
errorMessage: str = None
downloadId: str = None
protocol: str = None
Expand All @@ -686,15 +685,15 @@ class LidarrAlbum(BaseModel):
profileId: int = None
duration: int = None
albumType: str = None
secondaryTypes: Optional[list] = field(default_factory=list, default=None, init=None)
secondaryTypes: Optional[list] = None
mediumCount: int = None
ratings: Optional[dict] = field(default_factory=dict, default=None, init=None)
ratings: Optional[dict] = None
releaseDate: str = None
releases: Optional[list] = field(default_factory=list, default=None, init=None)
genres: Optional[list] = field(default_factory=list, default=None, init=None)
media: Optional[list] = field(default_factory=list, default=None, init=None)
artist: Optional[dict] = field(default_factory=dict, default=None, init=None)
images: Optional[list] = field(default_factory=list, default=None, init=None)
links: Optional[list] = field(default_factory=list, default=None, init=None)
statistics: dict = field(default_factory=dict, default={}, init=None)
releases: Optional[list] = None
genres: Optional[list] = None
media: Optional[list] = None
artist: Optional[dict] = None
images: Optional[list] = None
links: Optional[list] = None
statistics: dict = {}
id: int = None

0 comments on commit 41793fd

Please sign in to comment.