CodeCow/AutoUpdateJdCookie
1
0
Fork 0
mirror of https://github.com/icepage/AutoUpdateJdCookie.git synced 2026-06-05 21:02:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update main.py

Browse Source 
更新成功和更新失败的错误消息，进行合并发送，账号较多的时候，可以减少通知的数量。
This commit is contained in:
Shining Hu 2024-11-18 17:14:30 +08:00 committed by GitHub
parent 02d52c1d01
commit c5e1553878
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 27 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
main.py
View File

@ -578,12 +578,15 @@ async def main(mode: str = None):
# 登录JD获取pt_key
async with async_playwright() as playwright:
success_users = []
failed_users = []
for user in user_dict:
logger.info(f"开始更新{user}")
pt_key = await get_jd_pt_key(playwright, user, mode)
if pt_key is None:
logger.error(f"获取pt_key失败")
await send_msg(send_api, send_type=1, msg=f"{user} 更新失败")
failed_users.append(user)
continue
req_data = user_dict[user]
@ -593,19 +596,40 @@ async def main(mode: str = None):
response = await qlapi.set_envs(data=data)
if response['code'] == 200:
logger.info(f"{user}更新成功")
success_users.append(user)
else:
logger.error(f"{user}更新失败, response: {response}")
await send_msg(send_api, send_type=1, msg=f"{user} 更新失败")
failed_users.append(user)
continue
data = bytes(f"[{req_data['id']}]", 'utf-8')
response = await qlapi.envs_enable(data=data)
if response['code'] == 200:
logger.info(f"{user}启用成功")
await send_msg(send_api, send_type=0, msg=f"{user} 更新成功")
else:
logger.error(f"{user}启用失败, response: {response}")
# 处理用户名隐私
def mask_username(username):
if len(username) > 8: # 确保用户名长度足够进行替换
start = len(username) // 2 - 2
return username[:start] + '****' + username[start + 4:]
return username # 如果用户名长度不足，则不进行修改
masked_success_users = [mask_username(user) for user in success_users]
masked_failed_users = [mask_username(user) for user in failed_users]
# 附加消息
additional_message = "\n 解决办法：ABCD！"
# 最终发送消息
if masked_success_users:
success_msg = "更新成功的用户:\n" + "\n".join(masked_success_users) + "\n"
await send_msg(send_api, send_type=0, msg=success_msg)
if masked_failed_users:
failed_msg = "更新失败的用户:\n" + "\n".join(masked_failed_users) + "\n" + additional_message
await send_msg(send_api, send_type=1, msg=failed_msg)
except Exception as e:
traceback.print_exc()