Skip to content

Commit

Permalink
make sure we have a day in the dates
Browse files Browse the repository at this point in the history
  • Loading branch information
prochy-exe committed Apr 27, 2024
1 parent d06303c commit 6ceea01
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions malfetcher/mal_fetcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -414,6 +414,18 @@ def generate_upcoming_ep(release_date):
upcoming_ep = math.ceil(int((current_date - release_date).days)/7) + 1
return upcoming_ep

def ensure_day_in_date(date_str):
try:
datetime.strptime(date_str, '%Y-%m-%d')
return date_str
except ValueError:
try:
year_month = datetime.strptime(date_str, '%Y-%m')
full_date = year_month.replace(day=1)
return full_date.strftime('%Y-%m-%d')
except ValueError:
return None

anime_id = str(anime_info['id'])
anime_data = {}
anime_data['al_id'] = mal_to_al_id(anime_id)
Expand All @@ -426,8 +438,8 @@ def generate_upcoming_ep(release_date):
] + anime_info['alternative_titles']['synonyms']
anime_data['synonyms'] = [item for item in anime_data['synonyms'] if item is not None]
anime_data['status'] = mal_to_al_status[anime_info['status']]
anime_data['release_date'] = anime_info['start_date'] if 'start_date' in anime_info else None
anime_data['end_date'] = anime_info['end_date'] if 'end_date' in anime_info else None
anime_data['release_date'] = ensure_day_in_date(anime_info['start_date']) if 'start_date' in anime_info else None
anime_data['end_date'] = ensure_day_in_date(anime_info['end_date']) if 'end_date' in anime_info else None
anime_data['upcoming_ep'] = generate_upcoming_ep(anime_data['release_date']) if anime_data['status'] == "RELEASING" else None
anime_data['format'] = anime_info['media_type'].upper()
anime_data['related'] = getRelated()
Expand Down

0 comments on commit 6ceea01

Please sign in to comment.