def __init__(self):
self._priority = 90
self._common = OFDBCommon()
self._attrs = [
'name', 'alternative_names', 'photo', 'placeofbirth', 'imdbid',
'deathday', 'known_for'
]
def __init__(self):
self._priority = 90
self._common = OFDBCommon()
self._genrenorm = GenreNormalize('ofdb.genre')
self._attrs = {
'title', 'original_title', 'plot', 'imdbid', 'vote_count',
'rating', 'directors', 'writers', 'outline', 'year', 'poster',
'countries', 'genre', 'genre_norm', 'actors'
}
class OFDBMovie(provider.IMovieProvider):
""" OFDB Movie text metadata provider.
Interfaces implemented according to hugin.provider.interfaces.
"""
def __init__(self):
self._priority = 90
self._common = OFDBCommon()
self._genrenorm = GenreNormalize('ofdb.genre')
self._attrs = {
'title', 'original_title', 'plot', 'imdbid', 'vote_count',
'rating', 'directors', 'writers', 'outline', 'year', 'poster',
'countries', 'genre', 'genre_norm', 'actors'
}
def build_url(self, search_params):
# not enough search params
if search_params.title is None and search_params.imdbid is None:
return None
# try to search by imdbid if available, else use title
if search_params.imdbid:
path, query = 'imdb2ofdb_json', search_params.imdbid
else:
path, query = 'search_json', quote_plus(search_params.title)
return [self._common.get_base_url().format(path=path, query=query)]
def parse_response(self, url_response, search_params):
# validate the response data
status, retvalue, url, response = self._common.validate_url_response(
url_response
)
# CHECK IF CRITICAL MATCHES
if status in ['retry', 'critical']:
return retvalue
# validate the response status of the provider
status, retv = self._common.check_response_status(response)
if status in ['critical', 'unknown', 'no_data']:
return retv
response_type = response['status']['modul']
response = response['resultat']
if 'search' in response_type:
return self._parse_search_module(response, search_params), False
if 'imdb2ofdb' in response_type:
return self._parse_imdb2ofdb_module(response), False
if 'movie' in response_type:
return self._parse_movie_module(response), True
return None, True
def _parse_imdb2ofdb_module(self, response):
""" Invoke url build method for movie id found in response. """
return self._common.movieids_to_urllist([response['ofdbid']])
def _parse_search_module(self, response, search_params):
""" Parse a search response. Find high prio result. Build urllist."""
# create similarity matrix for title, check agains german and original
# title, higher ratio wins
similarity_map = []
for response in response['eintrag']:
# Get the title with the highest similarity ratio:
ratio = 0.0
if '-Serie' not in response['titel_de']:
for title_key in ['titel_de', 'titel_orig']:
ratio = max(
ratio,
string_similarity_ratio(
response[title_key],
search_params.title
)
)
similarity_map.append(
{'ofdbid': response['id'], 'ratio': ratio}
)
# sort by ratio, generate ofdbid list with requestet item count
similarity_map.sort(
key=lambda value: value['ratio'],
reverse=True
)
item_count = min(len(similarity_map), search_params.amount)
matches = [item['ofdbid'] for item in similarity_map[:item_count]]
return self._common.movieids_to_urllist(matches)
def _parse_movie_module(self, response):
""" Fill in result_dict. """
result_dict = {k: None for k in self._attrs}
#str attrs
result_dict['title'] = response['titel']
result_dict['original_title'] = response['alternativ']
result_dict['plot'] = response['beschreibung']
result_dict['outline'] = response['kurzbeschreibung']
result_dict['imdbid'] = 'tt{0}'.format(response['imdbid'])
result_dict['rating'] = response['bewertung']['note']
# number attrs
result_dict['vote_count'] = int(response['bewertung'].get('stimmen') or 0)
result_dict['year'] = int(response['jahr'] or 0)
# list attrs
result_dict['poster'] = [(None, response['bild'])]
result_dict['countries'] = response['produktionsland']
result_dict['actors'] = self._extract_person(
response['besetzung'], 'actors'
)
try:
result_dict['directors'] = [r['name'] for r in response['regie']]
except:
print('invalid value')
result_dict['writers'] = self._extract_person(
response['drehbuch'], 'directors'
)
result_dict['genre'] = response['genre']
result_dict['genre_norm'] = self._genrenorm.normalize_genre_list(
result_dict['genre']
)
return result_dict
def _extract_person(self, persons, person_type):
""" Extract person information from person response part. """
person_list = []
try:
for person in persons:
role, name = person.get('rolle'), person.get('name')
if person_type == 'actors':
person_list.append((role, name))
elif person_type == 'directors':
person_list.append(name)
except (AttributeError, TypeError):
return []
return person_list
@property
def supported_attrs(self):
return self._attrs
class OFDBPerson(provider.IPersonProvider):
""" OFDB Person text metadata provider.
Interfaces implemented according to hugin.provider.interfaces.
"""
def __init__(self):
self._priority = 90
self._common = OFDBCommon()
self._attrs = [
'name', 'alternative_names', 'photo', 'placeofbirth', 'imdbid',
'deathday', 'known_for'
]
def build_url(self, search_params):
if search_params.name:
return [
self._common.get_base_url().format(
path='searchperson_json',
query=quote(search_params.name))
]
def parse_response(self, url_response, search_params):
# validate the response data
status, retvalue, url, response = self._common.validate_url_response(
url_response
)
if status in ['retry', 'critical']:
return retvalue
# validate the response status of the provider
status, retv = self._common.check_response_status(response)
if status in ['critical', 'unknown', 'no_data']:
return retv
response_type = response['status']['modul']
response = response['resultat']
if 'searchperson' in response_type:
return self._parse_search_module(response, search_params), False
if 'person' in response_type:
return self._parse_person_module(response, search_params), True
return None, True
def _parse_search_module(self, response, search_params):
""" Parse a search response. Find high prio result. Build urllist."""
similarity_map = []
for response in response['eintrag']:
if response['wert'].isnumeric():
# a 'hack' for better name matching, as name string often looks
# like this 'Emma Roberts (10.02.1991) alias Emma Rose Roberts'
clean_name, *_ = response['name'].split('(')
ratio = string_similarity_ratio(
clean_name,
search_params.name
)
similarity_map.append(
{'ofdbid': response['wert'],
'ratio': ratio, 'name': response['name']}
)
# sort by ratio, generate ofdbid list with requestet item count
similarity_map.sort(
key=lambda value: value['ratio'],
reverse=True
)
item_count = min(len(similarity_map), search_params.amount)
personids = [item['ofdbid'] for item in similarity_map[:item_count]]
return self._common.personids_to_urllist(personids)
def _parse_person_module(self, response, _):
""" Fill in result_dict. """
result_dict = {k: None for k in self._attrs}
#str attrs
result_dict['name'] = response['name']
result_dict['imdbid'] = response['imdbid']
result_dict['deathday'] = response['gestorben']
#list attrs
result_dict['alternative_names'] = [response['alternativ']]
if response['bild']:
result_dict['photo'] = [(None, response['bild'])]
return result_dict
@property
def supported_attrs(self):
return self._attrs