python多线程运行脚本模板——适用于多用户

本文 首发于 Anyeの小站，转载请取得作者同意。

多用户执行python脚本的一种风格：

• account.csv.example：账户信息文件（示例），用于配置账户信息。
• main.py：主程序文件，用于完成学习任务。
• one_account.py：单账户执行所使用的程序，主算法内容均位于此文件。

简单示例

account.csv.example

username1,password1,notes1
username2,password2,notes2
username3,password3,notes3

one_account.py

import argparse

def main(username,password): # 主函数,用于多用户调用
    ...
    
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Login script with captcha for multiple users.')
    parser.add_argument('username', type=str, help='Username for login')
    parser.add_argument('password', type=str, help='Password for login')
    args = parser.parse_args()
    main(args.username, args.password)

main.py

import csv
import threading
import subprocess

def process_account(row, result_list):
    username, password = row[0], row[1]

    try:
        output_bytes = subprocess.check_output(["python", "one_account.py", username, password])
        output = output_bytes.decode('utf-8')
    except subprocess.CalledProcessError as e:
        output = f"Error: {e.output.decode('utf-8')}"

    result_list.append((username, output))  # 将结果存入结果列表

if __name__ == '__main__':
    success = 0
    count = 0
    all_results = []  # 用于存储每个线程的结果列表

    with open('account.csv', 'r', encoding='utf-8') as f:
        reader = csv.reader(f)
        threads = []

        for row in reader:
            count += 1
            print(f'用户{count}: {row[2]}({row[0]})')

            # 创建一个线程并启动，为每个线程创建独立的结果列表
            results = []
            thread = threading.Thread(target=process_account, args=(row, results))
            threads.append(thread)
            thread.start()

            # 将每个线程的结果列表存入总的结果列表
            all_results.append(results)

        # 等待所有线程执行完毕
        for thread in threads:
            thread.join()
    print('\n所有用户运行完毕\n')
    # 统计成功和失败的用户
    for results in all_results:
        for username, output in results:
           	if "Error" not in output:
                success += 1
            print(f'用户{username}运行结果：\n{output}')

    print(f'共{count}个用户，成功{success}个，失败{count - success}个')