class Scanner(object):
"""
Methods for scanning files into the database.
"""
def __init__(self, path):
self.logger = logging.getLogger('pymetadatamanager.scanner')
self.config = Config()
self.dbTV = TVShowDB(self.config.tvshowdb)
self.dbTV.init_db()
self.TVDB = TVDB()
self.FP = FileParser()
self.nfo_reader = NfoReader()
self.new_time = self.TVDB.get_server_time()
self.dbTV.clean_db_files()
self.series_list = []
self.set_file_list(path)
def __del__(self):
try:
self.dbTV.set_update_time(self.new_time)
except AttributeError:
self.logger.error("Error setting update time.")
def set_file_list(self, path):
self.file_list = self.FP.parse_files_by_path(path)
def set_series_list(self):
for file in self.file_list:
dir = file[0]
filename = file[1]
series_name = file[2]
if not series_name in self.series_list:
self.series_list.append(series_name)
def get_series_id_list(self, series_name):
series_id = self.dbTV.check_db_for_series(series_name)
if not series_id:
match_list = self.TVDB.find_series(series_name)
else:
match_list = [(series_id, match_list)]
return match_list
def add_series_to_db_by_id(self, series_id):
if not self.dbTV.check_db_for_series(series_id):
series = self.TVDB.get_all_series_info(series_id)
episodes = self.TVDB.get_series_episodes(series_id)
actors = self.TVDB.get_series_actors(series_id)
banners = self.TVDB.get_series_banners(series_id)
if series is not None:
self.dbTV.write_series_to_db(series)
series_name = self.dbTV.get_series_name(series_id)
self.logger.info("Adding series %s to DB" % (series_name,))
if episodes is not None:
self.dbTV.write_episodes_to_db(episodes, series_id)
if actors is not None:
self.dbTV.write_actors_to_db(actors)
if banners is not None:
self.dbTV.write_banners_to_db(banners)
def add_series_to_db_by_nfo(self, series_name):
episodes = []
episode_nfos = []
for file in self.file_list:
name = file[2]
if name == series_name:
episode_nfos.append(file[6])
series_nfo = file[5]
series = self.nfo_reader.get_series(series_nfo)
for episode_nfo in episode_nfos:
if not episode_nfo == '':
episodes.append(self.nfo_reader.get_episode(episode_nfo))
actors = self.nfo_reader.get_actors(series_nfo)
self.logger.debug(actors)
banners = self.nfo_reader.get_banners(series_nfo)
self.logger.info("Adding series %s to DB" % (series_name,))
if series is not None:
self.dbTV.write_series_to_db(series)
if episodes is not None:
series_id = self.dbTV.get_series_id(series_name)
self.dbTV.write_episodes_to_db(episodes, series_id)
if actors is not None:
self.dbTV.write_actors_to_db(actors)
if banners is not None:
self.dbTV.write_banners_to_db(banners)
def add_files_to_db(self, series_id, series_name):
#Create a list of files from this series
series_file_list = []
for file in self.file_list:
if file[2] == series_name:
if not self.dbTV.check_db_for_file(file[1], file[0]):
series_file_list.append(file)
#Add any new files to the DB
if len(series_file_list):
self.logger.info("Adding files from %s to DB" % (series_name,))
self.dbTV.write_files_to_db(series_file_list, series_id)
unlinked = self.dbTV.find_unlinked_files()
unlinked_list = []
for unlinked_file in unlinked:
for file in self.file_list:
file_path = file[0]
file_name = file[1]
if unlinked_file[1] == file_name \
and unlinked_file[2] == file_path:
unlinked_list.append(file)
if len(unlinked_list):
self.dbTV.write_files_to_db(series_file_list, series_id)
