diff --git a/dev.sh b/dev.sh new file mode 100644 index 00000000..b9bfe925 --- /dev/null +++ b/dev.sh @@ -0,0 +1,24 @@ +#!/bin/bash + + +# This script is used to run the development environment. + +python3 -m pip install -r requirements.txt -U + +cd src || exit + +CONFIG_DIR="config" + +if [ ! -d "$CONFIG_DIR" ]; then + echo "The directory '$CONFIG_DIR' is missing." + mkdir config +fi + +VERSION_FILE="module/__version__.py" + +if [ ! -f "$VERSION_FILE" ]; then + echo "The file '$VERSION_FILE' is missing." + echo "VERSION='DEV_VERSION'" >> "$VERSION_FILE" +fi + +python3 main.py \ No newline at end of file diff --git a/src/module/api/bangumi.py b/src/module/api/bangumi.py index 6def1480..c23f4152 100644 --- a/src/module/api/bangumi.py +++ b/src/module/api/bangumi.py @@ -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" diff --git a/src/module/manager/renamer.py b/src/module/manager/renamer.py index 44aaa7e4..7e2cbcf8 100644 --- a/src/module/manager/renamer.py +++ b/src/module/manager/renamer.py @@ -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__": diff --git a/src/module/rss/analyser.py b/src/module/rss/analyser.py index eee0e351..d3b93126 100644 --- a/src/module/rss/analyser.py +++ b/src/module/rss/analyser.py @@ -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.")