mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-25 09:18:50 -05:00
104 lines
2.5 KiB
Python
104 lines
2.5 KiB
Python
import fanart
|
|
from fanart.items import LeafItem, Immutable, ResourceItem
|
|
__all__ = (
|
|
'ArtItem',
|
|
'DiscItem',
|
|
'LogoItem',
|
|
'PosterItem',
|
|
'BackgroundItem',
|
|
'HdLogoItem',
|
|
'HdArtItem',
|
|
'BannerItem',
|
|
'ThumbItem',
|
|
'Movie',
|
|
)
|
|
|
|
|
|
class MovieItem(LeafItem):
|
|
|
|
@Immutable.mutablemethod
|
|
def __init__(self, id, url, likes, lang):
|
|
super(MovieItem, self).__init__(id, url, likes)
|
|
self.lang = lang
|
|
|
|
|
|
class DiscItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.DISC
|
|
|
|
@Immutable.mutablemethod
|
|
def __init__(self, id, url, likes, lang, disc, disc_type):
|
|
super(DiscItem, self).__init__(id, url, likes, lang)
|
|
self.disc = int(disc)
|
|
self.disc_type = disc_type
|
|
|
|
|
|
class ArtItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.ART
|
|
|
|
|
|
class LogoItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.LOGO
|
|
|
|
|
|
class PosterItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.POSTER
|
|
|
|
|
|
class BackgroundItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.BACKGROUND
|
|
|
|
|
|
class HdLogoItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.HDLOGO
|
|
|
|
|
|
class HdArtItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.HDART
|
|
|
|
|
|
class BannerItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.BANNER
|
|
|
|
|
|
class ThumbItem(MovieItem):
|
|
KEY = fanart.TYPE.MOVIE.THUMB
|
|
|
|
|
|
class Movie(ResourceItem):
|
|
WS = fanart.WS.MOVIE
|
|
|
|
@Immutable.mutablemethod
|
|
def __init__(self, name, imdbid, tmdbid, arts, logos, discs, posters, backgrounds, hdlogos, hdarts,
|
|
banners, thumbs):
|
|
self.name = name
|
|
self.imdbid = imdbid
|
|
self.tmdbid = tmdbid
|
|
self.arts = arts
|
|
self.posters = posters
|
|
self.logos = logos
|
|
self.discs = discs
|
|
self.backgrounds = backgrounds
|
|
self.hdlogos = hdlogos
|
|
self.hdarts = hdarts
|
|
self.banners = banners
|
|
self.thumbs = thumbs
|
|
|
|
@classmethod
|
|
def from_dict(cls, resource):
|
|
assert len(resource) == 1, 'Bad Format Map'
|
|
name, resource = resource.items()[0]
|
|
return cls(
|
|
name=name,
|
|
imdbid=resource['imdb_id'],
|
|
tmdbid=resource['tmdb_id'],
|
|
arts=ArtItem.extract(resource),
|
|
logos=LogoItem.extract(resource),
|
|
discs=DiscItem.extract(resource),
|
|
posters=PosterItem.extract(resource),
|
|
backgrounds=BackgroundItem.extract(resource),
|
|
hdlogos=HdLogoItem.extract(resource),
|
|
hdarts=HdArtItem.extract(resource),
|
|
banners=BannerItem.extract(resource),
|
|
thumbs=ThumbItem.extract(resource),
|
|
)
|