fix: date error

This commit is contained in:
Benexl
2025-07-22 01:23:49 +03:00
parent 9163b1394d
commit db1006a6b2

View File

@@ -20,6 +20,7 @@ from .types import (
AnilistBaseMediaDataSchema,
AnilistCurrentlyLoggedInUser,
AnilistDataSchema,
AnilistDateObject,
AnilistImage,
AnilistMediaList,
AnilistMediaLists,
@@ -48,6 +49,18 @@ status_map = {
}
def _to_generic_date(date: AnilistDateObject) -> Optional[datetime]:
return (
datetime(
date["year"],
date["month"],
date["day"],
)
if date and date["year"] and date["month"] and date["day"]
else None
)
def _to_generic_media_title(anilist_title: AnilistMediaTitle) -> MediaTitle:
"""Maps an AniList title object to a generic MediaTitle."""
romaji = anilist_title.get("romaji")
@@ -156,7 +169,7 @@ def _to_generic_user_status(
return
return UserListStatus(
id=anilist_media["mediaListEntry"]["id"],
status=anilist_media["mediaListEntry"]["status"],
status=anilist_media["mediaListEntry"]["status"], # type: ignore
progress=anilist_media["mediaListEntry"]["progress"],
)
@@ -186,16 +199,8 @@ def _to_generic_media_item(
popularity=data.get("popularity"),
favourites=data.get("favourites"),
next_airing=_to_generic_airing_schedule(data.get("nextAiringEpisode")),
start_date=datetime(
data["startDate"]["year"],
data["startDate"]["month"],
data["startDate"]["day"],
),
end_date=datetime(
data["startDate"]["year"],
data["startDate"]["month"],
data["startDate"]["day"],
),
start_date=_to_generic_date(data["startDate"]),
end_date=_to_generic_date(data["endDate"]),
streaming_episodes=_to_generic_streaming_episodes(
data.get("streamingEpisodes", [])
),