match_spider/ybw/utils/socks5.py

# -*- coding: utf-8 -*-
"""
Created on 2024-02-27
---------
@summary: socks5 proxy代理
---------
@author: Dzr
"""
import requests

import setting
from utils.log import logger


def get_proxy(scheme=None, default=None, socks5h=False):
    headers = {'Authorization': setting.JY_AUTH_TOKEN}
    try:
        resp = requests.get(setting.PROXY_API, headers=headers, timeout=5).json()
        proxies = resp.get("data")
        if resp and proxies:
            if socks5h:
                proxy_items = proxies.get("http")
                # 请求代理 vps dns域名解析
                proxies = dict(
                    http=proxy_items.replace("socks5", "socks5h"),
                    https=proxy_items.replace("socks5", "socks5h")
                )

            logger.debug(f"切换代理:{proxies}")
            return proxies if not scheme else proxies.get(scheme, default)
        else:
            logger.info("暂无代理...")
    except Exception:
        pass

    return default