refactor: IP代理池重构

proxy/providers/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+# @Author  : relakkes@gmail.com
+# @Time    : 2024/4/5 10:13
+# @Desc    :
+from .jishu_http_proxy import new_jisu_http_proxy
+from .kuaidl_proxy import new_kuai_daili_proxy

proxy/providers/jishu_http_proxy.py

@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*-
+# @Author  : relakkes@gmail.com
+# @Time    : 2024/4/5 09:32
+# @Desc    : 极速HTTP代理提供类实现,官网地址：https://www.jisuhttp.com?pl=zG3Jna
+import os
+from typing import Dict, List
+from urllib.parse import urlencode
+
+import httpx
+
+from proxy import IpGetError, ProxyProvider, RedisDbIpCache
+from proxy.types import IpInfoModel
+from tools import utils
+
+
+class JiSuHttpProxy(ProxyProvider):
+    def __init__(self, key: str, crypto: str, time_validity_period: int):
+        """
+        极速HTTP 代理IP实现
+        :param key: 提取key值 (去官网注册后获取)
+        :param crypto: 加密签名 (去官网注册后获取)
+        """
+        self.proxy_brand_name = "JISUHTTP"
+        self.api_path = "https://api.jisuhttp.com"
+        self.params = {
+            "key": key,
+            "crypto": crypto,
+            "time": time_validity_period,  # IP使用时长，支持3、5、10、15、30分钟时效
+            "type": "json",  # 数据结果为json
+            "port": "2",  # IP协议：1:HTTP、2:HTTPS、3:SOCKS5
+            "pw": "1",  # 是否使用账密验证， 1：是，0：否，否表示白名单验证；默认为0
+            "se": "1",  # 返回JSON格式时是否显示IP过期时间， 1：显示，0：不显示；默认为0
+        }
+        self.ip_cache = RedisDbIpCache()
+
+    async def get_proxies(self, num: int) -> List[IpInfoModel]:
+        """
+        :param num:
+        :return:
+        """
+
+        # 优先从缓存中拿 IP
+        ip_cache_list = self.ip_cache.load_all_ip(proxy_brand_name=self.proxy_brand_name)
+        if len(ip_cache_list) >= num:
+            return ip_cache_list[:num]
+
+        # 如果缓存中的数量不够，从IP代理商获取补上，再存入缓存中
+        need_get_count = num - len(ip_cache_list)
+        self.params.update({"num": need_get_count})
+        ip_infos = []
+        async with httpx.AsyncClient() as client:
+            url = self.api_path + "/fetchips" + '?' + urlencode(self.params)
+            utils.logger.info(f"[JiSuHttpProxy.get_proxies] get ip proxy url:{url}")
+            response = await client.get(url, headers={
+                "User-Agent": "MediaCrawler https://github.com/NanmiCoder/MediaCrawler"})
+            res_dict: Dict = response.json()
+            if res_dict.get("code") == 0:
+                data: List[Dict] = res_dict.get("data")
+                current_ts = utils.get_unix_timestamp()
+                for ip_item in data:
+                    ip_info_model = IpInfoModel(
+                        ip=ip_item.get("ip"),
+                        port=ip_item.get("port"),
+                        user=ip_item.get("user"),
+                        password=ip_item.get("pass"),
+                        expired_time_ts=utils.get_unix_time_from_time_str(ip_item.get("expire"))
+                    )
+                    ip_key = f"JISUHTTP_{ip_info_model.ip}_{ip_info_model.port}_{ip_info_model.user}_{ip_info_model.password}"
+                    ip_value = ip_info_model.json()
+                    ip_infos.append(ip_info_model)
+                    self.ip_cache.set_ip(ip_key, ip_value, ex=ip_info_model.expired_time_ts - current_ts)
+            else:
+                raise IpGetError(res_dict.get("msg", "unkown err"))
+        return ip_cache_list + ip_infos
+
+
+def new_jisu_http_proxy() -> JiSuHttpProxy:
+    """
+    构造极速HTTP实例
+    Returns:
+
+    """
+    return JiSuHttpProxy(
+        key=os.getenv("jisu_key", ""),  # 通过环境变量的方式获取极速HTTPIP提取key值
+        crypto=os.getenv("jisu_crypto", ""),  # 通过环境变量的方式获取极速HTTPIP提取加密签名
+        time_validity_period=30  # 30分钟（最长时效）
+    )

proxy/providers/kuaidl_proxy.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# @Author  : relakkes@gmail.com
+# @Time    : 2024/4/5 09:43
+# @Desc    : 快代理HTTP实现，官方文档：https://www.kuaidaili.com/?ref=ldwkjqipvz6c
+from typing import Dict, List
+
+from proxy import IpGetError, IpInfoModel, ProxyProvider, RedisDbIpCache
+
+
+class KuaiDaiLiProxy(ProxyProvider):
+    async def get_proxies(self, num: int) -> List[Dict]:
+        pass
+
+
+def new_kuai_daili_proxy() -> KuaiDaiLiProxy:
+    """
+    构造快代理HTTP实例
+    Returns:
+
+    """
+    return KuaiDaiLiProxy()