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

@ -54,14 +54,24 @@ class Utils(object):
urls = Urls()
try:
response = json.loads(requests.post(
url= urls.GET_XB_PATH, data={"param" : url}, headers=headers).text)
url=urls.GET_XB_PATH, data={"param": url}, headers=headers).text)
params = response["param"]
xb = response["X-Bogus"]
except Exception as e:
print('[ 错误 ]:X-Bogus接口异常, 可能是访问流量高, 接口限流请稍等几分钟再次尝试')
return
return params #, xb
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__":