bili-sync/commands.py

import asyncio

from loguru import logger

from constants import MediaStatus, MediaType
from models import FavoriteItem, Upper
from processor import download_content, process_video
from utils import aexists, amakedirs, aremove


async def recheck():
    """刷新数据库中视频的状态，如果发现文件不存在则标记未下载，以便在下次任务重新下载，在自己手动删除文件后调用"""
    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,
            )
            continue
        if not exist:
            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"])
    logger.info("Database updated.")


async def upper_thumb():
    """将up主的头像批量写入数据库，从不支持up主头像的版本升级上来后需要手动调用一次"""
    makedir_tasks = []
    other_tasks = []
    for upper in await Upper.all():
        if all(
            await asyncio.gather(
                aexists(upper.thumb_path), aexists(upper.meta_path)
            )
        ):
            logger.info(
                "Upper {} {} already exists, skipped.", upper.mid, upper.name
            )
        makedir_tasks.append(amakedirs(upper.thumb_path.parent, exist_ok=True))
        logger.info("Saving metadata for upper {} {}...", upper.mid, upper.name)
        other_tasks.extend(
            [
                upper.save_metadata(),
                download_content(upper.thumb, upper.thumb_path),
            ]
        )
    await asyncio.gather(*makedir_tasks)
    await asyncio.gather(*other_tasks)
    logger.info("All done.")


async def refresh_tags():
    """刷新已存在的视频的标签，从不支持标签的版本升级上来后需要手动调用一次"""
    items = await FavoriteItem.filter(
        downloaded=True,
        tags=None,
    ).prefetch_related("upper")
    await asyncio.gather(
        *[aremove(item.nfo_path) for item in items],
        return_exceptions=True,
    )
    await asyncio.gather(
        *[
            process_video(
                item,
                process_poster=False,
                process_video=False,
                process_nfo=True,
                process_upper=False,
            )
            for item in items
        ],
        return_exceptions=True,
    )