You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
flyapps/fir_ser/xsign/utils/iproxy.py

95 lines
3.4 KiB

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: 12月
# author: NinEveN
# date: 2021/12/10
import logging
import random
import time
from concurrent.futures import ThreadPoolExecutor
import requests
from django.core.cache import cache
from common.cache.storage import IpProxyListCache, IpProxyActiveCache
from common.core.sysconfig import Config
logger = logging.getLogger(__name__)
def get_best_proxy_ips(url='https://api.appstoreconnect.apple.com/agreement'):
try:
active_proxy_ips = [proxy_info['proxy'] for proxy_info in Config.APPLE_DEVELOPER_API_PROXY_LIST if
proxy_info.get('active')]
except Exception as e:
logger.error(f'get_best_proxy_ips failed Exception:{e}')
return
access_ip_info = []
if not active_proxy_ips:
return
def task(proxies):
start_time = time.time()
try:
r = requests.get(url, proxies={'http': proxies, 'https': proxies}, timeout=30)
access_ip_info.append({'ip': proxies, 'time': time.time() - start_time})
logger.info(f"ip:{proxies} code:{r.status_code} time: {time.time() - start_time}")
except Exception as e:
logger.warning(f"ip {proxies} check failed Exception:{e}")
pools = ThreadPoolExecutor(50)
for proxy_ip in active_proxy_ips:
pools.submit(task, proxy_ip)
pools.shutdown()
best_count = len(active_proxy_ips)
if len(active_proxy_ips) > 8:
best_count = int(best_count * 0.8)
best_sorted_ips = sorted(access_ip_info, key=lambda x: x.get('time'))[:best_count]
best_sorted_ips = [ip_proxy['ip'] for ip_proxy in best_sorted_ips]
IpProxyListCache().set_storage_cache(best_sorted_ips, 6 * 60 * 60)
return best_sorted_ips
def get_proxy_ip_from_cache(issuer_id, change_ip=False):
active_proxy_cache = IpProxyActiveCache(issuer_id)
active_ip_proxy = active_proxy_cache.get_storage_cache()
if not change_ip and active_ip_proxy:
logger.info(f"issuer_id:{issuer_id} get ip proxy cache {active_ip_proxy}")
return active_ip_proxy
list_proxy_cache = IpProxyListCache()
ip_proxy_result = list_proxy_cache.get_storage_cache()
if not ip_proxy_result:
with cache.lock('get_best_proxy_ips_locker', timeout=360):
ip_proxy_result = list_proxy_cache.get_storage_cache()
if not ip_proxy_result:
ip_proxy_result = get_best_proxy_ips()
if change_ip and ip_proxy_result:
try:
ip_proxy_result.remove(active_ip_proxy)
except Exception as e:
logger.warning(f'remove bad ip proxy failed {e}')
logger.error(f"remove bad ip proxy {active_ip_proxy}")
active_proxy_cache.del_storage_cache()
list_proxy_cache.set_storage_cache(ip_proxy_result, 24 * 60 * 60)
if ip_proxy_result and len(ip_proxy_result) > 0:
proxy_ip = ip_proxy_result[random.randint(0, 2 if len(ip_proxy_result) > 2 else len(ip_proxy_result) - 1)]
proxy_info = {
'http': proxy_ip,
'https': proxy_ip
}
else:
proxy_info = Config.APPLE_DEVELOPER_API_PROXY
logger.info(f"make ip proxy cache {proxy_info}")
active_proxy_cache.set_storage_cache(proxy_info, 60 * 60)
return proxy_info
def clean_ip_proxy_infos():
logger.info("clean ip proxy infos")
IpProxyListCache().del_storage_cache()
IpProxyActiveCache('*').del_many()