Spaces:
Running
Running
File size: 4,652 Bytes
de1ee14 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# This file is part of the AutoAnime distribution.
# Copyright (c) 2025 Kaif_00z
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# License can be found in <
# https://github.com/kaif-00z/AutoAnimeBot/blob/main/LICENSE > .
# if you are using this following code then don't forgot to give proper
# credit to t.me/kAiF_00z (github.com/kaif-00z)
from traceback import format_exc
import anitopy
from libs.kitsu import RawAnimeInfo
from libs.logger import LOGS
class AnimeInfo:
def __init__(self, name):
self.kitsu = RawAnimeInfo()
self.CAPTION = """
**{}
━━━━━━━━━━━━━━━
‣ Language:** `Japanese [ESub]`
**‣ Quality:** `480p|720p|1080p`
**‣ Season:** `{}`
**‣ Episode:** `{}`
**━━━━━━━━━━━━━━━**
"""
self.proper_name = self.get_proper_name_for_func(name)
self.name = name
self.data = anitopy.parse(name)
async def get_english(self):
anime_name = self.data.get("anime_title")
try:
anime = (await self.kitsu.search(self.proper_name)) or {}
return anime.get("english_title") or anime_name
except BaseException:
LOGS.error(str(format_exc()))
return anime_name.strip()
async def get_poster(self):
try:
if self.proper_name:
anime_poster = await self.kitsu.search(self.proper_name)
return anime_poster.get("poster_img") or None
except BaseException:
LOGS.error(str(format_exc()))
async def get_cover(self):
try:
if self.proper_name:
anime_poster = await self.kitsu.search(self.proper_name)
if anime_poster.get("anilist_id"):
return anime_poster.get("anilist_poster")
return None
except BaseException:
LOGS.error(str(format_exc()))
async def get_caption(self):
try:
if self.proper_name or self.data:
return self.CAPTION.format(
(await self.get_english()),
str(self.data.get("anime_season") or 1).zfill(2),
(
str(self.data.get("episode_number")).zfill(2)
if self.data.get("episode_number")
else "N/A"
),
)
except BaseException:
LOGS.error(str(format_exc()))
return ""
async def rename(self, original=False):
try:
anime_name = self.data.get("anime_title")
if anime_name and self.data.get("episode_number"):
return (
f"[S{self.data.get('anime_season') or 1}-{self.data.get('episode_number') or ''}] {(await self.get_english())} [{self.data.get('video_resolution')}].mkv".replace(
"‘", ""
)
.replace("’", "")
.strip()
)
if anime_name:
return (
f"{(await self.get_english())} [{self.data.get('video_resolution')}].mkv".replace(
"‘", ""
)
.replace("’", "")
.strip()
)
return self.name
except Exception as error:
LOGS.error(str(error))
LOGS.exception(format_exc())
return self.name
def get_proper_name_for_func(self, name):
try:
data = anitopy.parse(name)
anime_name = data.get("anime_title")
if anime_name and data.get("episode_number"):
return (
f"{anime_name} S{data.get('anime_season')} {data.get('episode_title')}"
if data.get("anime_season") and data.get("episode_title")
else (
f"{anime_name} S{data.get('anime_season')}"
if data.get("anime_season")
else anime_name
)
)
return anime_name
except Exception as error:
LOGS.error(str(error))
LOGS.exception(format_exc())
|