nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-04-24 18:40:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

make DownloadClient mother class for all need to use that

Browse Source
This commit is contained in:
EstrellaXD
2023-05-07 11:59:11 +08:00
parent 5c71b9d35d
commit 98787552ca
6 changed files with 42 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/module/manager/eps_complete.py
View File

@@ -10,8 +10,9 @@ from module.models import BangumiData, Config
logger = logging.getLogger(__name__)
class FullSeasonGet:
class FullSeasonGet(DownloadClient):
def __init__(self, settings: Config):
super().__init__(settings)
self.SEARCH_KEY = [
"group",
"title_raw",
@@ -61,27 +62,27 @@ class FullSeasonGet:
downloads.append(download_info)
return downloads
def download_season(self, data: BangumiData, download_client: DownloadClient):
def download_season(self, data: BangumiData):
logger.info(f"Start collecting {data.official_title} Season {data.season}...")
torrents = self.get_season_torrents(data)
downloads = self.collect_season_torrents(data, torrents)
for download in downloads:
download_client.add_torrent(download)
self.add_torrent(download)
logger.info("Completed!")
data.eps_collect = False
def eps_complete(self, datas: list[BangumiData], download_client: DownloadClient):
def eps_complete(self, datas: list[BangumiData]):
for data in datas:
if data.eps_collect:
self.download_season(data, download_client)
self.download_season(data)
def download_collection(
self, data: BangumiData, link, download_client: DownloadClient
self, data: BangumiData, link
):
with RequestContent() as req:
torrents = req.get_torrents(link)
downloads = self.collect_season_torrents(data, torrents)
logger.info(f"Starting download {data.official_title} Season {data.season}...")
for download in downloads:
download_client.add_torrent(download)
self.add_torrent(download)
logger.info("Completed!")

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

@@ -12,9 +12,9 @@ from module.models import Config
logger = logging.getLogger(__name__)
class Renamer:
def __init__(self, download_client: DownloadClient, settings: Config):
self._client = download_client
class Renamer(DownloadClient):
def __init__(self, settings: Config):
super().__init__(settings)
self._renamer = TitleParser()
self._notification = PostNotification()
self.settings = settings
@@ -28,7 +28,7 @@ class Renamer:
logger.debug(f"Checked {torrent_count} files")
def get_torrent_info(self, category="Bangumi"):
recent_info = self._client.get_torrent_info(category=category)
recent_info = self.get_torrent_info(category=category)
torrent_count = len(recent_info)
return recent_info, torrent_count
@@ -66,7 +66,7 @@ class Renamer:
)
if compare_name != new_path:
try:
self._client.rename_torrent_file(
self.rename_torrent_file(
_hash=info.hash, old_path=media_path, new_path=new_path
)
self._notification.send_msg(bangumi_name, f"{new_path}已经更新，已自动重命名。")
@@ -103,7 +103,7 @@ class Renamer:
)
if torrent_name != new_name:
try:
self._client.rename_torrent_file(
self.rename_torrent_file(
_hash=_hash, old_path=media_path, new_path=new_name
)
except Exception as e:
@@ -114,7 +114,7 @@ class Renamer:
logger.debug(e)
# Delete bad torrent.
self.delete_bad_torrent(info, remove_bad_torrents)
self._client.set_category(category="BangumiCollection", hashes=_hash)
self.set_category(category="BangumiCollection", hashes=_hash)
def rename_subtitles(
self,
@@ -137,7 +137,7 @@ class Renamer:
)
if old_name != new_name:
try:
self._client.rename_torrent_file(
self.rename_torrent_file(
_hash=_hash, old_path=subtitle_path, new_path=new_name
)
except Exception as e:
@@ -147,7 +147,7 @@ class Renamer:
def delete_bad_torrent(self, info, remove_bad_torrent: bool):
if remove_bad_torrent:
self._client.delete_torrent(info.hash)
self.delete_torrent(info.hash)
logger.info(f"{info.name} have been deleted.")
@staticmethod