fix(command): 修复头像封面音乐总是会下载

This commit is contained in:
imgyh 2023-02-27 10:27:45 +08:00
parent 0d0c77f882
commit 665ca47b08
2 changed files with 15 additions and 5 deletions

View File

@ -29,11 +29,11 @@ def argument():
parser.add_argument("--path", "-p", help="下载保存位置",
type=str, required=True)
parser.add_argument("--music", "-m", help="是否下载视频中的音乐(True/False), 默认为True",
type=bool, required=False, default=True)
type=Utils().str2bool, required=False, default=True)
parser.add_argument("--cover", "-c", help="是否下载视频的封面(True/False), 默认为True, 当下载视频时有效",
type=bool, required=False, default=True)
type=Utils().str2bool, required=False, default=True)
parser.add_argument("--avatar", "-a", help="是否下载作者的头像(True/False), 默认为True",
type=bool, required=False, default=True)
type=Utils().str2bool, required=False, default=True)
parser.add_argument("--mode", "-M", help="link是个人主页时, 设置下载发布的作品(post)或喜欢的作品(like)或者用户所有合集(mix), 默认为post",
type=str, required=False, default="post")
args = parser.parse_args()

View File

@ -63,6 +63,16 @@ class Utils(object):
return params # , xb
def str2bool(self, v):
if isinstance(v, bool):
return v
if v.lower() in ('yes', 'true', 't', 'y', '1'):
return True
elif v.lower() in ('no', 'false', 'f', 'n', '0'):
return False
else:
return True
if __name__ == "__main__":
pass