crawlo 1.2.0__py3-none-any.whl → 1.2.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of crawlo might be problematic. Click here for more details.

Files changed (220) hide show
  1. crawlo/__init__.py +61 -61
  2. crawlo/__version__.py +1 -1
  3. crawlo/cleaners/__init__.py +60 -60
  4. crawlo/cleaners/data_formatter.py +225 -225
  5. crawlo/cleaners/encoding_converter.py +125 -125
  6. crawlo/cleaners/text_cleaner.py +232 -232
  7. crawlo/cli.py +81 -65
  8. crawlo/commands/__init__.py +14 -14
  9. crawlo/commands/check.py +594 -594
  10. crawlo/commands/genspider.py +151 -151
  11. crawlo/commands/help.py +143 -133
  12. crawlo/commands/list.py +155 -155
  13. crawlo/commands/run.py +292 -292
  14. crawlo/commands/startproject.py +418 -418
  15. crawlo/commands/stats.py +188 -188
  16. crawlo/commands/utils.py +186 -186
  17. crawlo/config.py +312 -312
  18. crawlo/config_validator.py +252 -252
  19. crawlo/core/__init__.py +2 -2
  20. crawlo/core/engine.py +354 -354
  21. crawlo/core/processor.py +40 -40
  22. crawlo/core/scheduler.py +143 -143
  23. crawlo/crawler.py +1027 -1027
  24. crawlo/downloader/__init__.py +266 -266
  25. crawlo/downloader/aiohttp_downloader.py +220 -220
  26. crawlo/downloader/cffi_downloader.py +256 -256
  27. crawlo/downloader/httpx_downloader.py +259 -259
  28. crawlo/downloader/hybrid_downloader.py +213 -213
  29. crawlo/downloader/playwright_downloader.py +402 -402
  30. crawlo/downloader/selenium_downloader.py +472 -472
  31. crawlo/event.py +11 -11
  32. crawlo/exceptions.py +81 -81
  33. crawlo/extension/__init__.py +37 -37
  34. crawlo/extension/health_check.py +141 -141
  35. crawlo/extension/log_interval.py +57 -57
  36. crawlo/extension/log_stats.py +81 -81
  37. crawlo/extension/logging_extension.py +43 -43
  38. crawlo/extension/memory_monitor.py +104 -104
  39. crawlo/extension/performance_profiler.py +133 -133
  40. crawlo/extension/request_recorder.py +107 -107
  41. crawlo/filters/__init__.py +154 -154
  42. crawlo/filters/aioredis_filter.py +280 -280
  43. crawlo/filters/memory_filter.py +269 -269
  44. crawlo/items/__init__.py +23 -23
  45. crawlo/items/base.py +21 -21
  46. crawlo/items/fields.py +53 -53
  47. crawlo/items/items.py +104 -104
  48. crawlo/middleware/__init__.py +21 -21
  49. crawlo/middleware/default_header.py +132 -32
  50. crawlo/middleware/download_delay.py +105 -28
  51. crawlo/middleware/middleware_manager.py +135 -135
  52. crawlo/middleware/offsite.py +116 -0
  53. crawlo/middleware/proxy.py +366 -272
  54. crawlo/middleware/request_ignore.py +88 -30
  55. crawlo/middleware/response_code.py +164 -18
  56. crawlo/middleware/response_filter.py +138 -26
  57. crawlo/middleware/retry.py +124 -124
  58. crawlo/mode_manager.py +211 -211
  59. crawlo/network/__init__.py +21 -21
  60. crawlo/network/request.py +338 -338
  61. crawlo/network/response.py +359 -359
  62. crawlo/pipelines/__init__.py +21 -21
  63. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  64. crawlo/pipelines/console_pipeline.py +39 -39
  65. crawlo/pipelines/csv_pipeline.py +316 -316
  66. crawlo/pipelines/database_dedup_pipeline.py +224 -224
  67. crawlo/pipelines/json_pipeline.py +218 -218
  68. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  69. crawlo/pipelines/mongo_pipeline.py +131 -131
  70. crawlo/pipelines/mysql_pipeline.py +316 -316
  71. crawlo/pipelines/pipeline_manager.py +61 -61
  72. crawlo/pipelines/redis_dedup_pipeline.py +167 -167
  73. crawlo/project.py +187 -187
  74. crawlo/queue/pqueue.py +37 -37
  75. crawlo/queue/queue_manager.py +337 -337
  76. crawlo/queue/redis_priority_queue.py +298 -298
  77. crawlo/settings/__init__.py +7 -7
  78. crawlo/settings/default_settings.py +226 -219
  79. crawlo/settings/setting_manager.py +122 -122
  80. crawlo/spider/__init__.py +639 -639
  81. crawlo/stats_collector.py +59 -59
  82. crawlo/subscriber.py +130 -130
  83. crawlo/task_manager.py +30 -30
  84. crawlo/templates/crawlo.cfg.tmpl +10 -10
  85. crawlo/templates/project/__init__.py.tmpl +3 -3
  86. crawlo/templates/project/items.py.tmpl +17 -17
  87. crawlo/templates/project/middlewares.py.tmpl +118 -109
  88. crawlo/templates/project/pipelines.py.tmpl +96 -96
  89. crawlo/templates/project/run.py.tmpl +45 -45
  90. crawlo/templates/project/settings.py.tmpl +327 -326
  91. crawlo/templates/project/settings_distributed.py.tmpl +119 -119
  92. crawlo/templates/project/settings_gentle.py.tmpl +94 -94
  93. crawlo/templates/project/settings_high_performance.py.tmpl +151 -151
  94. crawlo/templates/project/settings_simple.py.tmpl +68 -68
  95. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  96. crawlo/templates/spider/spider.py.tmpl +143 -141
  97. crawlo/tools/__init__.py +182 -182
  98. crawlo/tools/anti_crawler.py +268 -268
  99. crawlo/tools/authenticated_proxy.py +240 -240
  100. crawlo/tools/data_validator.py +180 -180
  101. crawlo/tools/date_tools.py +35 -35
  102. crawlo/tools/distributed_coordinator.py +386 -386
  103. crawlo/tools/retry_mechanism.py +220 -220
  104. crawlo/tools/scenario_adapter.py +262 -262
  105. crawlo/utils/__init__.py +35 -35
  106. crawlo/utils/batch_processor.py +260 -260
  107. crawlo/utils/controlled_spider_mixin.py +439 -439
  108. crawlo/utils/date_tools.py +290 -290
  109. crawlo/utils/db_helper.py +343 -343
  110. crawlo/utils/enhanced_error_handler.py +359 -359
  111. crawlo/utils/env_config.py +105 -105
  112. crawlo/utils/error_handler.py +125 -125
  113. crawlo/utils/func_tools.py +82 -82
  114. crawlo/utils/large_scale_config.py +286 -286
  115. crawlo/utils/large_scale_helper.py +343 -343
  116. crawlo/utils/log.py +128 -128
  117. crawlo/utils/performance_monitor.py +284 -284
  118. crawlo/utils/queue_helper.py +175 -175
  119. crawlo/utils/redis_connection_pool.py +334 -334
  120. crawlo/utils/redis_key_validator.py +199 -199
  121. crawlo/utils/request.py +267 -267
  122. crawlo/utils/request_serializer.py +219 -219
  123. crawlo/utils/spider_loader.py +62 -62
  124. crawlo/utils/system.py +11 -11
  125. crawlo/utils/tools.py +4 -4
  126. crawlo/utils/url.py +39 -39
  127. {crawlo-1.2.0.dist-info → crawlo-1.2.2.dist-info}/METADATA +692 -697
  128. crawlo-1.2.2.dist-info/RECORD +220 -0
  129. examples/__init__.py +7 -7
  130. examples/aiohttp_settings.py +42 -0
  131. examples/curl_cffi_settings.py +41 -0
  132. examples/default_header_middleware_example.py +107 -0
  133. examples/default_header_spider_example.py +129 -0
  134. examples/download_delay_middleware_example.py +160 -0
  135. examples/httpx_settings.py +42 -0
  136. examples/multi_downloader_proxy_example.py +81 -0
  137. examples/offsite_middleware_example.py +55 -0
  138. examples/offsite_spider_example.py +107 -0
  139. examples/proxy_spider_example.py +166 -0
  140. examples/request_ignore_middleware_example.py +51 -0
  141. examples/request_ignore_spider_example.py +99 -0
  142. examples/response_code_middleware_example.py +52 -0
  143. examples/response_filter_middleware_example.py +67 -0
  144. examples/tong_hua_shun_settings.py +62 -0
  145. examples/tong_hua_shun_spider.py +170 -0
  146. tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +81 -81
  147. tests/__init__.py +7 -7
  148. tests/advanced_tools_example.py +275 -275
  149. tests/authenticated_proxy_example.py +236 -236
  150. tests/cleaners_example.py +160 -160
  151. tests/config_validation_demo.py +102 -102
  152. tests/controlled_spider_example.py +205 -205
  153. tests/date_tools_example.py +180 -180
  154. tests/dynamic_loading_example.py +523 -523
  155. tests/dynamic_loading_test.py +104 -104
  156. tests/env_config_example.py +133 -133
  157. tests/error_handling_example.py +171 -171
  158. tests/redis_key_validation_demo.py +130 -130
  159. tests/response_improvements_example.py +144 -144
  160. tests/test_advanced_tools.py +148 -148
  161. tests/test_all_redis_key_configs.py +145 -145
  162. tests/test_authenticated_proxy.py +141 -141
  163. tests/test_cleaners.py +54 -54
  164. tests/test_comprehensive.py +146 -146
  165. tests/test_config_validator.py +193 -193
  166. tests/test_crawlo_proxy_integration.py +173 -0
  167. tests/test_date_tools.py +123 -123
  168. tests/test_default_header_middleware.py +159 -0
  169. tests/test_double_crawlo_fix.py +207 -207
  170. tests/test_double_crawlo_fix_simple.py +124 -124
  171. tests/test_download_delay_middleware.py +222 -0
  172. tests/test_downloader_proxy_compatibility.py +269 -0
  173. tests/test_dynamic_downloaders_proxy.py +124 -124
  174. tests/test_dynamic_proxy.py +92 -92
  175. tests/test_dynamic_proxy_config.py +146 -146
  176. tests/test_dynamic_proxy_real.py +109 -109
  177. tests/test_edge_cases.py +303 -303
  178. tests/test_enhanced_error_handler.py +270 -270
  179. tests/test_env_config.py +121 -121
  180. tests/test_error_handler_compatibility.py +112 -112
  181. tests/test_final_validation.py +153 -153
  182. tests/test_framework_env_usage.py +103 -103
  183. tests/test_integration.py +356 -356
  184. tests/test_item_dedup_redis_key.py +122 -122
  185. tests/test_offsite_middleware.py +222 -0
  186. tests/test_parsel.py +29 -29
  187. tests/test_performance.py +327 -327
  188. tests/test_proxy_api.py +265 -0
  189. tests/test_proxy_health_check.py +32 -32
  190. tests/test_proxy_middleware.py +122 -0
  191. tests/test_proxy_middleware_enhanced.py +217 -0
  192. tests/test_proxy_middleware_integration.py +136 -136
  193. tests/test_proxy_providers.py +56 -56
  194. tests/test_proxy_stats.py +19 -19
  195. tests/test_proxy_strategies.py +59 -59
  196. tests/test_queue_manager_double_crawlo.py +173 -173
  197. tests/test_queue_manager_redis_key.py +176 -176
  198. tests/test_real_scenario_proxy.py +196 -0
  199. tests/test_redis_config.py +28 -28
  200. tests/test_redis_connection_pool.py +294 -294
  201. tests/test_redis_key_naming.py +181 -181
  202. tests/test_redis_key_validator.py +123 -123
  203. tests/test_redis_queue.py +224 -224
  204. tests/test_request_ignore_middleware.py +183 -0
  205. tests/test_request_serialization.py +70 -70
  206. tests/test_response_code_middleware.py +350 -0
  207. tests/test_response_filter_middleware.py +428 -0
  208. tests/test_response_improvements.py +152 -152
  209. tests/test_retry_middleware.py +242 -0
  210. tests/test_scheduler.py +241 -241
  211. tests/test_simple_response.py +61 -61
  212. tests/test_telecom_spider_redis_key.py +205 -205
  213. tests/test_template_content.py +87 -87
  214. tests/test_template_redis_key.py +134 -134
  215. tests/test_tools.py +153 -153
  216. tests/tools_example.py +257 -257
  217. crawlo-1.2.0.dist-info/RECORD +0 -190
  218. {crawlo-1.2.0.dist-info → crawlo-1.2.2.dist-info}/WHEEL +0 -0
  219. {crawlo-1.2.0.dist-info → crawlo-1.2.2.dist-info}/entry_points.txt +0 -0
  220. {crawlo-1.2.0.dist-info → crawlo-1.2.2.dist-info}/top_level.txt +0 -0
