crawlo 1.4.7__py3-none-any.whl → 1.4.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 (348) hide show
  1. crawlo/__init__.py +90 -90
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +75 -75
  4. crawlo/commands/__init__.py +14 -14
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +186 -186
  7. crawlo/commands/help.py +140 -140
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +379 -379
  10. crawlo/commands/startproject.py +460 -460
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +320 -320
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +52 -52
  16. crawlo/core/engine.py +451 -451
  17. crawlo/core/processor.py +47 -47
  18. crawlo/core/scheduler.py +290 -290
  19. crawlo/crawler.py +698 -698
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +280 -280
  23. crawlo/downloader/aiohttp_downloader.py +233 -233
  24. crawlo/downloader/cffi_downloader.py +250 -250
  25. crawlo/downloader/httpx_downloader.py +265 -265
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +425 -425
  28. crawlo/downloader/selenium_downloader.py +486 -486
  29. crawlo/event.py +45 -45
  30. crawlo/exceptions.py +214 -214
  31. crawlo/extension/__init__.py +64 -64
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +94 -94
  34. crawlo/extension/log_stats.py +70 -70
  35. crawlo/extension/logging_extension.py +53 -53
  36. crawlo/extension/memory_monitor.py +104 -104
  37. crawlo/extension/performance_profiler.py +133 -133
  38. crawlo/extension/request_recorder.py +107 -107
  39. crawlo/factories/__init__.py +27 -27
  40. crawlo/factories/base.py +68 -68
  41. crawlo/factories/crawler.py +104 -104
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/factories/utils.py +134 -134
  44. crawlo/filters/__init__.py +170 -170
  45. crawlo/filters/aioredis_filter.py +347 -347
  46. crawlo/filters/memory_filter.py +261 -261
  47. crawlo/framework.py +306 -306
  48. crawlo/initialization/__init__.py +44 -44
  49. crawlo/initialization/built_in.py +391 -391
  50. crawlo/initialization/context.py +141 -141
  51. crawlo/initialization/core.py +240 -240
  52. crawlo/initialization/phases.py +229 -229
  53. crawlo/initialization/registry.py +143 -143
  54. crawlo/initialization/utils.py +48 -48
  55. crawlo/interfaces.py +23 -23
  56. crawlo/items/__init__.py +23 -23
  57. crawlo/items/base.py +23 -23
  58. crawlo/items/fields.py +52 -52
  59. crawlo/items/items.py +104 -104
  60. crawlo/logging/__init__.py +42 -42
  61. crawlo/logging/config.py +280 -276
  62. crawlo/logging/factory.py +175 -175
  63. crawlo/logging/manager.py +104 -104
  64. crawlo/middleware/__init__.py +87 -87
  65. crawlo/middleware/default_header.py +132 -132
  66. crawlo/middleware/download_delay.py +104 -104
  67. crawlo/middleware/middleware_manager.py +142 -142
  68. crawlo/middleware/offsite.py +123 -123
  69. crawlo/middleware/proxy.py +209 -209
  70. crawlo/middleware/request_ignore.py +86 -86
  71. crawlo/middleware/response_code.py +150 -150
  72. crawlo/middleware/response_filter.py +136 -136
  73. crawlo/middleware/retry.py +124 -124
  74. crawlo/mode_manager.py +287 -287
  75. crawlo/network/__init__.py +21 -21
  76. crawlo/network/request.py +408 -376
  77. crawlo/network/response.py +598 -569
  78. crawlo/pipelines/__init__.py +52 -52
  79. crawlo/pipelines/base_pipeline.py +452 -452
  80. crawlo/pipelines/bloom_dedup_pipeline.py +145 -146
  81. crawlo/pipelines/console_pipeline.py +39 -39
  82. crawlo/pipelines/csv_pipeline.py +316 -316
  83. crawlo/pipelines/database_dedup_pipeline.py +196 -197
  84. crawlo/pipelines/json_pipeline.py +218 -218
  85. crawlo/pipelines/memory_dedup_pipeline.py +104 -105
  86. crawlo/pipelines/mongo_pipeline.py +140 -139
  87. crawlo/pipelines/mysql_pipeline.py +468 -469
  88. crawlo/pipelines/pipeline_manager.py +100 -100
  89. crawlo/pipelines/redis_dedup_pipeline.py +155 -155
  90. crawlo/project.py +347 -347
  91. crawlo/queue/__init__.py +9 -9
  92. crawlo/queue/pqueue.py +38 -38
  93. crawlo/queue/queue_manager.py +591 -591
  94. crawlo/queue/redis_priority_queue.py +518 -518
  95. crawlo/settings/__init__.py +7 -7
  96. crawlo/settings/default_settings.py +287 -284
  97. crawlo/settings/setting_manager.py +219 -219
  98. crawlo/spider/__init__.py +658 -657
  99. crawlo/stats_collector.py +81 -81
  100. crawlo/subscriber.py +129 -129
  101. crawlo/task_manager.py +138 -138
  102. crawlo/templates/crawlo.cfg.tmpl +10 -10
  103. crawlo/templates/project/__init__.py.tmpl +1 -1
  104. crawlo/templates/project/items.py.tmpl +13 -13
  105. crawlo/templates/project/middlewares.py.tmpl +38 -38
  106. crawlo/templates/project/pipelines.py.tmpl +35 -35
  107. crawlo/templates/project/settings.py.tmpl +113 -109
  108. crawlo/templates/project/settings_distributed.py.tmpl +160 -156
  109. crawlo/templates/project/settings_gentle.py.tmpl +174 -170
  110. crawlo/templates/project/settings_high_performance.py.tmpl +175 -171
  111. crawlo/templates/project/settings_minimal.py.tmpl +102 -98
  112. crawlo/templates/project/settings_simple.py.tmpl +172 -168
  113. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  114. crawlo/templates/run.py.tmpl +23 -23
  115. crawlo/templates/spider/spider.py.tmpl +32 -32
  116. crawlo/templates/spiders_init.py.tmpl +4 -4
  117. crawlo/tools/__init__.py +86 -86
  118. crawlo/tools/date_tools.py +289 -289
  119. crawlo/tools/distributed_coordinator.py +384 -384
  120. crawlo/tools/scenario_adapter.py +262 -262
  121. crawlo/tools/text_cleaner.py +232 -232
  122. crawlo/utils/__init__.py +74 -50
  123. crawlo/utils/batch_processor.py +276 -276
  124. crawlo/utils/config_manager.py +442 -442
  125. crawlo/utils/controlled_spider_mixin.py +439 -439
  126. crawlo/utils/db_helper.py +250 -250
  127. crawlo/utils/encoding_helper.py +190 -0
  128. crawlo/utils/error_handler.py +410 -410
  129. crawlo/utils/fingerprint.py +121 -121
  130. crawlo/utils/func_tools.py +82 -82
  131. crawlo/utils/large_scale_helper.py +344 -344
  132. crawlo/utils/leak_detector.py +335 -335
  133. crawlo/utils/misc.py +81 -81
  134. crawlo/utils/mongo_connection_pool.py +157 -157
  135. crawlo/utils/mysql_connection_pool.py +197 -197
  136. crawlo/utils/performance_monitor.py +285 -285
  137. crawlo/utils/queue_helper.py +175 -175
  138. crawlo/utils/redis_checker.py +90 -90
  139. crawlo/utils/redis_connection_pool.py +578 -578
  140. crawlo/utils/redis_key_validator.py +198 -198
  141. crawlo/utils/request.py +278 -278
  142. crawlo/utils/request_serializer.py +225 -225
  143. crawlo/utils/resource_manager.py +337 -337
  144. crawlo/utils/response_helper.py +113 -0
  145. crawlo/utils/selector_helper.py +138 -137
  146. crawlo/utils/singleton.py +69 -69
  147. crawlo/utils/spider_loader.py +201 -201
  148. crawlo/utils/text_helper.py +94 -94
  149. {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/METADATA +831 -689
  150. crawlo-1.4.8.dist-info/RECORD +347 -0
  151. examples/__init__.py +7 -7
  152. tests/__init__.py +7 -7
  153. tests/advanced_tools_example.py +217 -217
  154. tests/authenticated_proxy_example.py +110 -110
  155. tests/baidu_performance_test.py +108 -108
  156. tests/baidu_test.py +59 -59
  157. tests/bug_check_test.py +250 -250
  158. tests/cleaners_example.py +160 -160
  159. tests/comprehensive_framework_test.py +212 -212
  160. tests/comprehensive_test.py +81 -81
  161. tests/comprehensive_testing_summary.md +186 -186
  162. tests/config_validation_demo.py +142 -142
  163. tests/controlled_spider_example.py +205 -205
  164. tests/date_tools_example.py +180 -180
  165. tests/debug_configure.py +69 -69
  166. tests/debug_framework_logger.py +84 -84
  167. tests/debug_log_config.py +126 -126
  168. tests/debug_log_levels.py +63 -63
  169. tests/debug_pipelines.py +66 -66
  170. tests/detailed_log_test.py +233 -233
  171. tests/direct_selector_helper_test.py +96 -96
  172. tests/distributed_dedup_test.py +467 -467
  173. tests/distributed_test.py +66 -66
  174. tests/distributed_test_debug.py +76 -76
  175. tests/dynamic_loading_example.py +523 -523
  176. tests/dynamic_loading_test.py +104 -104
  177. tests/error_handling_example.py +171 -171
  178. tests/explain_mysql_update_behavior.py +76 -76
  179. tests/final_comprehensive_test.py +151 -151
  180. tests/final_log_test.py +260 -260
  181. tests/final_validation_test.py +182 -182
  182. tests/fix_log_test.py +142 -142
  183. tests/framework_performance_test.py +202 -202
  184. tests/log_buffering_test.py +111 -111
  185. tests/log_generation_timing_test.py +153 -153
  186. tests/monitor_redis_dedup.sh +72 -72
  187. tests/ofweek_scrapy/ofweek_scrapy/items.py +12 -12
  188. tests/ofweek_scrapy/ofweek_scrapy/middlewares.py +100 -100
  189. tests/ofweek_scrapy/ofweek_scrapy/pipelines.py +13 -13
  190. tests/ofweek_scrapy/ofweek_scrapy/settings.py +84 -84
  191. tests/ofweek_scrapy/ofweek_scrapy/spiders/__init__.py +4 -4
  192. tests/ofweek_scrapy/scrapy.cfg +11 -11
  193. tests/optimized_performance_test.py +211 -211
  194. tests/performance_comparison.py +244 -244
  195. tests/queue_blocking_test.py +113 -113
  196. tests/queue_test.py +89 -89
  197. tests/redis_key_validation_demo.py +130 -130
  198. tests/request_params_example.py +150 -150
  199. tests/response_improvements_example.py +144 -144
  200. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  201. tests/scrapy_comparison/scrapy_test.py +133 -133
  202. tests/simple_cli_test.py +54 -54
  203. tests/simple_command_test.py +119 -119
  204. tests/simple_crawlo_test.py +126 -126
  205. tests/simple_follow_test.py +38 -38
  206. tests/simple_log_test2.py +137 -137
  207. tests/simple_optimization_test.py +128 -128
  208. tests/simple_queue_type_test.py +41 -41
  209. tests/simple_response_selector_test.py +94 -94
  210. tests/simple_selector_helper_test.py +154 -154
  211. tests/simple_selector_test.py +207 -207
  212. tests/simple_spider_test.py +49 -49
  213. tests/simple_url_test.py +73 -73
  214. tests/simulate_mysql_update_test.py +139 -139
  215. tests/spider_log_timing_test.py +177 -177
  216. tests/test_advanced_tools.py +148 -148
  217. tests/test_all_commands.py +230 -230
  218. tests/test_all_pipeline_fingerprints.py +133 -133
  219. tests/test_all_redis_key_configs.py +145 -145
  220. tests/test_asyncmy_usage.py +56 -56
  221. tests/test_batch_processor.py +178 -178
  222. tests/test_cleaners.py +54 -54
  223. tests/test_cli_arguments.py +118 -118
  224. tests/test_component_factory.py +174 -174
  225. tests/test_config_consistency.py +80 -80
  226. tests/test_config_merge.py +152 -152
  227. tests/test_config_validator.py +182 -182
  228. tests/test_controlled_spider_mixin.py +79 -79
  229. tests/test_crawler_process_import.py +38 -38
  230. tests/test_crawler_process_spider_modules.py +47 -47
  231. tests/test_crawlo_proxy_integration.py +114 -114
  232. tests/test_date_tools.py +123 -123
  233. tests/test_dedup_fix.py +220 -220
  234. tests/test_dedup_pipeline_consistency.py +124 -124
  235. tests/test_default_header_middleware.py +313 -313
  236. tests/test_distributed.py +65 -65
  237. tests/test_double_crawlo_fix.py +204 -204
  238. tests/test_double_crawlo_fix_simple.py +124 -124
  239. tests/test_download_delay_middleware.py +221 -221
  240. tests/test_downloader_proxy_compatibility.py +272 -272
  241. tests/test_edge_cases.py +305 -305
  242. tests/test_encoding_core.py +56 -56
  243. tests/test_encoding_detection.py +126 -126
  244. tests/test_enhanced_error_handler.py +270 -270
  245. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  246. tests/test_error_handler_compatibility.py +112 -112
  247. tests/test_factories.py +252 -252
  248. tests/test_factory_compatibility.py +196 -196
  249. tests/test_final_validation.py +153 -153
  250. tests/test_fingerprint_consistency.py +135 -135
  251. tests/test_fingerprint_simple.py +51 -51
  252. tests/test_get_component_logger.py +83 -83
  253. tests/test_hash_performance.py +99 -99
  254. tests/test_integration.py +169 -169
  255. tests/test_item_dedup_redis_key.py +122 -122
  256. tests/test_large_scale_helper.py +235 -235
  257. tests/test_logging_enhancements.py +374 -374
  258. tests/test_logging_final.py +184 -184
  259. tests/test_logging_integration.py +312 -312
  260. tests/test_logging_system.py +282 -282
  261. tests/test_middleware_debug.py +141 -141
  262. tests/test_mode_consistency.py +51 -51
  263. tests/test_multi_directory.py +67 -67
  264. tests/test_multiple_spider_modules.py +80 -80
  265. tests/test_mysql_pipeline_config.py +164 -164
  266. tests/test_mysql_pipeline_error.py +98 -98
  267. tests/test_mysql_pipeline_init_log.py +82 -82
  268. tests/test_mysql_pipeline_integration.py +132 -132
  269. tests/test_mysql_pipeline_refactor.py +143 -143
  270. tests/test_mysql_pipeline_refactor_simple.py +85 -85
  271. tests/test_mysql_pipeline_robustness.py +195 -195
  272. tests/test_mysql_pipeline_types.py +88 -88
  273. tests/test_mysql_update_columns.py +93 -93
  274. tests/test_offsite_middleware.py +244 -244
  275. tests/test_offsite_middleware_simple.py +203 -203
  276. tests/test_optimized_selector_naming.py +100 -100
  277. tests/test_parsel.py +29 -29
  278. tests/test_performance.py +327 -327
  279. tests/test_performance_monitor.py +115 -115
  280. tests/test_pipeline_fingerprint_consistency.py +86 -86
  281. tests/test_priority_behavior.py +211 -211
  282. tests/test_priority_consistency.py +151 -151
  283. tests/test_priority_consistency_fixed.py +249 -249
  284. tests/test_proxy_health_check.py +32 -32
  285. tests/test_proxy_middleware.py +217 -217
  286. tests/test_proxy_middleware_enhanced.py +212 -212
  287. tests/test_proxy_middleware_integration.py +142 -142
  288. tests/test_proxy_middleware_refactored.py +207 -207
  289. tests/test_proxy_only.py +83 -83
  290. tests/test_proxy_providers.py +56 -56
  291. tests/test_proxy_stats.py +19 -19
  292. tests/test_proxy_strategies.py +59 -59
  293. tests/test_proxy_with_downloader.py +152 -152
  294. tests/test_queue_empty_check.py +41 -41
  295. tests/test_queue_manager_double_crawlo.py +173 -173
  296. tests/test_queue_manager_redis_key.py +179 -179
  297. tests/test_queue_naming.py +154 -154
  298. tests/test_queue_type.py +106 -106
  299. tests/test_queue_type_redis_config_consistency.py +130 -130
  300. tests/test_random_headers_default.py +322 -322
  301. tests/test_random_headers_necessity.py +308 -308
  302. tests/test_random_user_agent.py +72 -72
  303. tests/test_redis_config.py +28 -28
  304. tests/test_redis_connection_pool.py +294 -294
  305. tests/test_redis_key_naming.py +181 -181
  306. tests/test_redis_key_validator.py +123 -123
  307. tests/test_redis_queue.py +224 -224
  308. tests/test_redis_queue_name_fix.py +175 -175
  309. tests/test_redis_queue_type_fallback.py +129 -129
  310. tests/test_request_ignore_middleware.py +182 -182
  311. tests/test_request_params.py +111 -111
  312. tests/test_request_serialization.py +70 -70
  313. tests/test_response_code_middleware.py +349 -349
  314. tests/test_response_filter_middleware.py +427 -427
  315. tests/test_response_follow.py +104 -104
  316. tests/test_response_improvements.py +152 -152
  317. tests/test_response_selector_methods.py +92 -92
  318. tests/test_response_url_methods.py +70 -70
  319. tests/test_response_urljoin.py +86 -86
  320. tests/test_retry_middleware.py +333 -333
  321. tests/test_retry_middleware_realistic.py +273 -273
  322. tests/test_scheduler.py +252 -252
  323. tests/test_scheduler_config_update.py +133 -133
  324. tests/test_scrapy_style_encoding.py +112 -112
  325. tests/test_selector_helper.py +100 -100
  326. tests/test_selector_optimizations.py +146 -146
  327. tests/test_simple_response.py +61 -61
  328. tests/test_spider_loader.py +49 -49
  329. tests/test_spider_loader_comprehensive.py +69 -69
  330. tests/test_spider_modules.py +84 -84
  331. tests/test_spiders/test_spider.py +9 -9
  332. tests/test_telecom_spider_redis_key.py +205 -205
  333. tests/test_template_content.py +87 -87
  334. tests/test_template_redis_key.py +134 -134
  335. tests/test_tools.py +159 -159
  336. tests/test_user_agent_randomness.py +176 -176
  337. tests/test_user_agents.py +96 -96
  338. tests/untested_features_report.md +138 -138
  339. tests/verify_debug.py +51 -51
  340. tests/verify_distributed.py +117 -117
  341. tests/verify_log_fix.py +111 -111
  342. tests/verify_mysql_warnings.py +109 -109
  343. crawlo/utils/log.py +0 -80
  344. crawlo/utils/url_utils.py +0 -40
  345. crawlo-1.4.7.dist-info/RECORD +0 -347
  346. {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/WHEEL +0 -0
  347. {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/entry_points.txt +0 -0
  348. {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/top_level.txt +0 -0
@@ -1,570 +1,599 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- """
4
- HTTP Response 封装模块
5
- =====================
6
- 提供功能丰富的HTTP响应封装,支持:
7
- - 智能编码检测和解码
8
- - XPath/CSS 选择器
9
- - JSON 解析和缓存
10
- - 正则表达式支持
11
- - Cookie 处理
12
- """
13
- import re
14
- from http.cookies import SimpleCookie
15
- from typing import Dict, Any, List, Optional, Tuple
16
- from urllib.parse import urljoin as _urljoin, urlparse as _urlparse, urlsplit as _urlsplit, parse_qs as _parse_qs, \
17
- urlencode as _urlencode, quote as _quote, unquote as _unquote, urldefrag as _urldefrag
18
-
19
- import ujson
20
- from parsel import Selector, SelectorList
21
-
22
- # 尝试导入 w3lib 编码检测函数
23
- try:
24
- from w3lib.encoding import (
25
- html_body_declared_encoding,
26
- html_to_unicode,
27
- http_content_type_encoding,
28
- read_bom,
29
- resolve_encoding,
30
- )
31
- W3LIB_AVAILABLE = True
32
- except ImportError:
33
- W3LIB_AVAILABLE = False
34
-
35
- from crawlo.exceptions import DecodeError
36
- from crawlo.utils import (
37
- extract_text,
38
- extract_texts,
39
- extract_attr,
40
- extract_attrs,
41
- is_xpath
42
- )
43
-
44
-
45
- def memoize_method_noargs(func):
46
- """
47
- 装饰器,用于缓存无参数方法的结果
48
- """
49
- cache_attr = f'_cache_{func.__name__}'
50
-
51
- def wrapper(self):
52
- if not hasattr(self, cache_attr):
53
- setattr(self, cache_attr, func(self))
54
- return getattr(self, cache_attr)
55
-
56
- return wrapper
57
-
58
-
59
- class Response:
60
- """
61
- HTTP响应的封装,提供数据解析的便捷方法。
62
-
63
- 功能特性:
64
- - 智能编码检测和缓存
65
- - 懒加载 Selector 实例
66
- - JSON 解析和缓存
67
- - 多类型数据提取
68
- """
69
-
70
- # 默认编码
71
- _DEFAULT_ENCODING = "ascii"
72
-
73
- def __init__(
74
- self,
75
- url: str,
76
- *,
77
- headers: Dict[str, Any] = None,
78
- body: bytes = b"",
79
- method: str = 'GET',
80
- request: 'Request' = None, # 使用字符串注解避免循环导入
81
- status_code: int = 200,
82
- ):
83
- # 基本属性
84
- self.url = url
85
- self.headers = headers or {}
86
- self.body = body
87
- self.method = method.upper()
88
- self.request = request
89
- self.status_code = status_code
90
-
91
- # 编码处理
92
- self.encoding = self._determine_encoding()
93
-
94
- # 缓存属性
95
- self._text_cache = None
96
- self._json_cache = None
97
- self._selector_instance = None
98
-
99
- # 状态标记
100
- self._is_success = 200 <= status_code < 300
101
- self._is_redirect = 300 <= status_code < 400
102
- self._is_client_error = 400 <= status_code < 500
103
- self._is_server_error = status_code >= 500
104
-
105
- # ==================== 编码检测相关方法 ====================
106
-
107
- def _determine_encoding(self) -> str:
108
- """
109
- 智能检测响应编码
110
-
111
- 编码检测优先级:
112
- 1. Request 中指定的编码
113
- 2. BOM 字节顺序标记
114
- 3. HTTP Content-Type 头部
115
- 4. HTML meta 标签声明
116
- 5. 内容自动检测
117
- 6. 默认编码 (utf-8)
118
- """
119
- # 1. 优先使用声明的编码
120
- declared_encoding = self._declared_encoding()
121
- if declared_encoding:
122
- return declared_encoding
123
-
124
- # 2. 使用 w3lib 进行编码检测(如果可用)
125
- if W3LIB_AVAILABLE:
126
- return self._body_inferred_encoding()
127
- else:
128
- # 如果 w3lib 不可用,使用原有的检测逻辑
129
- # Content-Type 头中检测
130
- content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
131
- if content_type:
132
- charset_match = re.search(r"charset=([\w-]+)", content_type, re.I)
133
- if charset_match:
134
- return charset_match.group(1).lower()
135
-
136
- # 从 HTML meta 标签中检测(仅对HTML内容)
137
- if b'<html' in self.body[:1024].lower():
138
- # 查找 <meta charset="xxx"> 或 <meta http-equiv="Content-Type" content="...charset=xxx">
139
- html_start = self.body[:4096] # 只检查前4KB
140
- try:
141
- html_text = html_start.decode('ascii', errors='ignore')
142
- # <meta charset="utf-8">
143
- charset_match = re.search(r'<meta[^>]+charset=["\']?([\w-]+)', html_text, re.I)
144
- if charset_match:
145
- return charset_match.group(1).lower()
146
-
147
- # <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
148
- content_match = re.search(r'<meta[^>]+content=["\'][^"\'>]*charset=([\w-]+)', html_text, re.I)
149
- if content_match:
150
- return content_match.group(1).lower()
151
- except Exception:
152
- pass
153
-
154
- # 3. 默认使用 utf-8
155
- return 'utf-8'
156
-
157
- def _declared_encoding(self) -> Optional[str]:
158
- """
159
- 获取声明的编码
160
- 优先级:Request编码 > BOM > HTTP头部 > HTML meta标签
161
- """
162
- # 1. Request 中指定的编码
163
- if self.request and getattr(self.request, 'encoding', None):
164
- return self.request.encoding
165
-
166
- # 2. BOM 字节顺序标记
167
- bom_encoding = self._bom_encoding()
168
- if bom_encoding:
169
- return bom_encoding
170
-
171
- # 3. HTTP Content-Type 头部
172
- headers_encoding = self._headers_encoding()
173
- if headers_encoding:
174
- return headers_encoding
175
-
176
- # 4. HTML meta 标签声明编码
177
- body_declared_encoding = self._body_declared_encoding()
178
- if body_declared_encoding:
179
- return body_declared_encoding
180
-
181
- return None
182
-
183
- @memoize_method_noargs
184
- def _bom_encoding(self) -> Optional[str]:
185
- """BOM 字节顺序标记编码检测"""
186
- if not W3LIB_AVAILABLE:
187
- return None
188
- return read_bom(self.body)[0]
189
-
190
- @memoize_method_noargs
191
- def _headers_encoding(self) -> Optional[str]:
192
- """HTTP Content-Type 头部编码检测"""
193
- if not W3LIB_AVAILABLE:
194
- return None
195
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
196
- if isinstance(content_type, bytes):
197
- content_type = content_type.decode('latin-1')
198
- return http_content_type_encoding(content_type)
199
-
200
- @memoize_method_noargs
201
- def _body_declared_encoding(self) -> Optional[str]:
202
- """HTML meta 标签声明编码检测"""
203
- if not W3LIB_AVAILABLE:
204
- return None
205
- return html_body_declared_encoding(self.body)
206
-
207
- @memoize_method_noargs
208
- def _body_inferred_encoding(self) -> str:
209
- """内容自动检测编码"""
210
- if not W3LIB_AVAILABLE:
211
- # 回退到简单的自动检测
212
- for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
213
- try:
214
- self.body.decode(enc)
215
- except UnicodeError:
216
- continue
217
- return enc
218
- return self._DEFAULT_ENCODING
219
-
220
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
221
- if isinstance(content_type, bytes):
222
- content_type = content_type.decode('latin-1')
223
-
224
- # 使用 w3lib 的 html_to_unicode 函数进行编码检测
225
- benc, _ = html_to_unicode(
226
- content_type,
227
- self.body,
228
- auto_detect_fun=self._auto_detect_fun,
229
- default_encoding=self._DEFAULT_ENCODING,
230
- )
231
- return benc
232
-
233
- def _auto_detect_fun(self, text: bytes) -> Optional[str]:
234
- """自动检测编码的回调函数"""
235
- if not W3LIB_AVAILABLE:
236
- return None
237
- for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
238
- try:
239
- text.decode(enc)
240
- except UnicodeError:
241
- continue
242
- return resolve_encoding(enc)
243
- return None
244
-
245
- @property
246
- def text(self) -> str:
247
- """将响应体(body)以正确的编码解码为字符串,并缓存结果。"""
248
- if self._text_cache is not None:
249
- return self._text_cache
250
-
251
- if not self.body:
252
- self._text_cache = ""
253
- return self._text_cache
254
-
255
- # 如果可用,使用 w3lib 进行更准确的解码
256
- if W3LIB_AVAILABLE:
257
- try:
258
- content_type = self.headers.get(b"Content-Type", b"") or self.headers.get("content-type", b"")
259
- if isinstance(content_type, bytes):
260
- content_type = content_type.decode('latin-1')
261
-
262
- # 使用 w3lib 的 html_to_unicode 函数
263
- _, self._text_cache = html_to_unicode(
264
- content_type,
265
- self.body,
266
- default_encoding=self.encoding
267
- )
268
- return self._text_cache
269
- except Exception:
270
- # 如果 w3lib 解码失败,回退到原有方法
271
- pass
272
-
273
- # 尝试多种编码
274
- encodings_to_try = [self.encoding]
275
- if self.encoding != 'utf-8':
276
- encodings_to_try.append('utf-8')
277
- if 'gbk' not in encodings_to_try:
278
- encodings_to_try.append('gbk')
279
- if 'gb2312' not in encodings_to_try:
280
- encodings_to_try.append('gb2312')
281
- encodings_to_try.append('latin1') # 最后的回退选项
282
-
283
- for encoding in encodings_to_try:
284
- if not encoding:
285
- continue
286
- try:
287
- self._text_cache = self.body.decode(encoding)
288
- return self._text_cache
289
- except (UnicodeDecodeError, LookupError):
290
- continue
291
-
292
- # 所有编码都失败,使用容错解码
293
- try:
294
- self._text_cache = self.body.decode('utf-8', errors='replace')
295
- return self._text_cache
296
- except Exception as e:
297
- raise DecodeError(f"Failed to decode response from {self.url}: {e}")
298
-
299
- # ==================== 状态检查相关属性 ====================
300
-
301
- @property
302
- def is_success(self) -> bool:
303
- """检查响应是否成功 (2xx)"""
304
- return self._is_success
305
-
306
- @property
307
- def is_redirect(self) -> bool:
308
- """检查响应是否为重定向 (3xx)"""
309
- return self._is_redirect
310
-
311
- @property
312
- def is_client_error(self) -> bool:
313
- """检查响应是否为客户端错误 (4xx)"""
314
- return self._is_client_error
315
-
316
- @property
317
- def is_server_error(self) -> bool:
318
- """检查响应是否为服务器错误 (5xx)"""
319
- return self._is_server_error
320
-
321
- # ==================== 响应头相关属性 ====================
322
-
323
- @property
324
- def content_type(self) -> str:
325
- """获取响应的 Content-Type"""
326
- return self.headers.get('content-type', '') or self.headers.get('Content-Type', '')
327
-
328
- @property
329
- def content_length(self) -> Optional[int]:
330
- """获取响应的 Content-Length"""
331
- length = self.headers.get('content-length') or self.headers.get('Content-Length')
332
- return int(length) if length else None
333
-
334
- # ==================== JSON处理方法 ====================
335
-
336
- def json(self, default: Any = None) -> Any:
337
- """将响应文本解析为 JSON 对象。"""
338
- if self._json_cache is not None:
339
- return self._json_cache
340
-
341
- try:
342
- self._json_cache = ujson.loads(self.text)
343
- return self._json_cache
344
- except (ujson.JSONDecodeError, ValueError) as e:
345
- if default is not None:
346
- return default
347
- raise DecodeError(f"Failed to parse JSON from {self.url}: {e}")
348
-
349
- # ==================== URL处理方法 ====================
350
-
351
- def urljoin(self, url: str) -> str:
352
- """拼接 URL,自动处理相对路径。"""
353
- return _urljoin(self.url, url)
354
-
355
- # ==================== 选择器相关方法 ====================
356
-
357
- @property
358
- def _selector(self) -> Selector:
359
- """懒加载 Selector 实例"""
360
- if self._selector_instance is None:
361
- self._selector_instance = Selector(self.text)
362
- return self._selector_instance
363
-
364
- def xpath(self, query: str) -> SelectorList:
365
- """使用 XPath 选择器查询文档。"""
366
- return self._selector.xpath(query)
367
-
368
- def css(self, query: str) -> SelectorList:
369
- """使用 CSS 选择器查询文档。"""
370
- return self._selector.css(query)
371
-
372
- def _is_xpath(self, query: str) -> bool:
373
- """判断查询语句是否为XPath"""
374
- return is_xpath(query)
375
-
376
- def _extract_text(self, elements: SelectorList, join_str: str = " ") -> str:
377
- """
378
- 从元素列表中提取文本并拼接
379
-
380
- :param elements: SelectorList元素列表
381
- :param join_str: 文本拼接分隔符
382
- :return: 拼接后的文本
383
- """
384
- return extract_text(elements, join_str)
385
-
386
- def extract_text(self, xpath_or_css: str, join_str: str = " ", default: str = '') -> str:
387
- """
388
- 提取单个元素的文本内容,支持CSS和XPath选择器
389
-
390
- 参数:
391
- xpath_or_css: XPath或CSS选择器
392
- join_str: 文本拼接分隔符(默认为空格)
393
- default: 默认返回值,当未找到元素时返回
394
-
395
- 返回:
396
- 拼接后的纯文本字符串
397
-
398
- 示例:
399
- # 使用CSS选择器
400
- title = response.extract_text('title')
401
- content = response.extract_text('.content p', join_str=' ')
402
-
403
- # 使用XPath选择器
404
- title = response.extract_text('//title')
405
- content = response.extract_text('//div[@class="content"]//p', join_str=' ')
406
- """
407
- try:
408
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
409
- if not elements:
410
- return default
411
- return self._extract_text(elements, join_str)
412
- except Exception:
413
- return default
414
-
415
- def extract_texts(self, xpath_or_css: str, join_str: str = " ", default: List[str] = None) -> List[str]:
416
- """
417
- 提取多个元素的文本内容列表,支持CSS和XPath选择器
418
-
419
- 参数:
420
- xpath_or_css: XPath或CSS选择器
421
- join_str: 单个节点内文本拼接分隔符
422
- default: 默认返回值,当未找到元素时返回
423
-
424
- 返回:
425
- 纯文本列表(每个元素对应一个节点的文本)
426
-
427
- 示例:
428
- # 使用CSS选择器
429
- paragraphs = response.extract_texts('.content p')
430
- titles = response.extract_texts('h1, h2, h3')
431
-
432
- # 使用XPath选择器
433
- paragraphs = response.extract_texts('//div[@class="content"]//p')
434
- titles = response.extract_texts('//h1 | //h2 | //h3')
435
- """
436
- if default is None:
437
- default = []
438
-
439
- try:
440
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
441
- if not elements:
442
- return default
443
-
444
- result = extract_texts(elements, join_str)
445
- return result if result else default
446
- except Exception:
447
- return default
448
-
449
- def extract_attr(self, xpath_or_css: str, attr_name: str, default: Any = None) -> Any:
450
- """
451
- 提取单个元素的属性值,支持CSS和XPath选择器
452
-
453
- 参数:
454
- xpath_or_css: XPath或CSS选择器
455
- attr_name: 属性名称
456
- default: 默认返回值
457
-
458
- 返回:
459
- 属性值或默认值
460
-
461
- 示例:
462
- # 使用CSS选择器
463
- link_href = response.extract_attr('a', 'href')
464
- img_src = response.extract_attr('.image', 'src')
465
-
466
- # 使用XPath选择器
467
- link_href = response.extract_attr('//a[@class="link"]', 'href')
468
- img_src = response.extract_attr('//img[@alt="example"]', 'src')
469
- """
470
- try:
471
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
472
- if not elements:
473
- return default
474
- return extract_attr(elements, attr_name, default)
475
- except Exception:
476
- return default
477
-
478
- def extract_attrs(self, xpath_or_css: str, attr_name: str, default: List[Any] = None) -> List[Any]:
479
- """
480
- 提取多个元素的属性值列表,支持CSS和XPath选择器
481
-
482
- 参数:
483
- xpath_or_css: XPath或CSS选择器
484
- attr_name: 属性名称
485
- default: 默认返回值
486
-
487
- 返回:
488
- 属性值列表
489
-
490
- 示例:
491
- # 使用CSS选择器
492
- links = response.extract_attrs('a', 'href')
493
- images = response.extract_attrs('img', 'src')
494
-
495
- # 使用XPath选择器
496
- links = response.extract_attrs('//a[@class="link"]', 'href')
497
- images = response.extract_attrs('//img[@alt]', 'src')
498
- """
499
- if default is None:
500
- default = []
501
-
502
- try:
503
- elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
504
- if not elements:
505
- return default
506
-
507
- result = extract_attrs(elements, attr_name)
508
- return result if result else default
509
- except Exception:
510
- return default
511
-
512
- # ==================== 正则表达式相关方法 ====================
513
-
514
- def re_search(self, pattern: str, flags: int = re.DOTALL) -> Optional[re.Match]:
515
- """在响应文本上执行正则表达式搜索。"""
516
- if not isinstance(pattern, str):
517
- raise TypeError("Pattern must be a string")
518
- return re.search(pattern, self.text, flags=flags)
519
-
520
- def re_findall(self, pattern: str, flags: int = re.DOTALL) -> List[Any]:
521
- """在响应文本上执行正则表达式查找。"""
522
- if not isinstance(pattern, str):
523
- raise TypeError("Pattern must be a string")
524
- return re.findall(pattern, self.text, flags=flags)
525
-
526
- # ==================== Cookie处理方法 ====================
527
-
528
- def get_cookies(self) -> Dict[str, str]:
529
- """从响应头中解析并返回Cookies。"""
530
- cookie_header = self.headers.get("Set-Cookie", "")
531
- if isinstance(cookie_header, list):
532
- cookie_header = ", ".join(cookie_header)
533
- cookies = SimpleCookie()
534
- cookies.load(cookie_header)
535
- return {key: morsel.value for key, morsel in cookies.items()}
536
-
537
- # ==================== 请求相关方法 ====================
538
-
539
- def follow(self, url: str, callback=None, **kwargs):
540
- """
541
- 创建一个跟随链接的请求。
542
-
543
- Args:
544
- url: 要跟随的URL(可以是相对URL)
545
- callback: 回调函数
546
- **kwargs: 传递给Request的其他参数
547
-
548
- Returns:
549
- Request: 新的请求对象
550
- """
551
- # 延迟导入Request以避免循环导入
552
- from ..network.request import Request
553
-
554
- # 使用urljoin处理相对URL
555
- absolute_url = self.urljoin(url)
556
-
557
- # 创建新的请求
558
- return Request(
559
- url=absolute_url,
560
- callback=callback or getattr(self.request, 'callback', None),
561
- **kwargs
562
- )
563
-
564
- @property
565
- def meta(self) -> Dict:
566
- """获取关联的 Request 对象的 meta 字典。"""
567
- return self.request.meta if self.request else {}
568
-
569
- def __str__(self):
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ """
4
+ HTTP Response 封装模块
5
+ =====================
6
+ 提供功能丰富的HTTP响应封装,支持:
7
+ - 智能编码检测和解码
8
+ - XPath/CSS 选择器
9
+ - JSON 解析和缓存
10
+ - 正则表达式支持
11
+ - Cookie 处理
12
+ """
13
+ import re
14
+ import ujson
15
+ from typing import Dict, Any, List, Optional
16
+ from urllib.parse import urljoin as _urljoin
17
+ from parsel import Selector, SelectorList
18
+
19
+ # 尝试导入 w3lib 编码检测函数
20
+ try:
21
+ from w3lib.encoding import (
22
+ html_body_declared_encoding,
23
+ html_to_unicode,
24
+ http_content_type_encoding,
25
+ read_bom,
26
+ resolve_encoding,
27
+ )
28
+ W3LIB_AVAILABLE = True
29
+ except ImportError:
30
+ W3LIB_AVAILABLE = False
31
+ # w3lib 不可用时,从 utils 导入替代函数
32
+ from crawlo.utils import (
33
+ html_body_declared_encoding,
34
+ html_to_unicode,
35
+ http_content_type_encoding,
36
+ read_bom,
37
+ resolve_encoding,
38
+ )
39
+
40
+ from crawlo.exceptions import DecodeError
41
+ from crawlo.utils import (
42
+ extract_text,
43
+ extract_texts,
44
+ extract_attr,
45
+ extract_attrs,
46
+ is_xpath,
47
+ parse_cookies,
48
+ regex_search,
49
+ regex_findall,
50
+ get_header_value
51
+ )
52
+
53
+
54
+ def memoize_method_noargs(func):
55
+ """
56
+ 装饰器,用于缓存无参数方法的结果
57
+ """
58
+ cache_attr = f'_cache_{func.__name__}'
59
+
60
+ def wrapper(self):
61
+ if not hasattr(self, cache_attr):
62
+ setattr(self, cache_attr, func(self))
63
+ return getattr(self, cache_attr)
64
+
65
+ return wrapper
66
+
67
+
68
+ class Response:
69
+ """
70
+ HTTP响应的封装,提供数据解析的便捷方法。
71
+
72
+ 功能特性:
73
+ - 智能编码检测和缓存
74
+ - 懒加载 Selector 实例
75
+ - JSON 解析和缓存
76
+ - 多类型数据提取
77
+ """
78
+
79
+ # 默认编码
80
+ _DEFAULT_ENCODING = "ascii"
81
+
82
+ def __init__(
83
+ self,
84
+ url: str,
85
+ *,
86
+ headers: Optional[Dict[str, Any]] = None,
87
+ body: bytes = b"",
88
+ method: str = 'GET',
89
+ request: Optional['Request'] = None, # 使用字符串注解避免循环导入
90
+ status_code: int = 200,
91
+ ):
92
+ # 基本属性
93
+ self.url = url
94
+ self.headers = headers or {}
95
+ self.body = body
96
+ self.method = method.upper()
97
+ self.request = request
98
+ self.status_code = status_code
99
+
100
+ # 编码处理
101
+ self.encoding = self._determine_encoding()
102
+
103
+ # 缓存属性
104
+ self._text_cache = None
105
+ self._json_cache = None
106
+ self._selector_instance = None
107
+
108
+ # 状态标记
109
+ self._is_success = 200 <= status_code < 300
110
+ self._is_redirect = 300 <= status_code < 400
111
+ self._is_client_error = 400 <= status_code < 500
112
+ self._is_server_error = status_code >= 500
113
+
114
+ # ==================== 编码检测相关方法 ====================
115
+
116
+ def _determine_encoding(self) -> str:
117
+ """
118
+ 智能检测响应编码
119
+
120
+ 编码检测优先级:
121
+ 1. Request 中指定的编码
122
+ 2. BOM 字节顺序标记
123
+ 3. HTTP Content-Type 头部
124
+ 4. HTML meta 标签声明
125
+ 5. 内容自动检测
126
+ 6. 默认编码 (utf-8)
127
+ """
128
+ # 1. 优先使用声明的编码
129
+ declared_encoding = self._declared_encoding()
130
+ if declared_encoding:
131
+ return declared_encoding
132
+
133
+ # 2. 使用 w3lib 进行编码检测(如果可用)
134
+ if W3LIB_AVAILABLE:
135
+ return self._body_inferred_encoding()
136
+ else:
137
+ # 如果 w3lib 不可用,使用原有的检测逻辑
138
+ # Content-Type 头中检测
139
+ content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
140
+ if content_type:
141
+ charset_match = re.search(r"charset=([\w-]+)", content_type, re.I)
142
+ if charset_match:
143
+ return charset_match.group(1).lower()
144
+
145
+ # 从 HTML meta 标签中检测(仅对HTML内容)
146
+ if b'<html' in self.body[:1024].lower():
147
+ # 查找 <meta charset="xxx"> 或 <meta http-equiv="Content-Type" content="...charset=xxx">
148
+ html_start = self.body[:4096] # 只检查前4KB
149
+ try:
150
+ html_text = html_start.decode('ascii', errors='ignore')
151
+ # <meta charset="utf-8">
152
+ charset_match = re.search(r'<meta[^>]+charset=["\']?([\w-]+)', html_text, re.I)
153
+ if charset_match:
154
+ return charset_match.group(1).lower()
155
+
156
+ # <meta http-equiv="Content-Type" content="text/html; charset=utf-8">
157
+ content_match = re.search(r'<meta[^>]+content=["\'][^"\'>]*charset=([\w-]+)', html_text, re.I)
158
+ if content_match:
159
+ return content_match.group(1).lower()
160
+ except Exception:
161
+ pass
162
+
163
+ # 3. 默认使用 utf-8
164
+ return 'utf-8'
165
+
166
+ def _declared_encoding(self) -> Optional[str]:
167
+ """
168
+ 获取声明的编码
169
+ 优先级:Request编码 > BOM > HTTP头部 > HTML meta标签
170
+ """
171
+ # 1. Request 中指定的编码
172
+ if self.request and getattr(self.request, 'encoding', None):
173
+ return self.request.encoding
174
+
175
+ # 2. BOM 字节顺序标记
176
+ bom_encoding = self._bom_encoding()
177
+ if bom_encoding:
178
+ return bom_encoding
179
+
180
+ # 3. HTTP Content-Type 头部
181
+ headers_encoding = self._headers_encoding()
182
+ if headers_encoding:
183
+ return headers_encoding
184
+
185
+ # 4. HTML meta 标签声明编码
186
+ body_declared_encoding = self._body_declared_encoding()
187
+ if body_declared_encoding:
188
+ return body_declared_encoding
189
+
190
+ return None
191
+
192
+ @memoize_method_noargs
193
+ def _bom_encoding(self) -> Optional[str]:
194
+ """BOM 字节顺序标记编码检测"""
195
+ if not W3LIB_AVAILABLE:
196
+ # 使用替代函数
197
+ encoding, _ = read_bom(self.body)
198
+ return encoding
199
+ return read_bom(self.body)[0]
200
+
201
+ @memoize_method_noargs
202
+ def _headers_encoding(self) -> Optional[str]:
203
+ """HTTP Content-Type 头部编码检测"""
204
+ if not W3LIB_AVAILABLE:
205
+ # 使用替代函数
206
+ content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
207
+ return http_content_type_encoding(content_type)
208
+ content_type = self.headers.get("Content-Type", b"") or self.headers.get("content-type", b"")
209
+ if isinstance(content_type, bytes):
210
+ content_type = content_type.decode('latin-1')
211
+ return http_content_type_encoding(content_type)
212
+
213
+ @memoize_method_noargs
214
+ def _body_declared_encoding(self) -> Optional[str]:
215
+ """HTML meta 标签声明编码检测"""
216
+ if not W3LIB_AVAILABLE:
217
+ # 使用替代函数
218
+ return html_body_declared_encoding(self.body)
219
+ return html_body_declared_encoding(self.body)
220
+
221
+ @memoize_method_noargs
222
+ def _body_inferred_encoding(self) -> str:
223
+ """内容自动检测编码"""
224
+ if not W3LIB_AVAILABLE:
225
+ # 使用替代函数
226
+ content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
227
+ # 使用 html_to_unicode 函数进行编码检测
228
+ encoding, _ = html_to_unicode(
229
+ content_type,
230
+ self.body,
231
+ auto_detect_fun=self._auto_detect_fun,
232
+ default_encoding=self._DEFAULT_ENCODING,
233
+ )
234
+ return encoding
235
+
236
+ content_type = self.headers.get("Content-Type", b"") or self.headers.get("content-type", b"")
237
+ if isinstance(content_type, bytes):
238
+ content_type = content_type.decode('latin-1')
239
+
240
+ # 使用 w3lib 的 html_to_unicode 函数进行编码检测
241
+ benc, _ = html_to_unicode(
242
+ content_type,
243
+ self.body,
244
+ auto_detect_fun=self._auto_detect_fun,
245
+ default_encoding=self._DEFAULT_ENCODING,
246
+ )
247
+ return benc
248
+
249
+ def _auto_detect_fun(self, text: bytes) -> Optional[str]:
250
+ """自动检测编码的回调函数"""
251
+ if not W3LIB_AVAILABLE:
252
+ # 使用替代函数
253
+ for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
254
+ try:
255
+ text.decode(enc)
256
+ except UnicodeError:
257
+ continue
258
+ return resolve_encoding(enc)
259
+ return None
260
+ for enc in (self._DEFAULT_ENCODING, "utf-8", "cp1252"):
261
+ try:
262
+ text.decode(enc)
263
+ except UnicodeError:
264
+ continue
265
+ return resolve_encoding(enc)
266
+ return None
267
+
268
+ @property
269
+ def text(self) -> str:
270
+ """将响应体(body)以正确的编码解码为字符串,并缓存结果。"""
271
+ if self._text_cache is not None:
272
+ return self._text_cache
273
+
274
+ if not self.body:
275
+ self._text_cache = ""
276
+ return self._text_cache
277
+
278
+ # 如果可用,使用 w3lib 进行更准确的解码
279
+ if W3LIB_AVAILABLE:
280
+ try:
281
+ content_type = self.headers.get("Content-Type", b"") or self.headers.get("content-type", b"")
282
+ if isinstance(content_type, bytes):
283
+ content_type = content_type.decode('latin-1')
284
+
285
+ # 使用 w3lib 的 html_to_unicode 函数
286
+ _, self._text_cache = html_to_unicode(
287
+ content_type,
288
+ self.body,
289
+ default_encoding=self.encoding
290
+ )
291
+ return self._text_cache
292
+ except Exception:
293
+ # 如果 w3lib 解码失败,回退到原有方法
294
+ pass
295
+ else:
296
+ # 使用替代函数
297
+ try:
298
+ content_type = self.headers.get("content-type", "") or self.headers.get("Content-Type", "")
299
+ # 使用 html_to_unicode 函数
300
+ _, self._text_cache = html_to_unicode(
301
+ content_type,
302
+ self.body,
303
+ default_encoding=self.encoding
304
+ )
305
+ return self._text_cache
306
+ except Exception:
307
+ # 如果解码失败,回退到原有方法
308
+ pass
309
+
310
+ # 尝试多种编码
311
+ encodings_to_try = [self.encoding]
312
+ if self.encoding != 'utf-8':
313
+ encodings_to_try.append('utf-8')
314
+ if 'gbk' not in encodings_to_try:
315
+ encodings_to_try.append('gbk')
316
+ if 'gb2312' not in encodings_to_try:
317
+ encodings_to_try.append('gb2312')
318
+ encodings_to_try.append('latin1') # 最后的回退选项
319
+
320
+ for encoding in encodings_to_try:
321
+ if not encoding:
322
+ continue
323
+ try:
324
+ self._text_cache = self.body.decode(encoding)
325
+ return self._text_cache
326
+ except (UnicodeDecodeError, LookupError):
327
+ continue
328
+
329
+ # 所有编码都失败,使用容错解码
330
+ try:
331
+ self._text_cache = self.body.decode('utf-8', errors='replace')
332
+ return self._text_cache
333
+ except Exception as e:
334
+ raise DecodeError(f"Failed to decode response from {self.url}: {e}")
335
+
336
+ # ==================== 状态检查相关属性 ====================
337
+
338
+ @property
339
+ def is_success(self) -> bool:
340
+ """检查响应是否成功 (2xx)"""
341
+ return self._is_success
342
+
343
+ @property
344
+ def is_redirect(self) -> bool:
345
+ """检查响应是否为重定向 (3xx)"""
346
+ return self._is_redirect
347
+
348
+ @property
349
+ def is_client_error(self) -> bool:
350
+ """检查响应是否为客户端错误 (4xx)"""
351
+ return self._is_client_error
352
+
353
+ @property
354
+ def is_server_error(self) -> bool:
355
+ """检查响应是否为服务器错误 (5xx)"""
356
+ return self._is_server_error
357
+
358
+ # ==================== 响应头相关属性 ====================
359
+
360
+ @property
361
+ def content_type(self) -> str:
362
+ """获取响应的 Content-Type"""
363
+ return get_header_value(self.headers, 'content-type', '')
364
+
365
+ @property
366
+ def content_length(self) -> Optional[int]:
367
+ """获取响应的 Content-Length"""
368
+ length = get_header_value(self.headers, 'content-length')
369
+ return int(length) if length else None
370
+
371
+ # ==================== JSON处理方法 ====================
372
+
373
+ def json(self, default: Any = None) -> Any:
374
+ """将响应文本解析为 JSON 对象。"""
375
+ if self._json_cache is not None:
376
+ return self._json_cache
377
+
378
+ try:
379
+ self._json_cache = ujson.loads(self.text)
380
+ return self._json_cache
381
+ except (ujson.JSONDecodeError, ValueError) as e:
382
+ if default is not None:
383
+ return default
384
+ raise DecodeError(f"Failed to parse JSON from {self.url}: {e}")
385
+
386
+ # ==================== URL处理方法 ====================
387
+
388
+ def urljoin(self, url: str) -> str:
389
+ """拼接 URL,自动处理相对路径。"""
390
+ return _urljoin(self.url, url)
391
+
392
+ # ==================== 选择器相关方法 ====================
393
+
394
+ @property
395
+ def _selector(self) -> Selector:
396
+ """懒加载 Selector 实例"""
397
+ if self._selector_instance is None:
398
+ self._selector_instance = Selector(self.text)
399
+ return self._selector_instance
400
+
401
+ def xpath(self, query: str) -> SelectorList:
402
+ """使用 XPath 选择器查询文档。"""
403
+ return self._selector.xpath(query)
404
+
405
+ def css(self, query: str) -> SelectorList:
406
+ """使用 CSS 选择器查询文档。"""
407
+ return self._selector.css(query)
408
+
409
+ def _is_xpath(self, query: str) -> bool:
410
+ """判断查询语句是否为XPath"""
411
+ return is_xpath(query)
412
+
413
+ def _extract_text(self, elements: SelectorList, join_str: str = " ") -> str:
414
+ """
415
+ 从元素列表中提取文本并拼接
416
+
417
+ :param elements: SelectorList元素列表
418
+ :param join_str: 文本拼接分隔符
419
+ :return: 拼接后的文本
420
+ """
421
+ return extract_text(elements, join_str)
422
+
423
+ def extract_text(self, xpath_or_css: str, join_str: str = " ", default: str = '') -> str:
424
+ """
425
+ 提取单个元素的文本内容,支持CSS和XPath选择器
426
+
427
+ 参数:
428
+ xpath_or_css: XPath或CSS选择器
429
+ join_str: 文本拼接分隔符(默认为空格)
430
+ default: 默认返回值,当未找到元素时返回
431
+
432
+ 返回:
433
+ 拼接后的纯文本字符串
434
+
435
+ 示例:
436
+ # 使用CSS选择器
437
+ title = response.extract_text('title')
438
+ content = response.extract_text('.content p', join_str=' ')
439
+
440
+ # 使用XPath选择器
441
+ title = response.extract_text('//title')
442
+ content = response.extract_text('//div[@class="content"]//p', join_str=' ')
443
+ """
444
+ try:
445
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
446
+ if not elements:
447
+ return default
448
+ return self._extract_text(elements, join_str)
449
+ except Exception:
450
+ return default
451
+
452
+ def extract_texts(self, xpath_or_css: str, join_str: str = " ", default: Optional[List[str]] = None) -> List[str]:
453
+ """
454
+ 提取多个元素的文本内容列表,支持CSS和XPath选择器
455
+
456
+ 参数:
457
+ xpath_or_css: XPath或CSS选择器
458
+ join_str: 单个节点内文本拼接分隔符
459
+ default: 默认返回值,当未找到元素时返回
460
+
461
+ 返回:
462
+ 纯文本列表(每个元素对应一个节点的文本)
463
+
464
+ 示例:
465
+ # 使用CSS选择器
466
+ paragraphs = response.extract_texts('.content p')
467
+ titles = response.extract_texts('h1, h2, h3')
468
+
469
+ # 使用XPath选择器
470
+ paragraphs = response.extract_texts('//div[@class="content"]//p')
471
+ titles = response.extract_texts('//h1 | //h2 | //h3')
472
+ """
473
+ if default is None:
474
+ default = []
475
+
476
+ try:
477
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
478
+ if not elements:
479
+ return default
480
+
481
+ result = extract_texts(elements, join_str)
482
+ return result if result else default
483
+ except Exception:
484
+ return default
485
+
486
+ def extract_attr(self, xpath_or_css: str, attr_name: str, default: Any = None) -> Any:
487
+ """
488
+ 提取单个元素的属性值,支持CSS和XPath选择器
489
+
490
+ 参数:
491
+ xpath_or_css: XPath或CSS选择器
492
+ attr_name: 属性名称
493
+ default: 默认返回值
494
+
495
+ 返回:
496
+ 属性值或默认值
497
+
498
+ 示例:
499
+ # 使用CSS选择器
500
+ link_href = response.extract_attr('a', 'href')
501
+ img_src = response.extract_attr('.image', 'src')
502
+
503
+ # 使用XPath选择器
504
+ link_href = response.extract_attr('//a[@class="link"]', 'href')
505
+ img_src = response.extract_attr('//img[@alt="example"]', 'src')
506
+ """
507
+ try:
508
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
509
+ if not elements:
510
+ return default
511
+ return extract_attr(elements, attr_name, default)
512
+ except Exception:
513
+ return default
514
+
515
+ def extract_attrs(self, xpath_or_css: str, attr_name: str, default: Optional[List[Any]] = None) -> List[Any]:
516
+ """
517
+ 提取多个元素的属性值列表,支持CSS和XPath选择器
518
+
519
+ 参数:
520
+ xpath_or_css: XPath或CSS选择器
521
+ attr_name: 属性名称
522
+ default: 默认返回值
523
+
524
+ 返回:
525
+ 属性值列表
526
+
527
+ 示例:
528
+ # 使用CSS选择器
529
+ links = response.extract_attrs('a', 'href')
530
+ images = response.extract_attrs('img', 'src')
531
+
532
+ # 使用XPath选择器
533
+ links = response.extract_attrs('//a[@class="link"]', 'href')
534
+ images = response.extract_attrs('//img[@alt]', 'src')
535
+ """
536
+ if default is None:
537
+ default = []
538
+
539
+ try:
540
+ elements = self.xpath(xpath_or_css) if self._is_xpath(xpath_or_css) else self.css(xpath_or_css)
541
+ if not elements:
542
+ return default
543
+
544
+ result = extract_attrs(elements, attr_name)
545
+ return result if result else default
546
+ except Exception:
547
+ return default
548
+
549
+ # ==================== 正则表达式相关方法 ====================
550
+
551
+ def re_search(self, pattern: str, flags: int = re.DOTALL) -> Optional[re.Match]:
552
+ """在响应文本上执行正则表达式搜索。"""
553
+ return regex_search(pattern, self.text, flags)
554
+
555
+ def re_findall(self, pattern: str, flags: int = re.DOTALL) -> List[Any]:
556
+ """在响应文本上执行正则表达式查找。"""
557
+ return regex_findall(pattern, self.text, flags)
558
+
559
+ # ==================== Cookie处理方法 ====================
560
+
561
+ def get_cookies(self) -> Dict[str, str]:
562
+ """从响应头中解析并返回Cookies。"""
563
+ cookie_header = self.headers.get("Set-Cookie", "")
564
+ return parse_cookies(cookie_header)
565
+
566
+ # ==================== 请求相关方法 ====================
567
+
568
+ def follow(self, url: str, callback=None, **kwargs):
569
+ """
570
+ 创建一个跟随链接的请求。
571
+
572
+ Args:
573
+ url: 要跟随的URL(可以是相对URL)
574
+ callback: 回调函数
575
+ **kwargs: 传递给Request的其他参数
576
+
577
+ Returns:
578
+ Request: 新的请求对象
579
+ """
580
+ # 延迟导入Request以避免循环导入
581
+ from ..network.request import Request
582
+
583
+ # 使用urljoin处理相对URL
584
+ absolute_url = self.urljoin(url)
585
+
586
+ # 创建新的请求
587
+ return Request(
588
+ url=absolute_url,
589
+ callback=callback or getattr(self.request, 'callback', None),
590
+ **kwargs
591
+ )
592
+
593
+ @property
594
+ def meta(self) -> Dict:
595
+ """获取关联的 Request 对象的 meta 字典。"""
596
+ return self.request.meta if self.request else {}
597
+
598
+ def __str__(self):
570
599
  return f"<{self.status_code} {self.url}>"