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 }
import sys from hugin.harvest.session import Session USAGE = """Usage: python dummydownloader <foldertosave> <filewithimdbids> <provider> """ if __name__ == '__main__': if len(sys.argv) < 4: print(USAGE) sys.exit(1) s = Session(parallel_jobs=4, parallel_downloads_per_job=2) 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(
def __init__(self): self._session = Session()