diff --git a/sukuinote/slave-plugins/anilist.py b/sukuinote/slave-plugins/anilist.py
index a1e61c6..0aec1aa 100644
--- a/sukuinote/slave-plugins/anilist.py
+++ b/sukuinote/slave-plugins/anilist.py
@@ -35,6 +35,12 @@ MEDIA_QUERY = '''query ($id: Int, $search: String) {
airingAt
timeUntilAiring
}
+ airingSchedule(notYetAired: true) {
+ nodes {
+ airingAt
+ episode
+ }
+ }
siteUrl
}
}
@@ -105,7 +111,12 @@ async def generate_media(anilist):
if average_score is not None:
text += f'Average Score: {average_score}%\n'
if episodes:
- text += f'Episodes: {episodes}\n'
+ text += f'Episodes: '
+ if anilist['airingSchedule']:
+ for i in anilist['airingSchedule']['nodes']:
+ if i['airingAt'] == anilist['nextAiringEpisode']['airingAt']:
+ text += f'{i["episode"] - 1}/'
+ text += f'{episodes}\n'
if duration:
text += f'Duration: {duration} minutes per episode\n'
if chapters: