38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import imdb
|
|
from .result import Metadata, DataResult, Summary
|
|
|
|
class metadata():
|
|
name: str = None
|
|
imdb = imdb.Cinemagoer()
|
|
|
|
def __init__(self, name) -> None:
|
|
self.name = name
|
|
|
|
|
|
def lookup(self) -> DataResult:
|
|
""""""
|
|
try:
|
|
query = self.imdb.search_movie(self.name)
|
|
imdbId = query[0].movieID
|
|
result = self.imdb.get_movie(imdbId)
|
|
meta = Metadata(
|
|
title = result.get("title", None),
|
|
altTitle = [result.get("localized title", [])],
|
|
cover = result.get("cover url", None),
|
|
summary = [
|
|
Summary(
|
|
language = "eng",
|
|
summary = result.get("plot outline", None)
|
|
)
|
|
],
|
|
type = 'movie' if result.get('kind', '').lower() == 'movie' else 'serie',
|
|
genres = result.get('genres', []),
|
|
source="imdb",
|
|
usedTitle=self.name
|
|
)
|
|
if (meta.title is None) or (meta.type is None):
|
|
return DataResult(status="COMPLETED", message= None, data= None)
|
|
|
|
return DataResult(status="COMPLETED", message= None, data= meta)
|
|
except Exception as e:
|
|
return DataResult(status="ERROR", data=None, message=str(e)) |