diff --git a/biliarchiver/bili_get_bvids.py b/biliarchiver/bili_get_bvids.py index 54aadec..927c7b1 100644 --- a/biliarchiver/bili_get_bvids.py +++ b/biliarchiver/bili_get_bvids.py @@ -103,7 +103,7 @@ async def by_up_videos(url_or_mid: str) -> Path: return Path(abs_filepath) -async def main(): +async def _main(): args = arg_parse() if args.by_ranking: by_raning(args.ranking_rid) @@ -111,5 +111,8 @@ async def main(): assert args.up_videos_mid is not None, 'up_videos_mid 不能为空' await by_up_videos(args.up_videos_mid) +def main(): + asyncio.run(_main()) + if __name__ == '__main__': - asyncio.run(main()) \ No newline at end of file + main() \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index f372138..d5b66fa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,6 +14,7 @@ internetarchive = "^3.5.0" [tool.poetry.scripts] bili_archive_bvids = "biliarchiver:bili_archive_bvids.main" bili_uploade = "biliarchiver:bili_uploade.main" +bili_get_bvids = "biliarchiver:bili_get_bvids.main" [build-system]