mirror of
https://github.com/zc-zhangchen/any-auto-register.git
synced 2026-05-08 16:24:07 +08:00
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
"""代理池 - 从数据库读取代理，支持轮询和按区域选取"""
|
|
from typing import Optional
|
|
from sqlmodel import Session, select
|
|
from .db import ProxyModel, engine
|
|
import time, threading, random
|
|
from datetime import datetime, timezone
|
|
|
|
|
|
class ProxyPool:
|
|
def __init__(self):
|
|
self._index = 0
|
|
self._lock = threading.Lock()
|
|
|
|
def get_next(self, region: str = "") -> Optional[str]:
|
|
"""加权轮询取一个可用代理，在高成功率代理间轮换"""
|
|
with Session(engine) as s:
|
|
q = select(ProxyModel).where(ProxyModel.is_active == True)
|
|
if region:
|
|
q = q.where(ProxyModel.region == region)
|
|
proxies = s.exec(q).all()
|
|
if not proxies:
|
|
return None
|
|
proxies.sort(
|
|
key=lambda p: p.success_count / max(p.success_count + p.fail_count, 1),
|
|
reverse=True
|
|
)
|
|
with self._lock:
|
|
idx = self._index % len(proxies)
|
|
self._index += 1
|
|
return proxies[idx].url
|
|
|
|
def report_success(self, url: str) -> None:
|
|
with Session(engine) as s:
|
|
p = s.exec(select(ProxyModel).where(ProxyModel.url == url)).first()
|
|
if p:
|
|
p.success_count += 1
|
|
p.last_checked = datetime.now(timezone.utc)
|
|
s.add(p)
|
|
s.commit()
|
|
|
|
def report_fail(self, url: str) -> None:
|
|
with Session(engine) as s:
|
|
p = s.exec(select(ProxyModel).where(ProxyModel.url == url)).first()
|
|
if p:
|
|
p.fail_count += 1
|
|
p.last_checked = datetime.now(timezone.utc)
|
|
# 连续失败超过10次自动禁用
|
|
if p.fail_count > 0 and p.success_count == 0 and p.fail_count >= 5:
|
|
p.is_active = False
|
|
s.add(p)
|
|
s.commit()
|
|
|
|
def check_all(self) -> dict:
|
|
"""检测所有代理可用性"""
|
|
import requests
|
|
with Session(engine) as s:
|
|
proxies = s.exec(select(ProxyModel)).all()
|
|
results = {"ok": 0, "fail": 0}
|
|
for p in proxies:
|
|
try:
|
|
r = requests.get("https://httpbin.org/ip",
|
|
proxies={"http": p.url, "https": p.url},
|
|
timeout=8)
|
|
if r.status_code == 200:
|
|
self.report_success(p.url)
|
|
results["ok"] += 1
|
|
continue
|
|
except Exception:
|
|
pass
|
|
self.report_fail(p.url)
|
|
results["fail"] += 1
|
|
return results
|
|
|
|
|
|
proxy_pool = ProxyPool()
|