@@ -1,273 +1,367 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- import asyncio
4
- import socket
5
- from typing import Optional, Dict, Any, Callable, Union, TYPE_CHECKING
6
- from urllib.parse import urlparse
7
-
8
- from crawlo import Request, Response
9
- from crawlo.exceptions import NotConfiguredError
10
- from crawlo.utils.log import get_logger
11
-
12
- if TYPE_CHECKING:
13
- import aiohttp
14
-
15
- try:
16
- import httpx
17
- HTTPX_EXCEPTIONS = (httpx.NetworkError, httpx.TimeoutException, httpx.ReadError, httpx.ConnectError)
18
- except ImportError:
19
- HTTPX_EXCEPTIONS = ()
20
- httpx = None
21
-
22
- try:
23
- import aiohttp
24
- AIOHTTP_EXCEPTIONS = (
25
- aiohttp.ClientError, aiohttp.ClientConnectorError, aiohttp.ClientResponseError, aiohttp.ServerTimeoutError,
26
- aiohttp.ServerDisconnectedError)
27
- except ImportError:
28
- AIOHTTP_EXCEPTIONS = ()
29
- aiohttp = None
30
-
31
- try:
32
- from curl_cffi import requests as cffi_requests
33
- CURL_CFFI_EXCEPTIONS = (cffi_requests.RequestsError,)
34
- except (ImportError, AttributeError):
35
- CURL_CFFI_EXCEPTIONS = ()
36
- cffi_requests = None
37
-
38
- NETWORK_EXCEPTIONS = (
39
- asyncio.TimeoutError,
40
- socket.gaierror,
41
- ConnectionError,
42
- TimeoutError,
43
- ) + HTTPX_EXCEPTIONS + AIOHTTP_EXCEPTIONS + CURL_CFFI_EXCEPTIONS
44
-
45
- ProxyExtractor = Callable[[Dict[str, Any]], Union[None, str, Dict[str, str]]]
46
-
47
-
48
- class ProxyMiddleware:
49
- def __init__(self, settings, log_level):
50
- self.logger = get_logger(self.__class__.__name__, log_level)
51
-
52
- self._session: Optional[Any] = None # aiohttp.ClientSession when aiohttp is available
53
- self._current_proxy: Optional[Union[str, Dict[str, str]]] = None
54
- self._last_fetch_time: float = 0
55
-
56
- self.proxy_extractor = settings.get("PROXY_EXTRACTOR", "proxy")
57
- self.refresh_interval = settings.get_float("PROXY_REFRESH_INTERVAL", 60)
58
- self.timeout = settings.get_float("PROXY_API_TIMEOUT", 10)
59
-
60
- self.enabled = settings.get_bool("PROXY_ENABLED", True)
61
-
62
- if not self.enabled:
63
- self.logger.info("ProxyMiddleware 已被禁用 (PROXY_ENABLED=False)")
64
- return
65
-
66
- self.api_url = settings.get("PROXY_API_URL")
67
- if not self.api_url:
68
- raise NotConfiguredError("PROXY_API_URL 未配置,ProxyMiddleware 已禁用")
69
-
70
- self.logger.info(f"代理中间件已启用 | API: {self.api_url} | 刷新间隔: {self.refresh_interval}s")
71
-
72
- @classmethod
73
- def create_instance(cls, crawler):
74
- return cls(settings=crawler.settings, log_level=crawler.settings.get("LOG_LEVEL"))
75
-
76
- def _compile_extractor(self) -> ProxyExtractor:
77
- if callable(self.proxy_extractor):
78
- return self.proxy_extractor
79
-
80
- if isinstance(self.proxy_extractor, str):
81
- keys = self.proxy_extractor.split(".")
82
-
83
- def extract(data: Dict[str, Any]) -> Union[None, str, Dict[str, str]]:
84
- for k in keys:
85
- if isinstance(data, dict):
86
- data = data.get(k)
87
- else:
88
- return None
89
- if data is None:
90
- break
91
- return data
92
-
93
- return extract
94
-
95
- raise ValueError(f"PROXY_EXTRACTOR 必须是 str 或 callable,当前类型: {type(self.proxy_extractor)}")
96
-
97
- async def _close_session(self):
98
- if self._session:
99
- try:
100
- await self._session.close()
101
- self.logger.debug("已关闭 aiohttp session.")
102
- except Exception as e:
103
- self.logger.warning(f"关闭 aiohttp session 时出错: {e}")
104
- finally:
105
- self._session = None
106
-
107
- async def _get_session(self) -> Any: # returns aiohttp.ClientSession when aiohttp is available
108
- if aiohttp is None:
109
- raise RuntimeError("aiohttp 未安装,无法使用 ProxyMiddleware")
110
-
111
- if self._session is None or self._session.closed:
112
- if self._session and self._session.closed:
113
- self.logger.debug("现有 session 已关闭,正在创建新 session...")
114
- timeout = aiohttp.ClientTimeout(total=self.timeout)
115
- self._session = aiohttp.ClientSession(timeout=timeout)
116
- self.logger.debug("已创建新的 aiohttp session.")
117
- return self._session
118
-
119
- async def _fetch_raw_data(self) -> Optional[Dict[str, Any]]:
120
- max_retries = 2
121
- retry_count = 0
122
-
123
- while retry_count <= max_retries:
124
- session = await self._get_session()
125
- try:
126
- async with session.get(self.api_url) as resp:
127
- content_type = resp.content_type.lower()
128
- if 'application/json' not in content_type:
129
- self.logger.warning(f"代理 API 返回非 JSON 内容类型: {content_type} (URL: {self.api_url})")
130
- try:
131
- text = await resp.text()
132
- return {"__raw_text__": text.strip(), "__content_type__": content_type}
133
- except Exception as e:
134
- self.logger.error(f"读取非 JSON 响应体失败: {repr(e)}")
135
- return None
136
-
137
- if resp.status != 200:
138
- try:
139
- error_text = await resp.text()
140
- except:
141
- error_text = "<无法读取响应体>"
142
- self.logger.error(f"代理 API 状态码异常: {resp.status}, 响应体: {error_text}")
143
- if 400 <= resp.status < 500:
144
- return None
145
- return None
146
-
147
- return await resp.json()
148
-
149
- except NETWORK_EXCEPTIONS as e:
150
- retry_count += 1
151
- self.logger.warning(f"请求代理 API 失败 (尝试 {retry_count}/{max_retries + 1}): {repr(e)}")
152
- if retry_count <= max_retries:
153
- self.logger.info("正在关闭并重建 session 以进行重试...")
154
- await self._close_session()
155
- else:
156
- self.logger.error(f"请求代理 API 失败,已达到最大重试次数 ({max_retries + 1}): {repr(e)}")
157
- return None
158
-
159
- except aiohttp.ContentTypeError as e:
160
- self.logger.error(f"代理 API 响应内容类型错误: {repr(e)}")
161
- return None
162
-
163
- except Exception as e:
164
- self.logger.critical(f"请求代理 API 时发生未预期错误: {repr(e)}", exc_info=True)
165
- return None
166
-
167
- return None
168
-
169
- async def _extract_proxy(self, data: Dict[str, Any]) -> Optional[Union[str, Dict[str, str]]]:
170
- extractor = self._compile_extractor()
171
- try:
172
- result = extractor(data)
173
- if isinstance(result, str) and result.strip():
174
- return result.strip()
175
- elif isinstance(result, dict):
176
- cleaned = {k: v.strip() if isinstance(v, str) else v for k, v in result.items()}
177
- return cleaned if cleaned else None
178
- return None
179
- except Exception as e:
180
- self.logger.error(f"执行 PROXY_EXTRACTOR 时出错: {repr(e)}")
181
- return None
182
-
183
- async def _get_proxy_from_api(self) -> Optional[Union[str, Dict[str, str]]]:
184
- raw_data = await self._fetch_raw_data()
185
- if not raw_data:
186
- return None
187
-
188
- if "__raw_text__" in raw_data:
189
- text = raw_data["__raw_text__"]
190
- if text.startswith("http://") or text.startswith("https://"):
191
- return text
192
-
193
- return await self._extract_proxy(raw_data)
194
-
195
- async def _get_cached_proxy(self) -> Optional[str]:
196
- if not self.enabled:
197
- self.logger.debug("ProxyMiddleware 已禁用,跳过代理获取。")
198
- return None
199
-
200
- now = asyncio.get_event_loop().time()
201
- if self._current_proxy and (now - self._last_fetch_time) < self.refresh_interval:
202
- pass
203
- else:
204
- proxy = await self._get_proxy_from_api()
205
- if proxy:
206
- self._current_proxy = proxy
207
- self._last_fetch_time = now
208
- self.logger.debug(f"更新代理缓存: {proxy}")
209
- else:
210
- self.logger.warning("无法获取新代理,请求将直连。")
211
-
212
- return self._current_proxy
213
-
214
- @staticmethod
215
- def _is_https(request: Request) -> bool:
216
- return urlparse(request.url).scheme == "https"
217
-
218
- async def process_request(self, request: Request, spider) -> Optional[Request]:
219
- if not self.enabled:
220
- self.logger.debug(f"ProxyMiddleware 已禁用,请求将直连: {request.url}")
221
- return None
222
-
223
- if request.proxy:
224
- return None
225
-
226
- proxy = await self._get_cached_proxy()
227
- if proxy:
228
- # 处理带认证的代理URL
229
- if isinstance(proxy, str) and "@" in proxy and "://" in proxy:
230
- # 解析带认证的代理URL
231
- parsed = urlparse(proxy)
232
- if parsed.username and parsed.password:
233
- # 对于AioHttp下载器,需要特殊处理认证信息
234
- downloader_type = spider.crawler.settings.get("DOWNLOADER_TYPE", "aiohttp")
235
- if downloader_type == "aiohttp":
236
- # 将认证信息存储在meta中,由下载器处理
237
- request.meta["proxy_auth"] = {
238
- "username": parsed.username,
239
- "password": parsed.password
240
- }
241
- # 清理URL中的认证信息
242
- clean_proxy = f"{parsed.scheme}://{parsed.hostname}"
243
- if parsed.port:
244
- clean_proxy += f":{parsed.port}"
245
- request.proxy = clean_proxy
246
- else:
247
- # 其他下载器可以直接使用带认证的URL
248
- request.proxy = proxy
249
- else:
250
- request.proxy = proxy
251
- else:
252
- request.proxy = proxy
253
- self.logger.info(f"分配代理 → {proxy} | {request.url}")
254
- else:
255
- self.logger.warning(f"未获取到代理,请求直连: {request.url}")
256
-
257
- return None
258
-
259
- def process_response(self, request: Request, response: Response, spider) -> Response:
260
- proxy = request.proxy
261
- if proxy:
262
- status_code = getattr(response, 'status_code', 'N/A')
263
- self.logger.debug(f"代理成功: {proxy} | {request.url} | Status: {status_code}")
264
- return response
265
-
266
- def process_exception(self, request: Request, exception: Exception, spider) -> Optional[Request]:
267
- proxy = request.proxy
268
- if proxy:
269
- self.logger.warning(f"代理请求失败: {proxy} | {request.url} | {repr(exception)}")
270
- return None
271
-
272
- async def close(self):
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ import asyncio
4
+ import socket
5
+ from typing import Optional, Dict, Any, Callable, Union, TYPE_CHECKING, List
6
+ from urllib.parse import urlparse
7
+ import time
8
+
9
+ from crawlo import Request, Response
10
+ from crawlo.exceptions import NotConfiguredError
11
+ from crawlo.utils.log import get_logger
12
+
13
+ if TYPE_CHECKING:
14
+ import aiohttp
15
+
16
+ try:
17
+ import httpx
18
+ HTTPX_EXCEPTIONS = (httpx.NetworkError, httpx.TimeoutException, httpx.ReadError, httpx.ConnectError)
19
+ except ImportError:
20
+ HTTPX_EXCEPTIONS = ()
21
+ httpx = None
22
+
23
+ try:
24
+ import aiohttp
25
+ AIOHTTP_EXCEPTIONS = (
26
+ aiohttp.ClientError, aiohttp.ClientConnectorError, aiohttp.ClientResponseError, aiohttp.ServerTimeoutError,
27
+ aiohttp.ServerDisconnectedError)
28
+ except ImportError:
29
+ AIOHTTP_EXCEPTIONS = ()
30
+ aiohttp = None
31
+
32
+ try:
33
+ from curl_cffi import requests as cffi_requests
34
+ CURL_CFFI_EXCEPTIONS = (cffi_requests.RequestsError,)
35
+ except (ImportError, AttributeError):
36
+ CURL_CFFI_EXCEPTIONS = ()
37
+ cffi_requests = None
38
+
39
+ NETWORK_EXCEPTIONS = (
40
+ asyncio.TimeoutError,
41
+ socket.gaierror,
42
+ ConnectionError,
43
+ TimeoutError,
44
+ ) + HTTPX_EXCEPTIONS + AIOHTTP_EXCEPTIONS + CURL_CFFI_EXCEPTIONS
45
+
46
+ ProxyExtractor = Callable[[Dict[str, Any]], Union[None, str, Dict[str, str]]]
47
+
48
+ class Proxy:
49
+ """代理对象,包含代理信息和统计数据"""
50
+ def __init__(self, proxy_str: str):
51
+ self.proxy_str = proxy_str
52
+ self.success_count = 0
53
+ self.failure_count = 0
54
+ self.last_used_time = 0.0
55
+ self.is_healthy = True
56
+
57
+ @property
58
+ def success_rate(self) -> float:
59
+ """计算代理成功率"""
60
+ total = self.success_count + self.failure_count
61
+ if total == 0:
62
+ return 1.0
63
+ return self.success_count / total
64
+
65
+ def mark_success(self):
66
+ """标记代理使用成功"""
67
+ self.success_count += 1
68
+ self.last_used_time = time.time()
69
+ self.is_healthy = True
70
+
71
+ def mark_failure(self):
72
+ """标记代理使用失败"""
73
+ self.failure_count += 1
74
+ self.last_used_time = time.time()
75
+ # 如果失败率过高,标记为不健康
76
+ if self.failure_count > 3 and self.success_rate < 0.5:
77
+ self.is_healthy = False
78
+
79
+ class ProxyMiddleware:
80
+ def __init__(self, settings, log_level):
81
+ self.logger = get_logger(self.__class__.__name__, log_level)
82
+
83
+ self._session: Optional[Any] = None # aiohttp.ClientSession when aiohttp is available
84
+ # 将单个代理改为代理池
85
+ self._proxy_pool: List[Proxy] = []
86
+ self._current_proxy_index: int = 0
87
+ self._last_fetch_time: float = 0
88
+
89
+ self.proxy_extractor = settings.get("PROXY_EXTRACTOR", "proxy")
90
+ self.refresh_interval = settings.get_float("PROXY_REFRESH_INTERVAL", 60)
91
+ self.timeout = settings.get_float("PROXY_API_TIMEOUT", 10)
92
+ # 新增配置:代理池大小
93
+ self.proxy_pool_size = settings.get_int("PROXY_POOL_SIZE", 5)
94
+ # 新增配置:健康检查阈值
95
+ self.health_check_threshold = settings.get_float("PROXY_HEALTH_CHECK_THRESHOLD", 0.5)
96
+
97
+ self.enabled = settings.get_bool("PROXY_ENABLED", True)
98
+
99
+ if not self.enabled:
100
+ self.logger.info("ProxyMiddleware 已被禁用 (PROXY_ENABLED=False)")
101
+ return
102
+
103
+ self.api_url = settings.get("PROXY_API_URL")
104
+ if not self.api_url:
105
+ raise NotConfiguredError("PROXY_API_URL 未配置,ProxyMiddleware 已禁用")
106
+
107
+ self.logger.info(f"代理中间件已启用 | API: {self.api_url} | 刷新间隔: {self.refresh_interval}s | 代理池大小: {self.proxy_pool_size}")
108
+
109
+ @classmethod
110
+ def create_instance(cls, crawler):
111
+ return cls(settings=crawler.settings, log_level=crawler.settings.get("LOG_LEVEL"))
112
+
113
+ def _compile_extractor(self) -> ProxyExtractor:
114
+ if callable(self.proxy_extractor):
115
+ return self.proxy_extractor
116
+
117
+ if isinstance(self.proxy_extractor, str):
118
+ keys = self.proxy_extractor.split(".")
119
+
120
+ def extract(data: Dict[str, Any]) -> Union[None, str, Dict[str, str]]:
121
+ for k in keys:
122
+ if isinstance(data, dict):
123
+ data = data.get(k)
124
+ else:
125
+ return None
126
+ if data is None:
127
+ break
128
+ return data
129
+
130
+ return extract
131
+
132
+ raise ValueError(f"PROXY_EXTRACTOR 必须是 str 或 callable,当前类型: {type(self.proxy_extractor)}")
133
+
134
+ async def _close_session(self):
135
+ if self._session:
136
+ try:
137
+ await self._session.close()
138
+ self.logger.debug("已关闭 aiohttp session.")
139
+ except Exception as e:
140
+ self.logger.warning(f"关闭 aiohttp session 时出错: {e}")
141
+ finally:
142
+ self._session = None
143
+
144
+ async def _get_session(self) -> Any: # returns aiohttp.ClientSession when aiohttp is available
145
+ if aiohttp is None:
146
+ raise RuntimeError("aiohttp 未安装,无法使用 ProxyMiddleware")
147
+
148
+ if self._session is None or self._session.closed:
149
+ if self._session and self._session.closed:
150
+ self.logger.debug("现有 session 已关闭,正在创建新 session...")
151
+ timeout = aiohttp.ClientTimeout(total=self.timeout)
152
+ self._session = aiohttp.ClientSession(timeout=timeout)
153
+ self.logger.debug("已创建新的 aiohttp session.")
154
+ return self._session
155
+
156
+ async def _fetch_raw_data(self) -> Optional[Dict[str, Any]]:
157
+ max_retries = 2
158
+ retry_count = 0
159
+
160
+ while retry_count <= max_retries:
161
+ session = await self._get_session()
162
+ try:
163
+ async with session.get(self.api_url) as resp:
164
+ content_type = resp.content_type.lower()
165
+ if 'application/json' not in content_type:
166
+ self.logger.warning(f"代理 API 返回非 JSON 内容类型: {content_type} (URL: {self.api_url})")
167
+ try:
168
+ text = await resp.text()
169
+ return {"__raw_text__": text.strip(), "__content_type__": content_type}
170
+ except Exception as e:
171
+ self.logger.error(f"读取非 JSON 响应体失败: {repr(e)}")
172
+ return None
173
+
174
+ if resp.status != 200:
175
+ try:
176
+ error_text = await resp.text()
177
+ except:
178
+ error_text = "<无法读取响应体>"
179
+ self.logger.error(f"代理 API 状态码异常: {resp.status}, 响应体: {error_text}")
180
+ if 400 <= resp.status < 500:
181
+ return None
182
+ return None
183
+
184
+ return await resp.json()
185
+
186
+ except NETWORK_EXCEPTIONS as e:
187
+ retry_count += 1
188
+ self.logger.warning(f"请求代理 API 失败 (尝试 {retry_count}/{max_retries + 1}): {repr(e)}")
189
+ if retry_count <= max_retries:
190
+ self.logger.info("正在关闭并重建 session 以进行重试...")
191
+ await self._close_session()
192
+ else:
193
+ self.logger.error(f"请求代理 API 失败,已达到最大重试次数 ({max_retries + 1}): {repr(e)}")
194
+ return None
195
+
196
+ except aiohttp.ContentTypeError as e:
197
+ self.logger.error(f"代理 API 响应内容类型错误: {repr(e)}")
198
+ return None
199
+
200
+ except Exception as e:
201
+ self.logger.critical(f"请求代理 API 时发生未预期错误: {repr(e)}", exc_info=True)
202
+ return None
203
+
204
+ return None
205
+
206
+ async def _extract_proxy(self, data: Dict[str, Any]) -> Optional[Union[str, Dict[str, str]]]:
207
+ extractor = self._compile_extractor()
208
+ try:
209
+ result = extractor(data)
210
+ if isinstance(result, str) and result.strip():
211
+ return result.strip()
212
+ elif isinstance(result, dict):
213
+ cleaned = {k: v.strip() if isinstance(v, str) else v for k, v in result.items()}
214
+ return cleaned if cleaned else None
215
+ return None
216
+ except Exception as e:
217
+ self.logger.error(f"执行 PROXY_EXTRACTOR 时出错: {repr(e)}")
218
+ return None
219
+
220
+ async def _get_proxy_from_api(self) -> Optional[Union[str, Dict[str, str]]]:
221
+ raw_data = await self._fetch_raw_data()
222
+ if not raw_data:
223
+ return None
224
+
225
+ if "__raw_text__" in raw_data:
226
+ text = raw_data["__raw_text__"]
227
+ if text.startswith("http://") or text.startswith("https://"):
228
+ return text
229
+
230
+ return await self._extract_proxy(raw_data)
231
+
232
+ async def _update_proxy_pool(self):
233
+ """更新代理池"""
234
+ if not self.enabled:
235
+ self.logger.debug("ProxyMiddleware 已禁用,跳过代理获取。")
236
+ return
237
+
238
+ now = asyncio.get_event_loop().time()
239
+ if (now - self._last_fetch_time) < self.refresh_interval:
240
+ return
241
+
242
+ # 获取新的代理列表
243
+ proxy_data = await self._get_proxy_from_api()
244
+ if not proxy_data:
245
+ self.logger.warning("无法获取新代理,代理池将保持现状。")
246
+ return
247
+
248
+ # 解析代理数据
249
+ new_proxies = []
250
+ if isinstance(proxy_data, str):
251
+ # 单个代理
252
+ new_proxies = [proxy_data]
253
+ elif isinstance(proxy_data, dict):
254
+ # 如果是字典,尝试提取代理列表
255
+ for key, value in proxy_data.items():
256
+ if isinstance(value, str) and (value.startswith("http://") or value.startswith("https://")):
257
+ new_proxies.append(value)
258
+ elif isinstance(value, list):
259
+ # 如果值是列表,添加所有有效的代理
260
+ for item in value:
261
+ if isinstance(item, str) and (item.startswith("http://") or item.startswith("https://")):
262
+ new_proxies.append(item)
263
+
264
+ # 创建新的代理池
265
+ if new_proxies:
266
+ self._proxy_pool = [Proxy(proxy_str) for proxy_str in new_proxies[:self.proxy_pool_size]]
267
+ self._current_proxy_index = 0
268
+ self._last_fetch_time = now
269
+ self.logger.info(f"更新代理池,新增 {len(self._proxy_pool)} 个代理")
270
+ else:
271
+ self.logger.warning("未解析到有效代理,代理池将保持现状。")
272
+
273
+ async def _get_healthy_proxy(self) -> Optional[Proxy]:
274
+ """从代理池中获取一个健康的代理"""
275
+ if not self._proxy_pool:
276
+ await self._update_proxy_pool()
277
+
278
+ if not self._proxy_pool:
279
+ return None
280
+
281
+ # 查找健康的代理
282
+ healthy_proxies = [p for p in self._proxy_pool if p.is_healthy and p.success_rate >= self.health_check_threshold]
283
+
284
+ if not healthy_proxies:
285
+ # 如果没有健康的代理,尝试更新代理池
286
+ await self._update_proxy_pool()
287
+ healthy_proxies = [p for p in self._proxy_pool if p.is_healthy and p.success_rate >= self.health_check_threshold]
288
+
289
+ if not healthy_proxies:
290
+ return None
291
+
292
+ # 使用轮询方式选择代理
293
+ self._current_proxy_index = (self._current_proxy_index + 1) % len(healthy_proxies)
294
+ selected_proxy = healthy_proxies[self._current_proxy_index]
295
+ return selected_proxy
296
+
297
+ @staticmethod
298
+ def _is_https(request: Request) -> bool:
299
+ return urlparse(request.url).scheme == "https"
300
+
301
+ async def process_request(self, request: Request, spider) -> Optional[Request]:
302
+ if not self.enabled:
303
+ self.logger.debug(f"ProxyMiddleware 已禁用,请求将直连: {request.url}")
304
+ return None
305
+
306
+ if request.proxy:
307
+ return None
308
+
309
+ proxy_obj = await self._get_healthy_proxy()
310
+ if proxy_obj:
311
+ proxy = proxy_obj.proxy_str
312
+ # 处理带认证的代理URL
313
+ if isinstance(proxy, str) and "@" in proxy and "://" in proxy:
314
+ # 解析带认证的代理URL
315
+ parsed = urlparse(proxy)
316
+ if parsed.username and parsed.password:
317
+ # 对于AioHttp下载器,需要特殊处理认证信息
318
+ downloader_type = spider.crawler.settings.get("DOWNLOADER_TYPE", "aiohttp")
319
+ if downloader_type == "aiohttp":
320
+ # 将认证信息存储在meta中,由下载器处理
321
+ request.meta["proxy_auth"] = {
322
+ "username": parsed.username,
323
+ "password": parsed.password
324
+ }
325
+ # 清理URL中的认证信息
326
+ clean_proxy = f"{parsed.scheme}://{parsed.hostname}"
327
+ if parsed.port:
328
+ clean_proxy += f":{parsed.port}"
329
+ request.proxy = clean_proxy
330
+ else:
331
+ # 其他下载器可以直接使用带认证的URL
332
+ request.proxy = proxy
333
+ else:
334
+ request.proxy = proxy
335
+ else:
336
+ request.proxy = proxy
337
+
338
+ # 记录使用的代理
339
+ request.meta["_used_proxy"] = proxy_obj
340
+ self.logger.info(f"分配代理 → {proxy} | {request.url}")
341
+ else:
342
+ self.logger.warning(f"未获取到代理,请求直连: {request.url}")
343
+
344
+ return None
345
+
346
+ def process_response(self, request: Request, response: Response, spider) -> Response:
347
+ proxy_obj = request.meta.get("_used_proxy")
348
+ if proxy_obj and isinstance(proxy_obj, Proxy):
349
+ proxy_obj.mark_success()
350
+ status_code = getattr(response, 'status_code', 'N/A')
351
+ self.logger.debug(f"代理成功: {proxy_obj.proxy_str} | {request.url} | Status: {status_code}")
352
+ elif request.proxy:
353
+ status_code = getattr(response, 'status_code', 'N/A')
354
+ self.logger.debug(f"代理成功: {request.proxy} | {request.url} | Status: {status_code}")
355
+ return response
356
+
357
+ def process_exception(self, request: Request, exception: Exception, spider) -> Optional[Request]:
358
+ proxy_obj = request.meta.get("_used_proxy")
359
+ if proxy_obj and isinstance(proxy_obj, Proxy):
360
+ proxy_obj.mark_failure()
361
+ self.logger.warning(f"代理请求失败: {proxy_obj.proxy_str} | {request.url} | {repr(exception)}")
362
+ elif request.proxy:
363
+ self.logger.warning(f"代理请求失败: {request.proxy} | {request.url} | {repr(exception)}")
364
+ return None
365
+
366
+ async def close(self):
273
367
  await self._close_session()