weverse-script/main.py

import dao
import sys
import proxy
import w_token
from concurrent.futures import ThreadPoolExecutor, as_completed
import json
import os

f = open('config.json', 'r')
content = f.read()
f.close()
data = json.loads(content)
workers = data["workers"]


def task(aid, username, password):
    # proxy_host = proxy.get_proxy()
    # t = w_token.get_token(username, password, proxy_host[0], proxy_host[1])
    t = w_token.get_token(username, password, None, None)
    return [aid, username, t]


def main():
    print("输入演出num")
    num = input()
    r = dao.get_apply(num)
    if not r:
        print("演出不存在")
        return
    print("演出信息确认：")
    print(r[0][2], r[0][3])
    t = input("是否确认？(Y/N):")
    if t == "N" or t == "n":
        return
    print("请确认人员名单：")
    u = dao.get_apply_user(r[0][0])
    if not u:
        print("人员不存在")
        return
    print("weverse账号  区号  手机号")
    for row in u:
        print(row[1] + " " + row[2] + " " + row[3])
    t = input("是否确认？(Y/N):")
    if t == "N" or t == "n":
        return

    count = 0
    with ThreadPoolExecutor(max_workers=workers) as executor:
        # 提交多个任务给线程池
        futures = [executor.submit(task, row[0], row[1], row[4]) for row in u]

        # 等待所有任务完成并获取结果
        for future in as_completed(futures):
            result = future.result()
            if not result[2]:
                print("applyId:【%s】,username:【%s】获取失败" % (result[0], result[1]))
                continue
            dao.update_token(result[0], result[2])
            count = count + 1
            print("applyId:【%s】,username:【%s】获取成功！！！" % (result[0], result[1]))
    print("一共【%d】人，成功【%d】人" % (len(u), count))


if __name__ == "__main__":
    os.system('chcp 65001')
    main()
    print("按下任意键以退出程序")
    input()