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.
- crawlo/__init__.py +90 -90
- crawlo/__version__.py +1 -1
- crawlo/cli.py +75 -75
- crawlo/commands/__init__.py +14 -14
- crawlo/commands/check.py +594 -594
- crawlo/commands/genspider.py +186 -186
- crawlo/commands/help.py +140 -140
- crawlo/commands/list.py +155 -155
- crawlo/commands/run.py +379 -379
- crawlo/commands/startproject.py +460 -460
- crawlo/commands/stats.py +187 -187
- crawlo/commands/utils.py +196 -196
- crawlo/config.py +320 -320
- crawlo/config_validator.py +277 -277
- crawlo/core/__init__.py +52 -52
- crawlo/core/engine.py +451 -451
- crawlo/core/processor.py +47 -47
- crawlo/core/scheduler.py +290 -290
- crawlo/crawler.py +698 -698
- crawlo/data/__init__.py +5 -5
- crawlo/data/user_agents.py +194 -194
- crawlo/downloader/__init__.py +280 -280
- crawlo/downloader/aiohttp_downloader.py +233 -233
- crawlo/downloader/cffi_downloader.py +250 -250
- crawlo/downloader/httpx_downloader.py +265 -265
- crawlo/downloader/hybrid_downloader.py +212 -212
- crawlo/downloader/playwright_downloader.py +425 -425
- crawlo/downloader/selenium_downloader.py +486 -486
- crawlo/event.py +45 -45
- crawlo/exceptions.py +214 -214
- crawlo/extension/__init__.py +64 -64
- crawlo/extension/health_check.py +141 -141
- crawlo/extension/log_interval.py +94 -94
- crawlo/extension/log_stats.py +70 -70
- crawlo/extension/logging_extension.py +53 -53
- crawlo/extension/memory_monitor.py +104 -104
- crawlo/extension/performance_profiler.py +133 -133
- crawlo/extension/request_recorder.py +107 -107
- crawlo/factories/__init__.py +27 -27
- crawlo/factories/base.py +68 -68
- crawlo/factories/crawler.py +104 -104
- crawlo/factories/registry.py +84 -84
- crawlo/factories/utils.py +134 -134
- crawlo/filters/__init__.py +170 -170
- crawlo/filters/aioredis_filter.py +347 -347
- crawlo/filters/memory_filter.py +261 -261
- crawlo/framework.py +306 -306
- crawlo/initialization/__init__.py +44 -44
- crawlo/initialization/built_in.py +391 -391
- crawlo/initialization/context.py +141 -141
- crawlo/initialization/core.py +240 -240
- crawlo/initialization/phases.py +229 -229
- crawlo/initialization/registry.py +143 -143
- crawlo/initialization/utils.py +48 -48
- crawlo/interfaces.py +23 -23
- crawlo/items/__init__.py +23 -23
- crawlo/items/base.py +23 -23
- crawlo/items/fields.py +52 -52
- crawlo/items/items.py +104 -104
- crawlo/logging/__init__.py +42 -42
- crawlo/logging/config.py +280 -276
- crawlo/logging/factory.py +175 -175
- crawlo/logging/manager.py +104 -104
- crawlo/middleware/__init__.py +87 -87
- crawlo/middleware/default_header.py +132 -132
- crawlo/middleware/download_delay.py +104 -104
- crawlo/middleware/middleware_manager.py +142 -142
- crawlo/middleware/offsite.py +123 -123
- crawlo/middleware/proxy.py +209 -209
- crawlo/middleware/request_ignore.py +86 -86
- crawlo/middleware/response_code.py +150 -150
- crawlo/middleware/response_filter.py +136 -136
- crawlo/middleware/retry.py +124 -124
- crawlo/mode_manager.py +287 -287
- crawlo/network/__init__.py +21 -21
- crawlo/network/request.py +408 -376
- crawlo/network/response.py +598 -569
- crawlo/pipelines/__init__.py +52 -52
- crawlo/pipelines/base_pipeline.py +452 -452
- crawlo/pipelines/bloom_dedup_pipeline.py +145 -146
- crawlo/pipelines/console_pipeline.py +39 -39
- crawlo/pipelines/csv_pipeline.py +316 -316
- crawlo/pipelines/database_dedup_pipeline.py +196 -197
- crawlo/pipelines/json_pipeline.py +218 -218
- crawlo/pipelines/memory_dedup_pipeline.py +104 -105
- crawlo/pipelines/mongo_pipeline.py +140 -139
- crawlo/pipelines/mysql_pipeline.py +468 -469
- crawlo/pipelines/pipeline_manager.py +100 -100
- crawlo/pipelines/redis_dedup_pipeline.py +155 -155
- crawlo/project.py +347 -347
- crawlo/queue/__init__.py +9 -9
- crawlo/queue/pqueue.py +38 -38
- crawlo/queue/queue_manager.py +591 -591
- crawlo/queue/redis_priority_queue.py +518 -518
- crawlo/settings/__init__.py +7 -7
- crawlo/settings/default_settings.py +287 -284
- crawlo/settings/setting_manager.py +219 -219
- crawlo/spider/__init__.py +658 -657
- crawlo/stats_collector.py +81 -81
- crawlo/subscriber.py +129 -129
- crawlo/task_manager.py +138 -138
- crawlo/templates/crawlo.cfg.tmpl +10 -10
- crawlo/templates/project/__init__.py.tmpl +1 -1
- crawlo/templates/project/items.py.tmpl +13 -13
- crawlo/templates/project/middlewares.py.tmpl +38 -38
- crawlo/templates/project/pipelines.py.tmpl +35 -35
- crawlo/templates/project/settings.py.tmpl +113 -109
- crawlo/templates/project/settings_distributed.py.tmpl +160 -156
- crawlo/templates/project/settings_gentle.py.tmpl +174 -170
- crawlo/templates/project/settings_high_performance.py.tmpl +175 -171
- crawlo/templates/project/settings_minimal.py.tmpl +102 -98
- crawlo/templates/project/settings_simple.py.tmpl +172 -168
- crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
- crawlo/templates/run.py.tmpl +23 -23
- crawlo/templates/spider/spider.py.tmpl +32 -32
- crawlo/templates/spiders_init.py.tmpl +4 -4
- crawlo/tools/__init__.py +86 -86
- crawlo/tools/date_tools.py +289 -289
- crawlo/tools/distributed_coordinator.py +384 -384
- crawlo/tools/scenario_adapter.py +262 -262
- crawlo/tools/text_cleaner.py +232 -232
- crawlo/utils/__init__.py +74 -50
- crawlo/utils/batch_processor.py +276 -276
- crawlo/utils/config_manager.py +442 -442
- crawlo/utils/controlled_spider_mixin.py +439 -439
- crawlo/utils/db_helper.py +250 -250
- crawlo/utils/encoding_helper.py +190 -0
- crawlo/utils/error_handler.py +410 -410
- crawlo/utils/fingerprint.py +121 -121
- crawlo/utils/func_tools.py +82 -82
- crawlo/utils/large_scale_helper.py +344 -344
- crawlo/utils/leak_detector.py +335 -335
- crawlo/utils/misc.py +81 -81
- crawlo/utils/mongo_connection_pool.py +157 -157
- crawlo/utils/mysql_connection_pool.py +197 -197
- crawlo/utils/performance_monitor.py +285 -285
- crawlo/utils/queue_helper.py +175 -175
- crawlo/utils/redis_checker.py +90 -90
- crawlo/utils/redis_connection_pool.py +578 -578
- crawlo/utils/redis_key_validator.py +198 -198
- crawlo/utils/request.py +278 -278
- crawlo/utils/request_serializer.py +225 -225
- crawlo/utils/resource_manager.py +337 -337
- crawlo/utils/response_helper.py +113 -0
- crawlo/utils/selector_helper.py +138 -137
- crawlo/utils/singleton.py +69 -69
- crawlo/utils/spider_loader.py +201 -201
- crawlo/utils/text_helper.py +94 -94
- {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/METADATA +831 -689
- crawlo-1.4.8.dist-info/RECORD +347 -0
- examples/__init__.py +7 -7
- tests/__init__.py +7 -7
- tests/advanced_tools_example.py +217 -217
- tests/authenticated_proxy_example.py +110 -110
- tests/baidu_performance_test.py +108 -108
- tests/baidu_test.py +59 -59
- tests/bug_check_test.py +250 -250
- tests/cleaners_example.py +160 -160
- tests/comprehensive_framework_test.py +212 -212
- tests/comprehensive_test.py +81 -81
- tests/comprehensive_testing_summary.md +186 -186
- tests/config_validation_demo.py +142 -142
- tests/controlled_spider_example.py +205 -205
- tests/date_tools_example.py +180 -180
- tests/debug_configure.py +69 -69
- tests/debug_framework_logger.py +84 -84
- tests/debug_log_config.py +126 -126
- tests/debug_log_levels.py +63 -63
- tests/debug_pipelines.py +66 -66
- tests/detailed_log_test.py +233 -233
- tests/direct_selector_helper_test.py +96 -96
- tests/distributed_dedup_test.py +467 -467
- tests/distributed_test.py +66 -66
- tests/distributed_test_debug.py +76 -76
- tests/dynamic_loading_example.py +523 -523
- tests/dynamic_loading_test.py +104 -104
- tests/error_handling_example.py +171 -171
- tests/explain_mysql_update_behavior.py +76 -76
- tests/final_comprehensive_test.py +151 -151
- tests/final_log_test.py +260 -260
- tests/final_validation_test.py +182 -182
- tests/fix_log_test.py +142 -142
- tests/framework_performance_test.py +202 -202
- tests/log_buffering_test.py +111 -111
- tests/log_generation_timing_test.py +153 -153
- tests/monitor_redis_dedup.sh +72 -72
- tests/ofweek_scrapy/ofweek_scrapy/items.py +12 -12
- tests/ofweek_scrapy/ofweek_scrapy/middlewares.py +100 -100
- tests/ofweek_scrapy/ofweek_scrapy/pipelines.py +13 -13
- tests/ofweek_scrapy/ofweek_scrapy/settings.py +84 -84
- tests/ofweek_scrapy/ofweek_scrapy/spiders/__init__.py +4 -4
- tests/ofweek_scrapy/scrapy.cfg +11 -11
- tests/optimized_performance_test.py +211 -211
- tests/performance_comparison.py +244 -244
- tests/queue_blocking_test.py +113 -113
- tests/queue_test.py +89 -89
- tests/redis_key_validation_demo.py +130 -130
- tests/request_params_example.py +150 -150
- tests/response_improvements_example.py +144 -144
- tests/scrapy_comparison/ofweek_scrapy.py +138 -138
- tests/scrapy_comparison/scrapy_test.py +133 -133
- tests/simple_cli_test.py +54 -54
- tests/simple_command_test.py +119 -119
- tests/simple_crawlo_test.py +126 -126
- tests/simple_follow_test.py +38 -38
- tests/simple_log_test2.py +137 -137
- tests/simple_optimization_test.py +128 -128
- tests/simple_queue_type_test.py +41 -41
- tests/simple_response_selector_test.py +94 -94
- tests/simple_selector_helper_test.py +154 -154
- tests/simple_selector_test.py +207 -207
- tests/simple_spider_test.py +49 -49
- tests/simple_url_test.py +73 -73
- tests/simulate_mysql_update_test.py +139 -139
- tests/spider_log_timing_test.py +177 -177
- tests/test_advanced_tools.py +148 -148
- tests/test_all_commands.py +230 -230
- tests/test_all_pipeline_fingerprints.py +133 -133
- tests/test_all_redis_key_configs.py +145 -145
- tests/test_asyncmy_usage.py +56 -56
- tests/test_batch_processor.py +178 -178
- tests/test_cleaners.py +54 -54
- tests/test_cli_arguments.py +118 -118
- tests/test_component_factory.py +174 -174
- tests/test_config_consistency.py +80 -80
- tests/test_config_merge.py +152 -152
- tests/test_config_validator.py +182 -182
- tests/test_controlled_spider_mixin.py +79 -79
- tests/test_crawler_process_import.py +38 -38
- tests/test_crawler_process_spider_modules.py +47 -47
- tests/test_crawlo_proxy_integration.py +114 -114
- tests/test_date_tools.py +123 -123
- tests/test_dedup_fix.py +220 -220
- tests/test_dedup_pipeline_consistency.py +124 -124
- tests/test_default_header_middleware.py +313 -313
- tests/test_distributed.py +65 -65
- tests/test_double_crawlo_fix.py +204 -204
- tests/test_double_crawlo_fix_simple.py +124 -124
- tests/test_download_delay_middleware.py +221 -221
- tests/test_downloader_proxy_compatibility.py +272 -272
- tests/test_edge_cases.py +305 -305
- tests/test_encoding_core.py +56 -56
- tests/test_encoding_detection.py +126 -126
- tests/test_enhanced_error_handler.py +270 -270
- tests/test_enhanced_error_handler_comprehensive.py +245 -245
- tests/test_error_handler_compatibility.py +112 -112
- tests/test_factories.py +252 -252
- tests/test_factory_compatibility.py +196 -196
- tests/test_final_validation.py +153 -153
- tests/test_fingerprint_consistency.py +135 -135
- tests/test_fingerprint_simple.py +51 -51
- tests/test_get_component_logger.py +83 -83
- tests/test_hash_performance.py +99 -99
- tests/test_integration.py +169 -169
- tests/test_item_dedup_redis_key.py +122 -122
- tests/test_large_scale_helper.py +235 -235
- tests/test_logging_enhancements.py +374 -374
- tests/test_logging_final.py +184 -184
- tests/test_logging_integration.py +312 -312
- tests/test_logging_system.py +282 -282
- tests/test_middleware_debug.py +141 -141
- tests/test_mode_consistency.py +51 -51
- tests/test_multi_directory.py +67 -67
- tests/test_multiple_spider_modules.py +80 -80
- tests/test_mysql_pipeline_config.py +164 -164
- tests/test_mysql_pipeline_error.py +98 -98
- tests/test_mysql_pipeline_init_log.py +82 -82
- tests/test_mysql_pipeline_integration.py +132 -132
- tests/test_mysql_pipeline_refactor.py +143 -143
- tests/test_mysql_pipeline_refactor_simple.py +85 -85
- tests/test_mysql_pipeline_robustness.py +195 -195
- tests/test_mysql_pipeline_types.py +88 -88
- tests/test_mysql_update_columns.py +93 -93
- tests/test_offsite_middleware.py +244 -244
- tests/test_offsite_middleware_simple.py +203 -203
- tests/test_optimized_selector_naming.py +100 -100
- tests/test_parsel.py +29 -29
- tests/test_performance.py +327 -327
- tests/test_performance_monitor.py +115 -115
- tests/test_pipeline_fingerprint_consistency.py +86 -86
- tests/test_priority_behavior.py +211 -211
- tests/test_priority_consistency.py +151 -151
- tests/test_priority_consistency_fixed.py +249 -249
- tests/test_proxy_health_check.py +32 -32
- tests/test_proxy_middleware.py +217 -217
- tests/test_proxy_middleware_enhanced.py +212 -212
- tests/test_proxy_middleware_integration.py +142 -142
- tests/test_proxy_middleware_refactored.py +207 -207
- tests/test_proxy_only.py +83 -83
- tests/test_proxy_providers.py +56 -56
- tests/test_proxy_stats.py +19 -19
- tests/test_proxy_strategies.py +59 -59
- tests/test_proxy_with_downloader.py +152 -152
- tests/test_queue_empty_check.py +41 -41
- tests/test_queue_manager_double_crawlo.py +173 -173
- tests/test_queue_manager_redis_key.py +179 -179
- tests/test_queue_naming.py +154 -154
- tests/test_queue_type.py +106 -106
- tests/test_queue_type_redis_config_consistency.py +130 -130
- tests/test_random_headers_default.py +322 -322
- tests/test_random_headers_necessity.py +308 -308
- tests/test_random_user_agent.py +72 -72
- tests/test_redis_config.py +28 -28
- tests/test_redis_connection_pool.py +294 -294
- tests/test_redis_key_naming.py +181 -181
- tests/test_redis_key_validator.py +123 -123
- tests/test_redis_queue.py +224 -224
- tests/test_redis_queue_name_fix.py +175 -175
- tests/test_redis_queue_type_fallback.py +129 -129
- tests/test_request_ignore_middleware.py +182 -182
- tests/test_request_params.py +111 -111
- tests/test_request_serialization.py +70 -70
- tests/test_response_code_middleware.py +349 -349
- tests/test_response_filter_middleware.py +427 -427
- tests/test_response_follow.py +104 -104
- tests/test_response_improvements.py +152 -152
- tests/test_response_selector_methods.py +92 -92
- tests/test_response_url_methods.py +70 -70
- tests/test_response_urljoin.py +86 -86
- tests/test_retry_middleware.py +333 -333
- tests/test_retry_middleware_realistic.py +273 -273
- tests/test_scheduler.py +252 -252
- tests/test_scheduler_config_update.py +133 -133
- tests/test_scrapy_style_encoding.py +112 -112
- tests/test_selector_helper.py +100 -100
- tests/test_selector_optimizations.py +146 -146
- tests/test_simple_response.py +61 -61
- tests/test_spider_loader.py +49 -49
- tests/test_spider_loader_comprehensive.py +69 -69
- tests/test_spider_modules.py +84 -84
- tests/test_spiders/test_spider.py +9 -9
- tests/test_telecom_spider_redis_key.py +205 -205
- tests/test_template_content.py +87 -87
- tests/test_template_redis_key.py +134 -134
- tests/test_tools.py +159 -159
- tests/test_user_agent_randomness.py +176 -176
- tests/test_user_agents.py +96 -96
- tests/untested_features_report.md +138 -138
- tests/verify_debug.py +51 -51
- tests/verify_distributed.py +117 -117
- tests/verify_log_fix.py +111 -111
- tests/verify_mysql_warnings.py +109 -109
- crawlo/utils/log.py +0 -80
- crawlo/utils/url_utils.py +0 -40
- crawlo-1.4.7.dist-info/RECORD +0 -347
- {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/WHEEL +0 -0
- {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/entry_points.txt +0 -0
- {crawlo-1.4.7.dist-info → crawlo-1.4.8.dist-info}/top_level.txt +0 -0
crawlo/network/response.py
CHANGED
|
@@ -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
|
-
|
|
15
|
-
from typing import Dict, Any, List, Optional
|
|
16
|
-
from urllib.parse import urljoin as _urljoin
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
self.
|
|
96
|
-
self.
|
|
97
|
-
self.
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
self.
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
charset_match
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
#
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
return self.
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
content_type,
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
if not
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
278
|
-
|
|
279
|
-
if
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
|
|
333
|
-
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
345
|
-
|
|
346
|
-
|
|
347
|
-
|
|
348
|
-
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
377
|
-
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
|
|
398
|
-
|
|
399
|
-
|
|
400
|
-
|
|
401
|
-
|
|
402
|
-
|
|
403
|
-
|
|
404
|
-
|
|
405
|
-
|
|
406
|
-
"""
|
|
407
|
-
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
|
|
411
|
-
|
|
412
|
-
|
|
413
|
-
|
|
414
|
-
|
|
415
|
-
|
|
416
|
-
|
|
417
|
-
|
|
418
|
-
|
|
419
|
-
|
|
420
|
-
|
|
421
|
-
|
|
422
|
-
|
|
423
|
-
|
|
424
|
-
|
|
425
|
-
|
|
426
|
-
|
|
427
|
-
|
|
428
|
-
|
|
429
|
-
|
|
430
|
-
|
|
431
|
-
|
|
432
|
-
|
|
433
|
-
|
|
434
|
-
|
|
435
|
-
|
|
436
|
-
|
|
437
|
-
|
|
438
|
-
|
|
439
|
-
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
449
|
-
|
|
450
|
-
|
|
451
|
-
|
|
452
|
-
|
|
453
|
-
|
|
454
|
-
|
|
455
|
-
|
|
456
|
-
|
|
457
|
-
|
|
458
|
-
|
|
459
|
-
|
|
460
|
-
|
|
461
|
-
|
|
462
|
-
|
|
463
|
-
|
|
464
|
-
|
|
465
|
-
|
|
466
|
-
|
|
467
|
-
|
|
468
|
-
|
|
469
|
-
|
|
470
|
-
|
|
471
|
-
|
|
472
|
-
|
|
473
|
-
|
|
474
|
-
|
|
475
|
-
|
|
476
|
-
|
|
477
|
-
|
|
478
|
-
|
|
479
|
-
|
|
480
|
-
|
|
481
|
-
|
|
482
|
-
|
|
483
|
-
|
|
484
|
-
|
|
485
|
-
|
|
486
|
-
|
|
487
|
-
|
|
488
|
-
|
|
489
|
-
|
|
490
|
-
|
|
491
|
-
|
|
492
|
-
|
|
493
|
-
|
|
494
|
-
|
|
495
|
-
|
|
496
|
-
|
|
497
|
-
|
|
498
|
-
|
|
499
|
-
|
|
500
|
-
|
|
501
|
-
|
|
502
|
-
|
|
503
|
-
|
|
504
|
-
|
|
505
|
-
|
|
506
|
-
|
|
507
|
-
|
|
508
|
-
|
|
509
|
-
|
|
510
|
-
|
|
511
|
-
|
|
512
|
-
|
|
513
|
-
|
|
514
|
-
|
|
515
|
-
|
|
516
|
-
|
|
517
|
-
|
|
518
|
-
|
|
519
|
-
|
|
520
|
-
|
|
521
|
-
|
|
522
|
-
|
|
523
|
-
|
|
524
|
-
|
|
525
|
-
|
|
526
|
-
|
|
527
|
-
|
|
528
|
-
|
|
529
|
-
|
|
530
|
-
|
|
531
|
-
|
|
532
|
-
|
|
533
|
-
|
|
534
|
-
|
|
535
|
-
|
|
536
|
-
|
|
537
|
-
|
|
538
|
-
|
|
539
|
-
|
|
540
|
-
|
|
541
|
-
|
|
542
|
-
|
|
543
|
-
|
|
544
|
-
|
|
545
|
-
|
|
546
|
-
|
|
547
|
-
|
|
548
|
-
|
|
549
|
-
|
|
550
|
-
|
|
551
|
-
|
|
552
|
-
|
|
553
|
-
|
|
554
|
-
|
|
555
|
-
|
|
556
|
-
|
|
557
|
-
|
|
558
|
-
|
|
559
|
-
|
|
560
|
-
|
|
561
|
-
|
|
562
|
-
|
|
563
|
-
|
|
564
|
-
|
|
565
|
-
|
|
566
|
-
|
|
567
|
-
|
|
568
|
-
|
|
569
|
-
|
|
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}>"
|