style(tiktok): 改善提示信息

This commit is contained in:
imgyh 2023-02-22 12:35:22 +08:00
parent ac73a97c19
commit c5b890b693
2 changed files with 24 additions and 21 deletions

View File

@ -94,7 +94,7 @@ class TikTok(object):
# 传入 aweme_id
# 返回 数据 字典
def getAwemeInfo(self, aweme_id):
print('[ 提示 ]:正在请求的作品 id = %s\r\n' % aweme_id)
print('[ 提示 ]:正在请求的作品 id = %s\r' % aweme_id)
if aweme_id is None:
return None
@ -145,7 +145,7 @@ class TikTok(object):
times = 0
while True:
times = times + 1
print("[ 提示 ]:正在进行第 " + str(times) + " 次请求...\r")
print("[ 提示 ]:正在对 [主页] 进行第 " + str(times) + " 次请求...\r")
if mode == "post":
url = self.urls.USER_POST + self.utils.getXbogus(
url=f'sec_uid={sec_uid}&count={count}&max_cursor={max_cursor}')
@ -161,7 +161,8 @@ class TikTok(object):
try:
res = requests.get(url=url, headers=self.headers)
datadict = json.loads(res.text)
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["aweme_list"])) + ' 条数据')
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["aweme_list"])) + ' 条数据\r')
print('[ 提示 ]:开始对 ' + str(len(datadict["aweme_list"])) + ' 条数据请求作品详情\r\n')
if datadict is not None and datadict["status_code"] == 0:
break
except Exception as e:
@ -179,10 +180,10 @@ class TikTok(object):
# 退出条件
if datadict["has_more"] == 0 or datadict["has_more"] == False:
print("[ 提示 ]:所有作品数据获取完成...\r\n")
print("\r\n[ 提示 ]: [主页] 下所有作品数据获取完成...\r\n")
break
else:
print("[ 提示 ]:" + str(times) + " 次请求成功...\r\n")
print("\r\n[ 提示 ]:[主页] " + str(times) + " 次请求成功...\r\n")
return awemeList
@ -278,7 +279,7 @@ class TikTok(object):
times = 0
while True:
times = times + 1
print("[ 提示 ]:正在进行第 " + str(times) + " 次请求...\r")
print("[ 提示 ]:正在对 [合集] 进行第 " + str(times) + " 次请求...\r")
url = 'https://www.douyin.com/aweme/v1/web/mix/aweme/?' + self.utils.getXbogus(
url=f'device_platform=webapp&aid=6383&os_version=10&version_name=17.4.0&mix_id={mix_id}&cursor={cursor}&count={count}')
@ -288,7 +289,8 @@ class TikTok(object):
try:
res = requests.get(url=url, headers=self.headers)
datadict = json.loads(res.text)
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["aweme_list"])) + ' 条数据')
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["aweme_list"])) + ' 条数据\r')
print('[ 提示 ]:开始对 ' + str(len(datadict["aweme_list"])) + ' 条数据请求作品详情\r\n')
if datadict is not None:
break
except Exception as e:
@ -306,10 +308,10 @@ class TikTok(object):
# 退出条件
if datadict["has_more"] == 0 or datadict["has_more"] == False:
print("\r\n[ 提示 ]:合集下所有作品数据获取完成...\r\n")
print("\r\n[ 提示 ]:[合集] 下所有作品数据获取完成...\r\n")
break
else:
print("[ 提示 ]:" + str(times) + " 次请求成功...\r\n")
print("\r\n[ 提示 ]:[合集] " + str(times) + " 次请求成功...\r\n")
return awemeList
@ -321,12 +323,12 @@ class TikTok(object):
cursor = 0
mixIdNameDict = {}
print("[ 提示 ]:正在获取所有合集 id 数据请稍后...\r")
print("[ 提示 ]:正在获取主页下所有合集 id 数据请稍后...\r")
print("[ 提示 ]:会进行多次请求,等待时间较长...\r\n")
times = 0
while True:
times = times + 1
print("[ 提示 ]:正在进行第 " + str(times) + " 次请求...\r")
print("[ 提示 ]:正在对 [合集列表] 进行第 " + str(times) + " 次请求...\r")
url = self.urls.USER_MIX_LIST + self.utils.getXbogus(
url=f'device_platform=webapp&aid=6383&os_version=10&version_name=17.4.0&sec_user_id={sec_uid}&count={count}&cursor={cursor}')
@ -336,7 +338,8 @@ class TikTok(object):
try:
res = requests.get(url=url, headers=self.headers)
datadict = json.loads(res.text)
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["mix_infos"])) + ' 条数据')
print('[ 提示 ]:本次请求返回 ' + str(len(datadict["mix_infos"])) + ' 条数据\r')
print('[ 提示 ]:开始对 ' + str(len(datadict["mix_infos"])) + ' 条数据请求作品详情\r\n')
if datadict is not None and datadict["status_code"] == 0:
break
except Exception as e:
@ -350,10 +353,10 @@ class TikTok(object):
# 退出条件
if datadict["has_more"] == 0 or datadict["has_more"] == False:
print("[ 提示 ]:所有合集 id 数据获取完成...\r\n")
print("[ 提示 ]:[合集列表] 下所有合集 id 数据获取完成...\r\n")
break
else:
print("[ 提示 ]:" + str(times) + " 次请求成功...\r\n")
print("\r\n[ 提示 ]:[合集列表] " + str(times) + " 次请求成功...\r\n")
return mixIdNameDict
@ -486,7 +489,7 @@ class TikTok(object):
if not os.path.exists(savePath):
os.mkdir(savePath)
for ind, aweme in enumerate(awemeList):
print("[ 提示 ]:正在下载 %s 的作品 %s/%s\r"
print("[ 提示 ]:正在下载 [%s] 的作品 %s/%s\r\n"
% (aweme["author"]["nickname"], str(ind + 1), len(awemeList)))
self.awemeDownload(aweme, music, cover, avatar, savePath)

View File

@ -59,12 +59,12 @@ def main():
mixIdNameDict = tk.getUserAllMixInfo(key, 35)
for mix_id in mixIdNameDict:
print(f'\r\n[ 提示 ]:正在下载合集 [{mixIdNameDict[mix_id]}] 中的作品\r\n')
print(f'[ 提示 ]:正在下载合集 [{mixIdNameDict[mix_id]}] 中的作品\r\n')
mix_file_name = utils.replaceStr(mixIdNameDict[mix_id])
datalist = tk.getMixInfo(mix_id, 35)
tk.userDownload(awemeList=datalist, music=args.music, cover=args.cover, avatar=args.avatar,
savePath=os.path.join(args.path, mix_file_name))
print(f'\r\n[ 提示 ]:合集 [{mixIdNameDict[mix_id]}] 中的作品下载完成\r\n')
print(f'[ 提示 ]:合集 [{mixIdNameDict[mix_id]}] 中的作品下载完成\r\n')
elif key_type == "mix":
datalist = tk.getMixInfo(key,35)
tk.userDownload(awemeList=datalist, music=args.music, cover=args.cover, avatar=args.avatar,