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