32 lines
882 B
Python
32 lines
882 B
Python
import json
|
|
import yaml
|
|
import re
|
|
|
|
from constants import SERIES_NAME_YEAR_PATTERN
|
|
|
|
|
|
def get_credentials():
|
|
with open('credentials.yaml') as file:
|
|
sdarot_credentials = yaml.full_load(file)
|
|
return sdarot_credentials.get('username'), sdarot_credentials.get('password')
|
|
|
|
|
|
def save_cookies(cookies):
|
|
with open('cookie.json') as cookie_file:
|
|
json.dump(cookies, cookie_file)
|
|
|
|
|
|
def save_series_dictionary(series_json):
|
|
with open('series.json', 'w') as file:
|
|
json.dump(series_json, file)
|
|
|
|
|
|
def get_matched_series_ids(series_dictionary, series_name):
|
|
year = ''
|
|
match = re.match(SERIES_NAME_YEAR_PATTERN, series_name)
|
|
if match:
|
|
series_name = match.group(1)
|
|
year = match.group(2)
|
|
return list(filter(lambda obj: (series_name in obj['eng'].lower() or series_name in obj['heb']) and year in obj['year'], series_dictionary))
|
|
|