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.')