mirror/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr synced 2025-03-01 09:05:57 +00:00
Code Issues Releases Wiki Activity

WIP

Browse source
This commit is contained in:
morpheus65535 2021-09-20 23:49:21 -04:00
parent b4d75bad67
commit 9d5f9b2a9d
1 changed files with 57 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
bazarr/indexer/file_watcher.py
View file

@ -10,15 +10,19 @@ from watchdog.utils import WatchdogShutdown
from config import settings from config import settings
from bazarr.database import TableShowsRootfolder, TableMoviesRootfolder, TableShows, TableEpisodes, TableMovies from bazarr.database import TableShowsRootfolder, TableMoviesRootfolder, TableShows, TableEpisodes, TableMovies
from .video_prop_reader import VIDEO_EXTENSION from .video_prop_reader import VIDEO_EXTENSION
from .series.local.series_indexer import get_series_match, get_series_metadata
from .series.local.episodes_indexer import get_episode_metadata from .series.local.episodes_indexer import get_episode_metadata
from list_subtitles import store_subtitles from list_subtitles import store_subtitles
class FileWatcher: class FileWatcher:
def __init__(self): def __init__(self):
# make sure to only check for file with video extension
self.patterns = [f"*{x}" for x in VIDEO_EXTENSION] self.patterns = [f"*{x}" for x in VIDEO_EXTENSION]
self.ignore_patterns = None self.ignore_patterns = None
# ignore directory as it doesn't seems to work anyway... not with polling observer
self.ignore_directories = True self.ignore_directories = True
# enable case-sensitive monitoring to catch rename on other OS than Windows
self.case_sensitive = True self.case_sensitive = True
self.fs_event_handler = PatternMatchingEventHandler(self.patterns, self.fs_event_handler = PatternMatchingEventHandler(self.patterns,
self.ignore_patterns, self.ignore_patterns,
@ -33,72 +37,107 @@ class FileWatcher:
else: else:
logging.info(f'BAZARR file watcher is using the configured interval of {self.timeout} seconds.') logging.info(f'BAZARR file watcher is using the configured interval of {self.timeout} seconds.')
self.series_observer = Observer(timeout=self.timeout) # start
self.movies_observer = Observer(timeout=self.timeout) if settings.general.getboolean('use_series'):
self.series_directories = None self.series_observer = Observer(timeout=self.timeout)
self.movies_directories = None self.series_directories = None
if settings.general.getboolean('use_movies'):
self.movies_observer = Observer(timeout=self.timeout)
self.movies_directories = None
# get all root folders path
self.all_paths = list(TableShowsRootfolder.select()) self.all_paths = list(TableShowsRootfolder.select())
self.all_paths.extend(list(TableMoviesRootfolder.select())) self.all_paths.extend(list(TableMoviesRootfolder.select()))
def on_any_event(self, event): def on_any_event(self, event):
# get the root folder containing the file
root_dir = self.find_root_dir(event.src_path) root_dir = self.find_root_dir(event.src_path)
if root_dir: if root_dir:
if isinstance(root_dir, TableShowsRootfolder): if isinstance(root_dir, TableShowsRootfolder):
if event.event_type == 'deleted': if event.event_type == 'deleted':
# remove deleted episode
TableEpisodes.delete().where(TableEpisodes.path == event.src_path).execute() TableEpisodes.delete().where(TableEpisodes.path == event.src_path).execute()
elif event.event_type in ['created', 'modified']: elif event.event_type in ['created', 'modified']:
# add or update existing episode
series_metadata = self.get_series_from_episode_path(event.src_path) series_metadata = self.get_series_from_episode_path(event.src_path)
if not series_metadata:
# we can't find a series for this episode in database
if event.event_type == 'created':
# adding the first episode of a new series
series_dir = os.path.basename(os.path.dirname(os.path.dirname(event.src_path)))
# get matches from tmdb using the series directory name
series_matches = get_series_match(series_dir)
if series_matches:
# get series metadata for the first match
directory_metadata = get_series_metadata(series_matches[0]['tmdbId'], root_dir.rootId,
series_dir)
if directory_metadata:
try:
# insert the series in database
TableShows.insert(directory_metadata).execute()
except Exception as e:
logging.error(f'BAZARR is unable to insert this series to the database: '
f'"{directory_metadata["path"]}". The exception encountered is '
f'"{e}".')
if series_metadata: if series_metadata:
# we found an existing series in database and will get the episode metadata from tmdb
episode_metadata = get_episode_metadata(event.src_path, series_metadata['tmdbId'], episode_metadata = get_episode_metadata(event.src_path, series_metadata['tmdbId'],
series_metadata['seriesId']) series_metadata['seriesId'])
if episode_metadata: if episode_metadata:
TableEpisodes.insert(episode_metadata).on_conflict('update').execute() if event.event_type == 'created':
# insert the new episode
TableEpisodes.insert(episode_metadata).execute()
else:
# or update the existing one
TableEpisodes.update(episode_metadata) \
.where(TableEpisodes.path == event.src_path) \
.execute()
# store the embedded and external subtitles for that episode in database
store_subtitles(event.src_path, use_cache=False) store_subtitles(event.src_path, use_cache=False)
elif event.event_type in 'moved': elif event.event_type in 'moved':
# get the series metadata using the episode source path
series_metadata = self.get_series_from_episode_path(event.dest_path) series_metadata = self.get_series_from_episode_path(event.dest_path)
if series_metadata: if series_metadata:
# get episode metadata using the destination path
episode_metadata = get_episode_metadata(event.dest_path, series_metadata['tmdbId'], episode_metadata = get_episode_metadata(event.dest_path, series_metadata['tmdbId'],
series_metadata['seriesId']) series_metadata['seriesId'])
if episode_metadata: if episode_metadata:
# update the episode in database and store subtitles in database
TableEpisodes.update(episode_metadata).where(TableEpisodes.path == event.src_path).execute() TableEpisodes.update(episode_metadata).where(TableEpisodes.path == event.src_path).execute()
store_subtitles(event.src_path, use_cache=False) store_subtitles(event.src_path, use_cache=False)
else: else:
# here we'll deal with movies
pass pass
def find_root_dir(self, path): def find_root_dir(self, path):
# return the parent root folder for that episode/movie path
root_dir_list = [x for x in self.all_paths if x.path in path] root_dir_list = [x for x in self.all_paths if x.path in path]
if root_dir_list: if root_dir_list:
return root_dir_list[0] return root_dir_list[0]
@staticmethod @staticmethod
def get_series_from_episode_path(path): def get_series_from_episode_path(path):
series_dir = os.path.dirname(path) # return series row for a specific episode path by moving up 2 level. ie: Series/Season/episode.mkv
series_dir = os.path.dirname(os.path.dirname(path))
try: try:
series_metadata = TableShows.select().where(TableShows.path == series_dir).dicts().get() series_metadata = TableShows.select().where(TableShows.path == series_dir).dicts().get()
except TableShows.DoesNotExist: except TableShows.DoesNotExist:
series_dir = os.path.dirname(os.path.dirname(path)) return None
try:
series_metadata = TableShows.select().where(TableShows.path == series_dir).dicts().get()
except TableShows.DoesNotExist:
return None
else:
return series_metadata
else: else:
return series_metadata return series_metadata
def config(self): def config(self):
# config the event handler and observers
self.fs_event_handler.on_any_event = self.on_any_event self.fs_event_handler.on_any_event = self.on_any_event
# get all root folder paths and add them to the observers
if settings.general.getboolean('use_series'): if settings.general.getboolean('use_series'):
self.series_directories = [x['path'] for x in TableShowsRootfolder.select().dicts()] self.series_directories = [x['path'] for x in TableShowsRootfolder.select().dicts()]
if settings.general.getboolean('use_movies'):
self.movies_directories = [x['path'] for x in TableMoviesRootfolder.select().dicts()]
if settings.general.getboolean('use_series'):
for series_directory in self.series_directories: for series_directory in self.series_directories:
self.series_observer.schedule(self.fs_event_handler, series_directory, recursive=True) self.series_observer.schedule(self.fs_event_handler, series_directory, recursive=True)
if settings.general.getboolean('use_movies'): if settings.general.getboolean('use_movies'):
self.movies_directories = [x['path'] for x in TableMoviesRootfolder.select().dicts()]
for movies_directory in self.movies_directories: for movies_directory in self.movies_directories:
self.movies_observer.schedule(self.fs_event_handler, movies_directory, recursive=True) self.movies_observer.schedule(self.fs_event_handler, movies_directory, recursive=True)