def _get_playing_show(self, filename): if not self.active: # Don't do anything if the Tracker is disabled return (utils.TRACKER_NOVIDEO, None) if filename: if filename == self.last_filename: # It's the exact same filename, there's no need to do the processing again return (self.last_state, self.last_show_tuple) self.last_filename = filename # Do a regex to the filename to get # the show title and episode number aie = AnimeInfoExtractor(filename) (show_title, show_ep) = (aie.getName(), aie.getEpisode()) if not show_title: return (utils.TRACKER_UNRECOGNIZED, None ) # Format not recognized playing_show = utils.guess_show(show_title, self.list) if playing_show: return (utils.TRACKER_PLAYING, (playing_show, show_ep)) else: # Show not in list if self.not_found_prompt: # Dummy show to search for show = {'id': 0, 'title': show_title} return (utils.TRACKER_NOT_FOUND, (show, show_ep)) else: return (utils.TRACKER_NOT_FOUND, None) else: self.last_filename = None return (utils.TRACKER_NOVIDEO, None) # Not playing
def _get_playing_show(self, filename): if not self.active: # Don't do anything if the Tracker is disabled return (utils.TRACKER_NOVIDEO, None) if filename: if filename == self.last_filename: # It's the exact same filename, there's no need to do the processing again return (self.last_state, self.last_show_tuple) self.last_filename = filename # Do a regex to the filename to get # the show title and episode number aie = AnimeInfoExtractor(filename) (show_title, show_ep) = (aie.getName(), aie.getEpisode()) if not show_title: return (utils.TRACKER_UNRECOGNIZED, None) # Format not recognized playing_show = utils.guess_show(show_title, self.list) if playing_show: return (utils.TRACKER_PLAYING, (playing_show, show_ep)) else: # Show not in list if self.not_found_prompt: # Dummy show to search for show = {'id': 0, 'title': show_title} return (utils.TRACKER_NOT_FOUND, (show, show_ep)) else: return (utils.TRACKER_NOT_FOUND, None) else: self.last_filename = None return (utils.TRACKER_NOVIDEO, None) # Not playing
def get_show_info(self, showid=None, title=None, filename=None): """ Returns the show dictionary for the specified **showid**. """ showdict = self.data_handler.get() if showid: # Get show by ID try: return showdict[showid] except KeyError: raise utils.EngineError("Show not found.") elif title: showdict = self.data_handler.get() # Get show by title, slower for k, show in showdict.items(): if show['title'] == title: return show raise utils.EngineError("Show not found.") elif filename: # Guess show by filename self.msg.debug(self.name, "Guessing by filename.") aie = AnimeInfoExtractor(filename) (show_title, ep) = aie.getName(), aie.getEpisode() self.msg.debug(self.name, "Guessed {}".format(show_title)) if show_title: show = utils.guess_show(show_title, self._get_tracker_list()) if show: return (show, ep) else: raise utils.EngineError("Show not found.") else: raise utils.EngineError("File name not recognized.")
def scan_library(self, my_status=None, ignorecache=False): # Check if operation is supported by the API if not self.mediainfo.get('can_play'): raise utils.EngineError('Operation not supported by current site or mediatype.') if not self.config['searchdir']: raise utils.EngineError('Media directory is not set.') if not utils.dir_exists(self.config['searchdir']): raise utils.EngineError('The set media directory doesn\'t exist.') t = time.time() library = {} library_cache = self.data_handler.library_cache_get() if not my_status: my_status = self.mediainfo['status_start'] self.msg.info(self.name, "Scanning local library...") tracker_list = self._get_tracker_list(my_status) # Do a full listing of the media directory for fullpath, filename in utils.regex_find_videos('mkv|mp4|avi', self.config['searchdir']): show_id = None if not ignorecache and filename in library_cache.keys(): # If the filename was already seen before # use the cached information, if there's no information (None) # then it means it doesn't correspond to any show in the list # and can be safely skipped. if library_cache[filename]: show_id = library_cache[filename][0] show_ep = library_cache[filename][1] else: continue else: # If the filename has not been seen, extract # the information from the filename and do a fuzzy search # on the user's list. Cache the information. # If it fails, cache it as None. aie = AnimeInfoExtractor(filename) (show_title, show_ep) = (aie.getName(), aie.getEpisode()) if show_title: show = utils.guess_show(show_title, tracker_list) if show: show_id = show['id'] library_cache[filename] = (show['id'], show_ep) else: library_cache[filename] = None else: library_cache[filename] = None # After we got our information, add it to our library if show_id: if show_id not in library.keys(): library[show_id] = {} library[show_id][show_ep] = fullpath self.msg.debug(self.name, "Time: %s" % (time.time() - t)) self.data_handler.library_save(library) self.data_handler.library_cache_save(library_cache) return library
def _add_show_to_library(self, library, library_cache, rescan, fullpath, filename, tracker_list): show_id = None if not rescan and filename in library_cache: # If the filename was already seen before # use the cached information, if there's no information (None) # then it means it doesn't correspond to any show in the list # and can be safely skipped. if library_cache[filename]: (show_id, show_ep) = library_cache[filename] if type(show_ep) is tuple: (show_ep_start, show_ep_end) = show_ep else: show_ep_start = show_ep_end = show_ep self.msg.debug(self.name, "File in cache: {}".format(fullpath)) else: self.msg.debug( self.name, "File in cache but skipped: {}".format(fullpath)) return library, library_cache else: # If the filename has not been seen, extract # the information from the filename and do a fuzzy search # on the user's list. Cache the information. # If it fails, cache it as None. aie = AnimeInfoExtractor(filename) show_title = aie.getName() (show_ep_start, show_ep_end) = aie.getEpisodeNumbers(True) if show_title: show = utils.guess_show(show_title, tracker_list) if show: self.msg.debug(self.name, "Adding to library: {}".format(fullpath)) show_id = show['id'] if show_ep_start == show_ep_end: library_cache[filename] = (show['id'], show_ep_start) else: library_cache[filename] = (show['id'], (show_ep_start, show_ep_end)) else: self.msg.debug(self.name, "Not a show, skipping: {}".format(fullpath)) library_cache[filename] = None else: self.msg.debug(self.name, "Not recognized, skipping: {}".format(fullpath)) library_cache[filename] = None # After we got our information, add it to our library if show_id: if show_id not in library: library[show_id] = {} for show_ep in range(show_ep_start, show_ep_end + 1): library[show_id][show_ep] = fullpath return library, library_cache
def _add_show_to_library(self, library, library_cache, rescan, fullpath, filename, tracker_list): show_id = None if not rescan and filename in library_cache: # If the filename was already seen before # use the cached information, if there's no information (None) # then it means it doesn't correspond to any show in the list # and can be safely skipped. if library_cache[filename]: (show_id, show_ep) = library_cache[filename] if type(show_ep) is tuple: (show_ep_start, show_ep_end) = show_ep else: show_ep_start = show_ep_end = show_ep self.msg.debug(self.name, "File in cache: {}".format(fullpath)) else: self.msg.debug(self.name, "File in cache but skipped: {}".format(fullpath)) return library, library_cache else: # If the filename has not been seen, extract # the information from the filename and do a fuzzy search # on the user's list. Cache the information. # If it fails, cache it as None. aie = AnimeInfoExtractor(filename) show_title = aie.getName() (show_ep_start, show_ep_end) = aie.getEpisodeNumbers(True) if show_title: show = utils.guess_show(show_title, tracker_list) if show: self.msg.debug(self.name, "Adding to library: {}".format(fullpath)) if show_ep_start == show_ep_end: # TODO : Support redirections for episode ranges (show, show_ep) = utils.redirect_show((show, show_ep_start), self.redirections, tracker_list) show_ep_end = show_ep_start = show_ep self.msg.debug(self.name, "Redirected to {} {}".format(show['title'], show_ep)) library_cache[filename] = (show['id'], show_ep) else: library_cache[filename] = (show['id'], (show_ep_start, show_ep_end)) show_id = show['id'] else: self.msg.debug(self.name, "Not a show, skipping: {}".format(fullpath)) library_cache[filename] = None else: self.msg.debug(self.name, "Not recognized, skipping: {}".format(fullpath)) library_cache[filename] = None # After we got our information, add it to our library if show_id: if show_id not in library: library[show_id] = {} for show_ep in range(show_ep_start, show_ep_end+1): library[show_id][show_ep] = fullpath return library, library_cache
def _get_playing_show(self, filename): if not self.active: # Don't do anything if the Tracker is disabled return (utils.TRACKER_NOVIDEO, None) if filename: self.msg.debug(self.name, "Guessing filename: {}".format(filename)) # Trim out watch dir if os.path.isabs(filename): for watch_prefix in self.watch_dirs: if filename.startswith(watch_prefix): filename = filename[len(watch_prefix):] if filename.startswith(os.path.sep): filename = filename[len(os.path.sep):] break if filename == self.last_filename: # It's the exact same filename, there's no need to do the processing again self.msg.debug(self.name, "Same filename as before. Skipping.") return (self.last_state, self.last_show_tuple) self.last_filename = filename # Do a regex to the filename to get # the show title and episode number aie = AnimeInfoExtractor(filename) (show_title, show_ep) = (aie.getName(), aie.getEpisode()) if not show_title: # Format not recognized return (utils.TRACKER_UNRECOGNIZED, None) playing_show = utils.guess_show(show_title, self.list) self.msg.debug( self.name, "Show guess: {}: {}".format(show_title, playing_show)) if playing_show: (playing_show, show_ep) = utils.redirect_show( (playing_show, show_ep), self.redirections, self.list) return (utils.TRACKER_PLAYING, (playing_show, show_ep)) else: # Show not in list if self.config['tracker_not_found_prompt']: # Dummy show to search for show = {'id': 0, 'title': show_title} return (utils.TRACKER_NOT_FOUND, (show, show_ep)) else: return (utils.TRACKER_NOT_FOUND, None) else: self.last_filename = None return (utils.TRACKER_NOVIDEO, None) # Not playing
def _search_video(self, titles, episode): # DEPRECATED !!! self.msg.debug(self.name, "DEPRECATED: _search_video") best_candidate = (None, 0, None) matcher = difflib.SequenceMatcher() # Check over video files and propose our best candidate for (fullpath, filename) in utils.regex_find_videos('mkv|mp4|avi', self.config['searchdir']): # Analyze what's the title and episode of the file aie = AnimeInfoExtractor(filename) candidate_title = aie.getName() candidate_episode_start, candidate_episode_end = aie.getEpisodeNumbers( ) # Skip this file if we couldn't analyze it if not candidate_title: continue if candidate_episode_start is None: continue # Skip this file if it isn't the episode we want if candidate_episode_end is None: if episode != candidate_episode_start: continue else: if not (candidate_episode_start <= episode <= candidate_episode_end): continue matcher.set_seq1(candidate_title.lower()) # We remember to compare all titles (aliases and whatnot) for requested_title in titles: matcher.set_seq2(requested_title.lower()) ratio = matcher.ratio() # Propose as our new candidate if its ratio is # better than threshold and it's better than # what we've seen yet if ratio > 0.7 and ratio > best_candidate[1]: best_candidate = (fullpath, ratio, aie.getEpisode()) return best_candidate[0], best_candidate[2]
def _get_playing_show(self, filename): if not self.active: # Don't do anything if the Tracker is disabled return (utils.TRACKER_NOVIDEO, None) if filename: self.msg.debug(self.name, "Guessing filename: {}".format(filename)) # Trim out watch dir if os.path.isabs(filename): for watch_prefix in self.watch_dirs: if filename.startswith(watch_prefix): filename = filename[len(watch_prefix):] if filename.startswith(os.path.sep): filename = filename[len(os.path.sep):] break if filename == self.last_filename: # It's the exact same filename, there's no need to do the processing again return (self.last_state, self.last_show_tuple) self.last_filename = filename # Do a regex to the filename to get # the show title and episode number aie = AnimeInfoExtractor(filename) (show_title, show_ep) = (aie.getName(), aie.getEpisode()) if not show_title: return (utils.TRACKER_UNRECOGNIZED, None) # Format not recognized playing_show = utils.guess_show(show_title, self.list) if playing_show: (playing_show, show_ep) = utils.redirect_show((playing_show, show_ep), self.redirections, self.list) return (utils.TRACKER_PLAYING, (playing_show, show_ep)) else: # Show not in list if self.config['tracker_not_found_prompt']: # Dummy show to search for show = {'id': 0, 'title': show_title} return (utils.TRACKER_NOT_FOUND, (show, show_ep)) else: return (utils.TRACKER_NOT_FOUND, None) else: self.last_filename = None return (utils.TRACKER_NOVIDEO, None) # Not playing
def _search_video(self, titles, episode): # DEPRECATED !!! self.msg.debug(self.name, "DEPRECATED: _search_video") best_candidate = (None, 0, None) matcher = difflib.SequenceMatcher() # Check over video files and propose our best candidate for (fullpath, filename) in utils.regex_find_videos('mkv|mp4|avi', self.config['searchdir']): # Analyze what's the title and episode of the file aie = AnimeInfoExtractor(filename) candidate_title = aie.getName() candidate_episode_start, candidate_episode_end = aie.getEpisodeNumbers() # Skip this file if we couldn't analyze it if not candidate_title: continue if candidate_episode_start is None: continue # Skip this file if it isn't the episode we want if candidate_episode_end is None: if episode != candidate_episode_start: continue else: if not (candidate_episode_start <= episode <= candidate_episode_end): continue matcher.set_seq1(candidate_title.lower()) # We remember to compare all titles (aliases and whatnot) for requested_title in titles: matcher.set_seq2(requested_title.lower()) ratio = matcher.ratio() # Propose as our new candidate if its ratio is # better than threshold and it's better than # what we've seen yet if ratio > 0.7 and ratio > best_candidate[1]: best_candidate = (fullpath, ratio, aie.getEpisode()) return best_candidate[0], best_candidate[2]
def get_show_info(self, showid=None, title=None, filename=None): """ Returns the show dictionary for the specified **showid**. """ showdict = self.data_handler.get() if showid: # Get show by ID try: return showdict[showid] except KeyError: raise utils.EngineError("Show not found.") elif title: showdict = self.data_handler.get() # Get show by title, slower for show in showdict.values(): if show['title'] == title: return show raise utils.EngineError("Show not found.") elif filename: # Guess show by filename self.msg.debug(self.name, "Guessing by filename.") aie = AnimeInfoExtractor(filename) (show_title, ep) = aie.getName(), aie.getEpisode() self.msg.debug(self.name, "Guessed {}".format(show_title)) if show_title: tracker_list = self._get_tracker_list() show = utils.guess_show(show_title, tracker_list) if show: return utils.redirect_show((show, ep), self.redirections, tracker_list) else: raise utils.EngineError("Show not found.") else: raise utils.EngineError("File name not recognized.")