saveweb/biliarchiver
1
0
Fork 0
mirror of https://github.com/saveweb/biliarchiver.git synced 2024-09-19 11:05:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add config command to biliarchiver.py in cli_tools directory

Browse Source
This commit is contained in:
Ovler 2023-08-29 12:30:41 +08:00
parent d2a141e186
commit ca1273ed8c
No known key found for this signature in database
GPG Key ID: 87C2223333333333
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
biliarchiver/cli_tools/biliarchiver.py
View File

@ -3,6 +3,7 @@ from biliarchiver.i18n import _
from biliarchiver.cli_tools.up_command import up
from biliarchiver.cli_tools.down_command import down
from biliarchiver.cli_tools.get_command import get
from biliarchiver.cli_tools.conf_command import config
from biliarchiver.version import BILI_ARCHIVER_VERSION
@ -58,6 +59,7 @@ def init():
biliarchiver.add_command(up)
biliarchiver.add_command(down)
biliarchiver.add_command(get)
biliarchiver.add_command(config)
@biliarchiver.command(help=click.style(_("配置账号信息"), fg="cyan"))

24
biliarchiver/cli_tools/conf_command.py Normal file
View File

@ -0,0 +1,24 @@
import click
from dataclasses import dataclass
import os
import json
from pathlib import Path
from biliarchiver.i18n import _
from biliarchiver.config import _Config
@click.command(help=click.style(_("将传入参数写入配置文件"), fg="cyan"))
@click.option("--video_concurrency", "-v", type=click.INT, default=None, help=_("视频下载并发数"))
@click.option("--part_concurrency", "-p", type=click.INT, default=None, help=_("分P下载并发数"))
@click.option("--stream_retry", "-r", type=click.INT, default=None, help=_("流下载重试次数"))
@click.option("--storage_home_dir", "-s", type=click.STRING, default=None, help=_("存储目录"))
@click.option("--ia_key_file", "-i", type=click.STRING, default=None, help=_("IA key文件"))
@click.option("--cookies_file", "-c", type=click.STRING, default=None, help=_("cookies文件"))
def config(**kwargs):
config = _Config()
for k, v in kwargs.items():
if v is not None:
setattr(config, k, v)
config.save()
print(_("配置文件写入成功"))