GithubBackup/bili-sync
1
0
Fork 0
mirror of https://github.com/amtoaer/bili-sync.git synced 2026-07-09 22:43:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 支持分 p 视频下载，待额外测试 (#24)

Browse Source
This commit is contained in:
ᴀᴍᴛᴏᴀᴇʀ
2024-02-24 03:38:08 +08:00
committed by GitHub
parent 1dd760d445
commit a4c362d8ab
11 changed files with 563 additions and 349 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
commands.py
View File

@@ -13,27 +13,14 @@ from utils import aexists, aremove
async def recheck():
"""刷新数据库中视频的状态，如果发现文件不存在则标记未下载，以便在下次任务重新下载，在自己手动删除文件后调用"""
items = await FavoriteItem.filter(
type=MediaType.VIDEO,
status=MediaStatus.NORMAL,
downloaded=True,
)
items = await FavoriteItem.filter(type=MediaType.VIDEO, status=MediaStatus.NORMAL, downloaded=True)
exists = await asyncio.gather(*[aexists(item.video_path) for item in items])
for item, exist in zip(items, exists):
if isinstance(exist, Exception):
logger.error(
"Error when checking file {} {}: {}",
item.bvid,
item.name,
exist,
)
logger.error("Error when checking file {} {}: {}.", item.bvid, item.name, exist)
continue
if not exist:
logger.info(
"File {} {} not exists, mark as not downloaded.",
item.bvid,
item.name,
)
logger.info("File {} {} not exists, mark as not downloaded.", item.bvid, item.name)
item.downloaded = False
logger.info("Updating database...")
await FavoriteItem.bulk_update(items, fields=["downloaded"])
@@ -52,10 +39,7 @@ async def _refresh_favorite_item_info(
items = await FavoriteItem.filter(downloaded=True).prefetch_related("upper")
if force:
# 如果强制刷新，那么就先把现存的所有内容删除
await asyncio.gather(
*[aremove(path) for item in items for path in path_getter(item)],
return_exceptions=True,
)
await asyncio.gather(*[aremove(path) for item in items for path in path_getter(item)], return_exceptions=True)
await asyncio.gather(
*[
process_favorite_item(
@@ -72,30 +56,14 @@ async def _refresh_favorite_item_info(
)
refresh_nfo = functools.partial(
_refresh_favorite_item_info, lambda item: [item.nfo_path], process_nfo=True
)
refresh_nfo = functools.partial(_refresh_favorite_item_info, lambda item: [item.nfo_path], process_nfo=True)
refresh_poster = functools.partial(
_refresh_favorite_item_info,
lambda item: [item.poster_path],
process_poster=True,
)
refresh_poster = functools.partial(_refresh_favorite_item_info, lambda item: [item.poster_path], process_poster=True)
refresh_video = functools.partial(
_refresh_favorite_item_info,
lambda item: [item.video_path],
process_video=True,
)
refresh_video = functools.partial(_refresh_favorite_item_info, lambda item: [item.video_path], process_video=True)
refresh_upper = functools.partial(
_refresh_favorite_item_info,
lambda item: item.upper_path,
process_upper=True,
)
refresh_upper = functools.partial(_refresh_favorite_item_info, lambda item: item.upper_path, process_upper=True)
refresh_subtitle = functools.partial(
_refresh_favorite_item_info,
lambda item: [item.subtitle_path],
process_subtitle=True,
_refresh_favorite_item_info, lambda item: [item.subtitle_path], process_subtitle=True
)