mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-11-24 04:21:08 +00:00
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
|
"""
|
||
|
|
Define and check rules for a film object.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# Rules for the film object
|
||
|
|
parse_options = {
|
||
|
|
'type': 'film',
|
||
|
|
'runtime_min': 80,
|
||
|
|
'runtime_max': 140,
|
||
|
|
'inlude_unkown_runtime': False,
|
||
|
|
'score_range_min': '6.9',
|
||
|
|
'score_range_max': '10.0',
|
||
|
|
'include_unknown_score': False,
|
||
|
|
'year_range_oldest': 1990,
|
||
|
|
'year_range_newest': 2019,
|
||
|
|
'wanted_genres': ['drama'],
|
||
|
|
'unwanted_genres': ['romance', 'musical','horror', 'documentary'],
|
||
|
|
# Whether add or remove a film
|
||
|
|
# whose genre neither in wanted_genres nor unwanted_genres list
|
||
|
|
'add_not_unwanted_&_not_wanted': True,
|
||
|
|
'include_watched': False
|
||
|
|
}
|
||
|
|
|
||
|
|
def check_runtime(film_runtime):
|
||
|
|
if film_runtime == 'unknown':
|
||
|
|
return parse_options['inlude_unkown_runtime']
|
||
|
|
min_runtime = parse_options['runtime_min']
|
||
|
|
max_runtime = parse_options['runtime_max']
|
||
|
|
|
||
|
|
return film_runtime >= min_runtime and film_runtime <= max_runtime
|
||
|
|
|
||
|
|
def check_genre(film_genre_list):
|
||
|
|
for genre in film_genre_list:
|
||
|
|
if genre.lower() in parse_options['unwanted_genres']:
|
||
|
|
return False
|
||
|
|
if parse_options['wanted_genres'] is None or len(parse_options['wanted_genres']) == 0:
|
||
|
|
return True
|
||
|
|
for genre in film_genre_list:
|
||
|
|
if genre.lower() in parse_options['wanted_genres']:
|
||
|
|
return True
|
||
|
|
return parse_options['add_not_unwanted_&_not_wanted']
|
||
|
|
|
||
|
|
|
||
|
|
def check_score(score_range):
|
||
|
|
if score_range == 'unknown':
|
||
|
|
return parse_options['include_unknown_score']
|
||
|
|
min_score = float(parse_options['score_range_min']) * 10
|
||
|
|
max_score = float(parse_options['score_range_max']) * 10
|
||
|
|
return score_range >= min_score and score_range <= max_score
|
||
|
|
|
||
|
|
|
||
|
|
def check_year(year_range):
|
||
|
|
min_year = parse_options['year_range_oldest']
|
||
|
|
max_year = parse_options['year_range_newest']
|
||
|
|
return int(year_range) >= min_year and int(year_range) <= max_year
|
||
|
|
|
||
|
|
|
||
|
|
def check_type(film_type):
|
||
|
|
if parse_options['type'] == 'both':
|
||
|
|
return True
|
||
|
|
elif parse_options['type'] == film_type:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def watched_included():
|
||
|
|
return parse_options['include_watched']
|
||
|
|
|
||
|
|
def check_film_object(film_object, watched_films=None):
|
||
|
|
if not check_runtime(film_object.runtime):
|
||
|
|
return False
|
||
|
|
if not check_genre(film_object.genres):
|
||
|
|
return False
|
||
|
|
if not check_score(film_object.rating):
|
||
|
|
return False
|
||
|
|
if film_object.type == 'film' and not check_year(film_object.year):
|
||
|
|
return False
|
||
|
|
if not check_type(film_object.type):
|
||
|
|
return False
|
||
|
|
if watched_films is not None and film_object.name in watched_films:
|
||
|
|
return False
|
||
|
|
# All of the above rules applied for the object
|
||
|
|
return True
|