crawlo 1.2.2__py3-none-any.whl → 1.2.4__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 (222) 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 -81
  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 +144 -142
  12. crawlo/commands/list.py +155 -155
  13. crawlo/commands/run.py +323 -292
  14. crawlo/commands/startproject.py +420 -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 +251 -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 +1110 -1027
  24. crawlo/data/__init__.py +6 -0
  25. crawlo/data/user_agents.py +108 -0
  26. crawlo/downloader/__init__.py +266 -266
  27. crawlo/downloader/aiohttp_downloader.py +220 -220
  28. crawlo/downloader/cffi_downloader.py +256 -256
  29. crawlo/downloader/httpx_downloader.py +259 -259
  30. crawlo/downloader/hybrid_downloader.py +212 -213
  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 +37 -37
  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 +280 -280
  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 -53
  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 +135 -135
  54. crawlo/middleware/offsite.py +114 -115
  55. crawlo/middleware/proxy.py +367 -366
  56. crawlo/middleware/request_ignore.py +86 -87
  57. crawlo/middleware/response_code.py +163 -164
  58. crawlo/middleware/response_filter.py +136 -137
  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 -224
  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 -316
  73. crawlo/pipelines/pipeline_manager.py +61 -61
  74. crawlo/pipelines/redis_dedup_pipeline.py +165 -167
  75. crawlo/project.py +279 -187
  76. crawlo/queue/pqueue.py +37 -37
  77. crawlo/queue/queue_manager.py +337 -337
  78. crawlo/queue/redis_priority_queue.py +298 -298
  79. crawlo/settings/__init__.py +7 -7
  80. crawlo/settings/default_settings.py +217 -226
  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 -130
  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/run.py.tmpl +47 -45
  92. crawlo/templates/project/settings.py.tmpl +350 -327
  93. crawlo/templates/project/settings_distributed.py.tmpl +160 -119
  94. crawlo/templates/project/settings_gentle.py.tmpl +133 -94
  95. crawlo/templates/project/settings_high_performance.py.tmpl +155 -151
  96. crawlo/templates/project/settings_simple.py.tmpl +108 -68
  97. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  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 -260
  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 -359
  113. crawlo/utils/env_config.py +105 -105
  114. crawlo/utils/error_handler.py +123 -125
  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 -343
  118. crawlo/utils/log.py +128 -128
  119. crawlo/utils/performance_monitor.py +285 -284
  120. crawlo/utils/queue_helper.py +175 -175
  121. crawlo/utils/redis_connection_pool.py +334 -334
  122. crawlo/utils/redis_key_validator.py +198 -199
  123. crawlo/utils/request.py +267 -267
  124. crawlo/utils/request_serializer.py +218 -219
  125. crawlo/utils/spider_loader.py +61 -62
  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.2.dist-info → crawlo-1.2.4.dist-info}/METADATA +764 -692
  130. crawlo-1.2.4.dist-info/RECORD +206 -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_validator.py +193 -193
  152. tests/test_crawlo_proxy_integration.py +172 -172
  153. tests/test_date_tools.py +123 -123
  154. tests/test_default_header_middleware.py +158 -158
  155. tests/test_double_crawlo_fix.py +207 -207
  156. tests/test_double_crawlo_fix_simple.py +124 -124
  157. tests/test_download_delay_middleware.py +221 -221
  158. tests/test_downloader_proxy_compatibility.py +268 -268
  159. tests/test_dynamic_downloaders_proxy.py +124 -124
  160. tests/test_dynamic_proxy.py +92 -92
  161. tests/test_dynamic_proxy_config.py +146 -146
  162. tests/test_dynamic_proxy_real.py +109 -109
  163. tests/test_edge_cases.py +303 -303
  164. tests/test_enhanced_error_handler.py +270 -270
  165. tests/test_env_config.py +121 -121
  166. tests/test_error_handler_compatibility.py +112 -112
  167. tests/test_final_validation.py +153 -153
  168. tests/test_framework_env_usage.py +103 -103
  169. tests/test_integration.py +356 -356
  170. tests/test_item_dedup_redis_key.py +122 -122
  171. tests/test_offsite_middleware.py +221 -221
  172. tests/test_parsel.py +29 -29
  173. tests/test_performance.py +327 -327
  174. tests/test_proxy_api.py +264 -264
  175. tests/test_proxy_health_check.py +32 -32
  176. tests/test_proxy_middleware.py +121 -121
  177. tests/test_proxy_middleware_enhanced.py +216 -216
  178. tests/test_proxy_middleware_integration.py +136 -136
  179. tests/test_proxy_providers.py +56 -56
  180. tests/test_proxy_stats.py +19 -19
  181. tests/test_proxy_strategies.py +59 -59
  182. tests/test_queue_manager_double_crawlo.py +173 -173
  183. tests/test_queue_manager_redis_key.py +176 -176
  184. tests/test_real_scenario_proxy.py +195 -195
  185. tests/test_redis_config.py +28 -28
  186. tests/test_redis_connection_pool.py +294 -294
  187. tests/test_redis_key_naming.py +181 -181
  188. tests/test_redis_key_validator.py +123 -123
  189. tests/test_redis_queue.py +224 -224
  190. tests/test_request_ignore_middleware.py +182 -182
  191. tests/test_request_serialization.py +70 -70
  192. tests/test_response_code_middleware.py +349 -349
  193. tests/test_response_filter_middleware.py +427 -427
  194. tests/test_response_improvements.py +152 -152
  195. tests/test_retry_middleware.py +241 -241
  196. tests/test_scheduler.py +241 -241
  197. tests/test_simple_response.py +61 -61
  198. tests/test_telecom_spider_redis_key.py +205 -205
  199. tests/test_template_content.py +87 -87
  200. tests/test_template_redis_key.py +134 -134
  201. tests/test_tools.py +153 -153
  202. tests/tools_example.py +257 -257
  203. crawlo-1.2.2.dist-info/RECORD +0 -220
  204. examples/aiohttp_settings.py +0 -42
  205. examples/curl_cffi_settings.py +0 -41
  206. examples/default_header_middleware_example.py +0 -107
  207. examples/default_header_spider_example.py +0 -129
  208. examples/download_delay_middleware_example.py +0 -160
  209. examples/httpx_settings.py +0 -42
  210. examples/multi_downloader_proxy_example.py +0 -81
  211. examples/offsite_middleware_example.py +0 -55
  212. examples/offsite_spider_example.py +0 -107
  213. examples/proxy_spider_example.py +0 -166
  214. examples/request_ignore_middleware_example.py +0 -51
  215. examples/request_ignore_spider_example.py +0 -99
  216. examples/response_code_middleware_example.py +0 -52
  217. examples/response_filter_middleware_example.py +0 -67
  218. examples/tong_hua_shun_settings.py +0 -62
  219. examples/tong_hua_shun_spider.py +0 -170
  220. {crawlo-1.2.2.dist-info → crawlo-1.2.4.dist-info}/WHEEL +0 -0
  221. {crawlo-1.2.2.dist-info → crawlo-1.2.4.dist-info}/entry_points.txt +0 -0
  222. {crawlo-1.2.2.dist-info → crawlo-1.2.4.dist-info}/top_level.txt +0 -0
@@ -1,367 +1,368 @@
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):
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):
367
368
  await self._close_session()