nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-05-11 10:35:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Update api name, add dev.sh

Browse Source
This commit is contained in:
estrella
2023-05-27 21:45:05 +08:00
parent 84b8fac003
commit b0a294bbb7
4 changed files with 42 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/module/api/bangumi.py
View File

@@ -21,7 +21,7 @@ async def get_all_data(current_user=Depends(get_current_user)):
@router.get(
"/api/v1/bangumi/getData/{bangumi_id}", tags=["bangumi"], response_model=BangumiData
"/api/v1/bangumi/getRule/{bangumi_id}", tags=["bangumi"], response_model=BangumiData
)
async def get_data(bangumi_id: str, current_user=Depends(get_current_user)):
if not current_user:
@@ -32,8 +32,8 @@ async def get_data(bangumi_id: str, current_user=Depends(get_current_user)):
return torrent.search_data(bangumi_id)
@router.post("/api/v1/bangumi/updateData", tags=["bangumi"])
async def update_data(data: BangumiData, current_user=Depends(get_current_user)):
@router.post("/api/v1/bangumi/updateRule", tags=["bangumi"])
async def update_rule(data: BangumiData, current_user=Depends(get_current_user)):
if not current_user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, detail="invalid token"

20
src/module/manager/renamer.py
View File

@@ -44,7 +44,7 @@ class Renamer(DownloadClient):
elif method == "subtitle_advance":
return f"{bangumi_name} S{season}E{episode}.{file_info.language}{file_info.suffix}"
else:
logger.error(f"Unknown rename method: {method}")
logger.error(f"[Renamer] Unknown rename method: {method}")
return file_info.media_path
@staticmethod
@@ -61,9 +61,9 @@ class Renamer(DownloadClient):
with PostNotification() as notificator:
status = notificator.send_msg(n)
if status:
logger.info(f"Notification sent: {ep.title} S{ep.season}E{ep.episode}")
logger.info(f"[Renamer] Notification sent: {ep.title} S{ep.season}E{ep.episode}")
else:
logger.warning(f"Notification failed: {ep.title} S{ep.season}E{ep.episode}")
logger.warning(f"[Renamer] Notification failed: {ep.title} S{ep.season}E{ep.episode}")
def rename_file(
self,
@@ -90,7 +90,7 @@ class Renamer(DownloadClient):
if settings.notification.enable:
self.send_notification(bangumi_name, ep)
return True
logger.warning(f"{media_path} parse failed")
logger.warning(f"[Renamer] {media_path} parse failed")
if settings.bangumi_manage.remove_bad_torrent:
self.delete_torrent(hashes=_hash)
@@ -116,7 +116,7 @@ class Renamer(DownloadClient):
_hash=_hash, old_path=media_path, new_path=new_path
)
if not renamed:
logger.warning(f"{media_path} rename failed")
logger.warning(f"[Renamrr] {media_path} rename failed")
# Delete bad torrent.
if settings.bangumi_manage.remove_bad_torrent:
self.delete_torrent(_hash)
@@ -147,11 +147,11 @@ class Renamer(DownloadClient):
_hash=_hash, old_path=subtitle_path, new_path=new_path
)
if not renamed:
logger.warning(f"{subtitle_path} rename failed")
logger.warning(f"[Renamer] {subtitle_path} rename failed")
def rename(self):
# Get torrent info
logger.debug("Start rename process.")
logger.debug("[Renamer] Start rename process.")
rename_method = settings.bangumi_manage.rename_method
torrents_info = self.get_torrent_info()
for info in torrents_info:
@@ -172,14 +172,14 @@ class Renamer(DownloadClient):
self.rename_subtitles(subtitle_list=subtitle_list, **kwargs)
# Rename collection
elif len(media_list) > 1:
logger.info("Start rename collection")
logger.info("[Renamer] Start rename collection")
self.rename_collection(media_list=media_list, **kwargs)
if len(subtitle_list) > 0:
self.rename_subtitles(subtitle_list=subtitle_list, **kwargs)
self.set_category(info.hash, "BangumiCollection")
else:
logger.warning(f"{info.name} has no media file")
logger.debug("Rename process finished.")
logger.warning(f"[Renamer] {info.name} has no media file")
logger.debug("[Renamer] Rename process finished.")
if __name__ == "__main__":

10
src/module/rss/analyser.py
View File

@@ -58,7 +58,7 @@ class RSSAnalyser:
if not full_parse:
return [data]
new_data.append(data)
logger.debug(f"New title found: {data.official_title}")
logger.debug(f"[RSS] New title found: {data.official_title}")
return new_data
def torrent_to_data(self, torrent: TorrentInfo, rss_link: str | None = None) -> BangumiData:
@@ -80,7 +80,7 @@ class RSSAnalyser:
with BangumiDatabase() as database:
torrents_to_add = database.match_list(rss_torrents, rss_link)
if not torrents_to_add:
logger.debug("No new title found.")
logger.debug("[RSS] No new title has been found.")
return []
# New List
new_data = self.torrents_to_data(torrents_to_add, rss_link, full_parse)
@@ -89,9 +89,9 @@ class RSSAnalyser:
return new_data
def run(self, rss_link: str = settings.rss_link):
logger.info("Start collecting RSS info.")
logger.info("[RSS] Start collecting RSS info.")
try:
self.rss_to_data(rss_link)
except Exception as e:
logger.debug(e)
logger.error("Failed to collect RSS info.")
logger.debug(f"[RSS] {e}")
logger.error("[RSS] Failed to collect RSS info.")