def do_build_genres(input_directory, genre_path, is_series_dir, act_immediately, config, debug):
if not os.path.exists(input_directory):
print "Directory '%s' does not exist. Exiting." % input_directory,
return 1
try:
thetvdb = TheTvDb(config, debug)
moviedb = MovieDB(config, debug)
database = Database(config, debug)
file_manager = FileManager(config, database, thetvdb, moviedb, debug)
files = []
for root, dir, files_to_add in os.walk(input_directory):
for file in files_to_add:
if file_manager.is_media_file(file):
files.append((root, file))
for (root, file) in files:
if is_series_dir:
build_genres_for_episode_file(file_manager, genre_path, root, file, act_immediately)
else:
build_genres_for_movie_file(file_manager, genre_path, root, file, act_immediately)
return 0
except:
traceback.print_exc()
return 11
def do_process(config, debug, move):
thetvdb = TheTvDb(config, debug)
database = Database(config, debug)
file_manager = FileManager(config, database, thetvdb, debug)
lock = file_manager.get_process_lock()
if lock is None:
print "Another instance of media_library_manager is currently processing. Exiting."
return 3
else:
try:
input_path = config.getLibraryInputPath()
tv_path = config.getLibraryTvPath()
if not os.path.exists(input_path):
print "Input path '%s' does not exist. Exiting." % input_path,
return 1
if not os.path.exists(tv_path):
print "Library TV path '%s' does not exist. Exiting." % tv_path,
return 2
files = []
for root, dir, files_to_add in os.walk(input_path):
for file in files_to_add:
if file_manager.is_media_file(file):
files.append((root, file))
input_contents = os.listdir(input_path)
for (root, file) in files:
file_path = os.path.join(root, file)
if os.path.isfile(file_path):
process_file(file_manager, file_path, tv_path, debug, move)
file_manager.cleanup_recent_folder()
except:
traceback.print_exc()
return 11
finally:
file_manager.relinquish_process_lock()
def do_generate(input_directory, config, debug):
if not os.path.exists(input_directory):
print "Directory '%s' does not exist. Exiting." % input_directory,
return 1
try:
thetvdb = TheTvDb(config, debug)
database = Database(config, debug)
file_manager = FileManager(config, database, thetvdb, debug)
files = []
for root, dir, files_to_add in os.walk(input_directory):
for file in files_to_add:
if file_manager.is_media_file(file):
files.append((root, file))
for (root, file) in files:
generate_for_file(root, file)
return 0
except:
traceback.print_exc()
return 11
