fixed year filter and test error

This commit is contained in:
meisnate12 2021-08-06 13:46:13 -04:00
parent ce911b78ac
commit 6920e39d32
2 changed files with 34 additions and 26 deletions

View file

@ -1061,30 +1061,38 @@ class CollectionBuilder:
elif "trakt" in method: check_map(self.config.Trakt.get_items(method, value, self.library.is_movie)) elif "trakt" in method: check_map(self.config.Trakt.get_items(method, value, self.library.is_movie))
else: logger.error(f"Collection Error: {method} method not supported") else: logger.error(f"Collection Error: {method} method not supported")
def tmdb_filter(self, item_id, is_movie, item=None): def check_tmdb_filter(self, item_id, is_movie, item=None):
filter_missing = False
if self.tmdb_filters or self.details["released_missing_only"]: if self.tmdb_filters or self.details["released_missing_only"]:
try: try:
if item is None: if item is None:
item = self.config.TMDb.get_movie(item_id) if is_movie else self.config.TMDb.get_movie(self.config.Convert.tvdb_to_tmdb(item_id)) item = self.config.TMDb.get_movie(item_id) if is_movie else self.config.TMDb.get_show(self.config.Convert.tvdb_to_tmdb(item_id))
if self.details["released_missing_only"]: if self.details["released_missing_only"]:
if util.validate_date(item.release_date if is_movie else item.first_air_date, "") > self.current_time: if util.validate_date(item.release_date if is_movie else item.first_air_date, "") > self.current_time:
return True return False
for filter_method, filter_data in self.tmdb_filters: for filter_method, filter_data in self.tmdb_filters:
if (filter_method == "original_language" and item.original_language not in filter_data) \ filter_attr, modifier, filter_final = self._split(filter_method)
or (filter_method == "original_language.not" and item.original_language in filter_data) \ if filter_attr == "original_language":
or (filter_method == "tmdb_vote_count.gt" and item.vote_count <= filter_data) \ if (modifier == ".not" and item.original_language in filter_data) \
or (filter_method == "tmdb_vote_count.gte" and item.vote_count < filter_data) \ or (modifier == "" and item.original_language not in filter_data):
or (filter_method == "tmdb_vote_count.lt" and item.vote_count >= filter_data) \ return False
or (filter_method == "tmdb_vote_count.lte" and item.vote_count > filter_data) \ elif modifier in [".gt", ".gte", ".lt", ".lte"]:
or (filter_method == "year.gt" and item.year <= filter_data) \ attr = None
or (filter_method == "year.gte" and item.year < filter_data) \ if filter_attr == "tmdb_vote_count":
or (filter_method == "year.lt" and item.year >= filter_data) \ attr = item.vote_count
or (filter_method == "year.lte" and item.year > filter_data): elif filter_attr == "year" and is_movie:
return True attr = item.year
elif filter_attr == "year" and not is_movie:
air_date = item.first_air_date
if air_date:
attr = util.validate_date(air_date, "Year Filter").year
if attr is None or (modifier == ".gt" and attr <= filter_data) \
or (modifier == ".gte" and attr < filter_data) \
or (modifier == ".lt" and attr >= filter_data) \
or (modifier == ".lte" and attr > filter_data):
return False
except Failed: except Failed:
return True return False
return filter_missing return True
def build_filter(self, method, plex_filter, smart=False): def build_filter(self, method, plex_filter, smart=False):
if smart: if smart:
@ -1529,13 +1537,13 @@ class CollectionBuilder:
logger.error(e) logger.error(e)
continue continue
current_title = f"{movie.title} ({util.validate_date(movie.release_date, 'test').year})" if movie.release_date else movie.title current_title = f"{movie.title} ({util.validate_date(movie.release_date, 'test').year})" if movie.release_date else movie.title
if self.tmdb_filter(missing_id, True, item=movie): if self.check_tmdb_filter(missing_id, True, item=movie):
if self.details["show_filtered"] is True and self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | X | {current_title} (TMDb: {missing_id})")
else:
missing_movies_with_names.append((current_title, missing_id)) missing_movies_with_names.append((current_title, missing_id))
if self.details["show_missing"] is True: if self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | ? | {current_title} (TMDb: {missing_id})") logger.info(f"{self.name} Collection | ? | {current_title} (TMDb: {missing_id})")
else:
if self.details["show_filtered"] is True and self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | X | {current_title} (TMDb: {missing_id})")
logger.info("") logger.info("")
logger.info(f"{len(missing_movies_with_names)} Movie{'s' if len(missing_movies_with_names) > 1 else ''} Missing") logger.info(f"{len(missing_movies_with_names)} Movie{'s' if len(missing_movies_with_names) > 1 else ''} Missing")
if self.details["save_missing"] is True: if self.details["save_missing"] is True:
@ -1558,13 +1566,13 @@ class CollectionBuilder:
logger.error(e) logger.error(e)
continue continue
current_title = str(show.title.encode("ascii", "replace").decode()) current_title = str(show.title.encode("ascii", "replace").decode())
if self.tmdb_filter(missing_id, False): if self.check_tmdb_filter(missing_id, False):
if self.details["show_filtered"] is True and self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | X | {current_title} (TVDb: {missing_id})")
else:
missing_shows_with_names.append((current_title, missing_id)) missing_shows_with_names.append((current_title, missing_id))
if self.details["show_missing"] is True: if self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | ? | {current_title} (TVDB: {missing_id})") logger.info(f"{self.name} Collection | ? | {current_title} (TVDB: {missing_id})")
else:
if self.details["show_filtered"] is True and self.details["show_missing"] is True:
logger.info(f"{self.name} Collection | X | {current_title} (TVDb: {missing_id})")
logger.info("") logger.info("")
logger.info(f"{len(missing_shows_with_names)} Show{'s' if len(missing_shows_with_names) > 1 else ''} Missing") logger.info(f"{len(missing_shows_with_names)} Show{'s' if len(missing_shows_with_names) > 1 else ''} Missing")
if self.details["save_missing"] is True: if self.details["save_missing"] is True:

View file

@ -75,7 +75,7 @@ class Metadata:
logger.info("") logger.info("")
for mapping_name, meta in self.metadata.items(): for mapping_name, meta in self.metadata.items():
methods = {mm.lower(): mm for mm in meta} methods = {mm.lower(): mm for mm in meta}
if self.config.test and ("test" not in methods or meta[methods["test"]] is not True): if self.config.test_mode and ("test" not in methods or meta[methods["test"]] is not True):
continue continue
updated = False updated = False