kami_itunes_third_api/src/service/proxy.py

import copy
from time import time
from src.initialization import setting
import random

# 账户列表
account_list = {}


class ProxyService:
    def get_wrap_proxy(self, account_name: str):
        proxy = self.get_proxy(account_name)
        return {
            "http": proxy,
            "https": proxy,
        }

    def get_proxy(self, account_name: str):
        self.set_expire_strategy()
        if account_name in account_list:
            return account_list[account_name].address
        return self.set_proxy(account_name)

    def set_proxy(self, account_name: str):
        proxy = random.choice(setting.proxies.address)
        account_list[account_name] = {
            "address": proxy,
            "timeout": time.time() + 60 * 60 * 24,
        }
        return proxy

    # 设置触发账号过期策略
    def set_expire_strategy(self):
        for key, value in copy.deepcopy(account_list).items():
            if value.get("timeout") > time.time():
                account_list.pop(key)