class PlotLangChange(plugin.IModifier):
def __init__(self):
self._session = Session()
def modify(self, movie, attr_name='plot', change_to='en'):
query = self._session.create_query(
title=movie.attributes.get('title'),
providers=['tmdbmovie'],
amount=1,
language=change_to
)
result = self._session.submit(query)
if result:
movie.attributes[attr_name] = result.pop()._result_dict.get('plot')
def modify_all(self, db):
for movie in db.values():
self.movie(movie)
def parameters(self):
return {
'attr_name': str,
'change_to': str
}
if os.path.exists(sys.argv[1]):
print('{}, already exists.'.format(sys.argv[1]))
else:
print('Creating {}.'.format(sys.argv[1]))
os.mkdir(sys.argv[1])
with open(sys.argv[2], 'r') as f:
movieids = f.read().splitlines()
length = len(movieids)
cnt = 0
for movieid in movieids:
movie = s.submit(s.create_query(
imdbid=movieid, lang='de', providers=[sys.argv[3]], cache=True)
)
if movie == []:
print(movieid, ' not found.')
continue
movie = movie.pop()
title = movie.result_dict['title']
if '/' in title:
title = title.replace('/', '|')
moviefolder = '{0} ({1})'.format(title, movie.result_dict['year'])
moviepath = '{}/{}'.format(sys.argv[1], moviefolder)
if not os.path.exists(moviepath):
os.mkdir(moviepath)
else:
print(moviepath, movieid, ' exists.')
