PyPI - crawlo - Versions diffs - 1.1.9__py3-none-any.whl → 1.2.1__py3-none-any.whl - Mend

crawlo 1.1.9py3-none-any.whl → 1.2.1py3-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 (221) hide show

crawlo/__init__.py +61 -61
crawlo/__version__.py +1 -1
crawlo/cleaners/__init__.py +60 -60
crawlo/cleaners/data_formatter.py +225 -225
crawlo/cleaners/encoding_converter.py +125 -125
crawlo/cleaners/text_cleaner.py +232 -232
crawlo/cli.py +65 -65
crawlo/commands/__init__.py +14 -14
crawlo/commands/check.py +594 -594
crawlo/commands/genspider.py +151 -151
crawlo/commands/help.py +142 -132
crawlo/commands/list.py +155 -155
crawlo/commands/run.py +292 -292
crawlo/commands/startproject.py +418 -418
crawlo/commands/stats.py +188 -188
crawlo/commands/utils.py +186 -186
crawlo/config.py +312 -312
crawlo/config_validator.py +252 -252
crawlo/core/__init__.py +2 -2
crawlo/core/engine.py +354 -345
crawlo/core/processor.py +40 -40
crawlo/core/scheduler.py +143 -136
crawlo/crawler.py +1027 -1027
crawlo/downloader/__init__.py +266 -266
crawlo/downloader/aiohttp_downloader.py +220 -220
crawlo/downloader/cffi_downloader.py +256 -256
crawlo/downloader/httpx_downloader.py +259 -259
crawlo/downloader/hybrid_downloader.py +213 -213
crawlo/downloader/playwright_downloader.py +402 -402
crawlo/downloader/selenium_downloader.py +472 -472
crawlo/event.py +11 -11
crawlo/exceptions.py +81 -81
crawlo/extension/__init__.py +37 -37
crawlo/extension/health_check.py +141 -141
crawlo/extension/log_interval.py +57 -57
crawlo/extension/log_stats.py +81 -81
crawlo/extension/logging_extension.py +43 -43
crawlo/extension/memory_monitor.py +104 -104
crawlo/extension/performance_profiler.py +133 -133
crawlo/extension/request_recorder.py +107 -107
crawlo/filters/__init__.py +154 -154
crawlo/filters/aioredis_filter.py +280 -280
crawlo/filters/memory_filter.py +269 -269
crawlo/items/__init__.py +23 -23
crawlo/items/base.py +21 -21
crawlo/items/fields.py +53 -53
crawlo/items/items.py +104 -104
crawlo/middleware/__init__.py +21 -21
crawlo/middleware/default_header.py +132 -32
crawlo/middleware/download_delay.py +105 -28
crawlo/middleware/middleware_manager.py +135 -135
crawlo/middleware/offsite.py +116 -0
crawlo/middleware/proxy.py +366 -272
crawlo/middleware/request_ignore.py +88 -30
crawlo/middleware/response_code.py +164 -18
crawlo/middleware/response_filter.py +138 -26
crawlo/middleware/retry.py +124 -124
crawlo/mode_manager.py +211 -211
crawlo/network/__init__.py +21 -21
crawlo/network/request.py +338 -338
crawlo/network/response.py +359 -359
crawlo/pipelines/__init__.py +21 -21
crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
crawlo/pipelines/console_pipeline.py +39 -39
crawlo/pipelines/csv_pipeline.py +316 -316
crawlo/pipelines/database_dedup_pipeline.py +224 -224
crawlo/pipelines/json_pipeline.py +218 -218
crawlo/pipelines/memory_dedup_pipeline.py +115 -115
crawlo/pipelines/mongo_pipeline.py +131 -131
crawlo/pipelines/mysql_pipeline.py +316 -316
crawlo/pipelines/pipeline_manager.py +61 -61
crawlo/pipelines/redis_dedup_pipeline.py +167 -167
crawlo/project.py +187 -187
crawlo/queue/pqueue.py +37 -37
crawlo/queue/queue_manager.py +337 -334
crawlo/queue/redis_priority_queue.py +298 -298
crawlo/settings/__init__.py +7 -7
crawlo/settings/default_settings.py +226 -219
crawlo/settings/setting_manager.py +122 -122
crawlo/spider/__init__.py +639 -639
crawlo/stats_collector.py +59 -59
crawlo/subscriber.py +130 -130
crawlo/task_manager.py +30 -30
crawlo/templates/crawlo.cfg.tmpl +10 -10
crawlo/templates/project/__init__.py.tmpl +3 -3
crawlo/templates/project/items.py.tmpl +17 -17
crawlo/templates/project/middlewares.py.tmpl +118 -109
crawlo/templates/project/pipelines.py.tmpl +96 -96
crawlo/templates/project/run.py.tmpl +45 -45
crawlo/templates/project/settings.py.tmpl +327 -326
crawlo/templates/project/settings_distributed.py.tmpl +119 -119
crawlo/templates/project/settings_gentle.py.tmpl +94 -94
crawlo/templates/project/settings_high_performance.py.tmpl +151 -151
crawlo/templates/project/settings_simple.py.tmpl +68 -68
crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
crawlo/templates/spider/spider.py.tmpl +143 -141
crawlo/tools/__init__.py +182 -182
crawlo/tools/anti_crawler.py +268 -268
crawlo/tools/authenticated_proxy.py +240 -240
crawlo/tools/data_validator.py +180 -180
crawlo/tools/date_tools.py +35 -35
crawlo/tools/distributed_coordinator.py +386 -386
crawlo/tools/retry_mechanism.py +220 -220
crawlo/tools/scenario_adapter.py +262 -262
crawlo/utils/__init__.py +35 -35
crawlo/utils/batch_processor.py +260 -260
crawlo/utils/controlled_spider_mixin.py +439 -439
crawlo/utils/date_tools.py +290 -290
crawlo/utils/db_helper.py +343 -343
crawlo/utils/enhanced_error_handler.py +359 -359
crawlo/utils/env_config.py +105 -105
crawlo/utils/error_handler.py +125 -125
crawlo/utils/func_tools.py +82 -82
crawlo/utils/large_scale_config.py +286 -286
crawlo/utils/large_scale_helper.py +343 -343
crawlo/utils/log.py +128 -128
crawlo/utils/performance_monitor.py +284 -284
crawlo/utils/queue_helper.py +175 -175
crawlo/utils/redis_connection_pool.py +334 -334
crawlo/utils/redis_key_validator.py +199 -199
crawlo/utils/request.py +267 -267
crawlo/utils/request_serializer.py +219 -219
crawlo/utils/spider_loader.py +62 -62
crawlo/utils/system.py +11 -11
crawlo/utils/tools.py +4 -4
crawlo/utils/url.py +39 -39
crawlo-1.2.1.dist-info/METADATA +692 -0
crawlo-1.2.1.dist-info/RECORD +220 -0
examples/__init__.py +7 -7
examples/aiohttp_settings.py +42 -0
examples/curl_cffi_settings.py +41 -0
examples/default_header_middleware_example.py +107 -0
examples/default_header_spider_example.py +129 -0
examples/download_delay_middleware_example.py +160 -0
examples/httpx_settings.py +42 -0
examples/multi_downloader_proxy_example.py +81 -0
examples/offsite_middleware_example.py +55 -0
examples/offsite_spider_example.py +107 -0
examples/proxy_spider_example.py +166 -0
examples/request_ignore_middleware_example.py +51 -0
examples/request_ignore_spider_example.py +99 -0
examples/response_code_middleware_example.py +52 -0
examples/response_filter_middleware_example.py +67 -0
examples/tong_hua_shun_settings.py +62 -0
examples/tong_hua_shun_spider.py +170 -0
tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +81 -81
tests/__init__.py +7 -7
tests/advanced_tools_example.py +275 -275
tests/authenticated_proxy_example.py +236 -236
tests/cleaners_example.py +160 -160
tests/config_validation_demo.py +102 -102
tests/controlled_spider_example.py +205 -205
tests/date_tools_example.py +180 -180
tests/dynamic_loading_example.py +523 -523
tests/dynamic_loading_test.py +104 -104
tests/env_config_example.py +133 -133
tests/error_handling_example.py +171 -171
tests/redis_key_validation_demo.py +130 -130
tests/response_improvements_example.py +144 -144
tests/test_advanced_tools.py +148 -148
tests/test_all_redis_key_configs.py +145 -145
tests/test_authenticated_proxy.py +141 -141
tests/test_cleaners.py +54 -54
tests/test_comprehensive.py +146 -146
tests/test_config_validator.py +193 -193
tests/test_crawlo_proxy_integration.py +173 -0
tests/test_date_tools.py +123 -123
tests/test_default_header_middleware.py +159 -0
tests/test_double_crawlo_fix.py +207 -207
tests/test_double_crawlo_fix_simple.py +124 -124
tests/test_download_delay_middleware.py +222 -0
tests/test_downloader_proxy_compatibility.py +269 -0
tests/test_dynamic_downloaders_proxy.py +124 -124
tests/test_dynamic_proxy.py +92 -92
tests/test_dynamic_proxy_config.py +146 -146
tests/test_dynamic_proxy_real.py +109 -109
tests/test_edge_cases.py +303 -303
tests/test_enhanced_error_handler.py +270 -270
tests/test_env_config.py +121 -121
tests/test_error_handler_compatibility.py +112 -112
tests/test_final_validation.py +153 -153
tests/test_framework_env_usage.py +103 -103
tests/test_integration.py +356 -356
tests/test_item_dedup_redis_key.py +122 -122
tests/test_offsite_middleware.py +222 -0
tests/test_parsel.py +29 -29
tests/test_performance.py +327 -327
tests/test_proxy_api.py +265 -0
tests/test_proxy_health_check.py +32 -32
tests/test_proxy_middleware.py +122 -0
tests/test_proxy_middleware_enhanced.py +217 -0
tests/test_proxy_middleware_integration.py +136 -136
tests/test_proxy_providers.py +56 -56
tests/test_proxy_stats.py +19 -19
tests/test_proxy_strategies.py +59 -59
tests/test_queue_manager_double_crawlo.py +174 -231
tests/test_queue_manager_redis_key.py +176 -176
tests/test_real_scenario_proxy.py +196 -0
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_request_ignore_middleware.py +183 -0
tests/test_request_serialization.py +70 -70
tests/test_response_code_middleware.py +350 -0
tests/test_response_filter_middleware.py +428 -0
tests/test_response_improvements.py +152 -152
tests/test_retry_middleware.py +242 -0
tests/test_scheduler.py +241 -241
tests/test_simple_response.py +61 -61
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 +153 -153
tests/tools_example.py +257 -257
crawlo-1.1.9.dist-info/METADATA +0 -626
crawlo-1.1.9.dist-info/RECORD +0 -190
{crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/WHEEL +0 -0
{crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/entry_points.txt +0 -0
{crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/top_level.txt +0 -0

tests/test_response_improvements.py CHANGED Viewed

@@ -1,153 +1,153 @@
-#!/usr/bin/python
-# -*- coding:UTF-8 -*-
-"""
-Response 改进功能测试
-"""
-import unittest
-from crawlo.network.response import Response
-class TestResponseImprovements(unittest.TestCase):
-    """Response 改进功能测试类"""
-    def setUp(self):
-        """测试前准备"""
-        # 创建一个模拟的HTML响应
-        html_content = """
-        <html>
-        <head>
-            <title>测试页面</title>
-        </head>
-        <body>
-            <div class="content">
-                <h1>主标题</h1>
-                <p class="intro">这是介绍段落</p>
-                <ul class="list">
-                    <li>项目1</li>
-                    <li>项目2</li>
-                    <li>项目3</li>
-                </ul>
-                <a href="https://example.com" class="link">链接文本</a>
-                <img src="image.jpg" alt="图片描述" class="image">
-            </div>
-        </body>
-        </html>
-        """
-        self.response = Response(
-            url="https://example.com/test",
-            body=html_content.encode('utf-8'),
-            headers={"content-type": "text/html; charset=utf-8"}
-        )
-    def test_extract_text_with_css(self):
-        """测试使用CSS选择器提取文本"""
-        # 测试提取单个元素文本
-        title = self.response.extract_text('title')
-        self.assertEqual(title, "测试页面")
-        # 测试提取class元素文本
-        h1_text = self.response.extract_text('.content h1')
-        self.assertEqual(h1_text, "主标题")
-        # 测试提取带有默认值的情况
-        non_exist = self.response.extract_text('.non-exist', default='默认值')
-        self.assertEqual(non_exist, '默认值')
-    def test_extract_text_with_xpath(self):
-        """测试使用XPath选择器提取文本"""
-        # 测试提取单个元素文本
-        title = self.response.extract_text('//title')
-        self.assertEqual(title, "测试页面")
-        # 测试提取class元素文本
-        h1_text = self.response.extract_text('//div[@class="content"]/h1')
-        self.assertEqual(h1_text, "主标题")
-    def test_extract_texts_with_css(self):
-        """测试使用CSS选择器提取多个文本"""
-        # 测试提取多个li元素的文本
-        list_items = self.response.extract_texts('.list li')
-        expected = ["项目1", "项目2", "项目3"]
-        self.assertEqual(list_items, expected)
-        # 测试提取不存在元素的默认值
-        non_exist = self.response.extract_texts('.non-exist', default=['默认值'])
-        self.assertEqual(non_exist, ['默认值'])
-    def test_extract_texts_with_xpath(self):
-        """测试使用XPath选择器提取多个文本"""
-        # 测试提取多个li元素的文本
-        list_items = self.response.extract_texts('//ul[@class="list"]/li')
-        expected = ["项目1", "项目2", "项目3"]
-        self.assertEqual(list_items, expected)
-    def test_extract_attr(self):
-        """测试提取元素属性"""
-        # 测试提取链接的href属性
-        link_href = self.response.extract_attr('.link', 'href')
-        self.assertEqual(link_href, "https://example.com")
-        # 测试提取图片的alt属性
-        img_alt = self.response.extract_attr('.image', 'alt')
-        self.assertEqual(img_alt, "图片描述")
-        # 测试提取不存在属性的默认值
-        non_exist = self.response.extract_attr('.link', 'non-exist', default='默认值')
-        self.assertEqual(non_exist, '默认值')
-    def test_extract_attrs(self):
-        """测试提取多个元素的属性"""
-        # 测试提取所有li元素的属性（这里我们测试class属性）
-        list_classes = self.response.extract_attrs('.list li', 'class')
-        # 注意：在当前HTML中li元素没有class属性，所以应该返回空列表
-        self.assertEqual(list_classes, [])
-        # 测试提取所有图片元素的alt属性
-        img_alts = self.response.extract_attrs('.image', 'alt')
-        self.assertEqual(img_alts, ['图片描述'])
-        # 测试提取不存在元素时的默认值
-        non_exist = self.response.extract_attrs('.non-exist-elements', 'alt', default=['默认值'])
-        self.assertEqual(non_exist, ['默认值'])
-    def test_extract_text_from_elements(self):
-        """测试从复杂元素中提取文本"""
-        # 创建包含嵌套标签的HTML
-        complex_html = """
-        <div class="complex">
-            <p>段落文本 <strong>粗体文本</strong> 普通文本</p>
-            <p>第二段落 <em>斜体文本</em></p>
-        </div>
-        """
-        complex_response = Response(
-            url="https://example.com/complex",
-            body=complex_html.encode('utf-8')
-        )
-        # 测试提取复杂元素的文本
-        complex_text = complex_response.extract_text('.complex p', join_str=' ')
-        self.assertIn("段落文本", complex_text)
-        self.assertIn("粗体文本", complex_text)
-        self.assertIn("普通文本", complex_text)
-    def test_edge_cases(self):
-        """测试边界情况"""
-        # 测试空响应
-        empty_response = Response(url="https://example.com/empty", body=b"")
-        empty_text = empty_response.extract_text('title', default='默认标题')
-        self.assertEqual(empty_text, '默认标题')
-        # 测试只包含空白字符的元素
-        whitespace_html = "<div class='whitespace'>   </div>"
-        whitespace_response = Response(
-            url="https://example.com/whitespace",
-            body=whitespace_html.encode('utf-8')
-        )
-        whitespace_text = whitespace_response.extract_text('.whitespace')
-        self.assertEqual(whitespace_text, '')
-if __name__ == '__main__':
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+Response 改进功能测试
+"""
+import unittest
+from crawlo.network.response import Response
+class TestResponseImprovements(unittest.TestCase):
+    """Response 改进功能测试类"""
+    def setUp(self):
+        """测试前准备"""
+        # 创建一个模拟的HTML响应
+        html_content = """
+        <html>
+        <head>
+            <title>测试页面</title>
+        </head>
+        <body>
+            <div class="content">
+                <h1>主标题</h1>
+                <p class="intro">这是介绍段落</p>
+                <ul class="list">
+                    <li>项目1</li>
+                    <li>项目2</li>
+                    <li>项目3</li>
+                </ul>
+                <a href="https://example.com" class="link">链接文本</a>
+                <img src="image.jpg" alt="图片描述" class="image">
+            </div>
+        </body>
+        </html>
+        """
+        self.response = Response(
+            url="https://example.com/test",
+            body=html_content.encode('utf-8'),
+            headers={"content-type": "text/html; charset=utf-8"}
+        )
+    def test_extract_text_with_css(self):
+        """测试使用CSS选择器提取文本"""
+        # 测试提取单个元素文本
+        title = self.response.extract_text('title')
+        self.assertEqual(title, "测试页面")
+        # 测试提取class元素文本
+        h1_text = self.response.extract_text('.content h1')
+        self.assertEqual(h1_text, "主标题")
+        # 测试提取带有默认值的情况
+        non_exist = self.response.extract_text('.non-exist', default='默认值')
+        self.assertEqual(non_exist, '默认值')
+    def test_extract_text_with_xpath(self):
+        """测试使用XPath选择器提取文本"""
+        # 测试提取单个元素文本
+        title = self.response.extract_text('//title')
+        self.assertEqual(title, "测试页面")
+        # 测试提取class元素文本
+        h1_text = self.response.extract_text('//div[@class="content"]/h1')
+        self.assertEqual(h1_text, "主标题")
+    def test_extract_texts_with_css(self):
+        """测试使用CSS选择器提取多个文本"""
+        # 测试提取多个li元素的文本
+        list_items = self.response.extract_texts('.list li')
+        expected = ["项目1", "项目2", "项目3"]
+        self.assertEqual(list_items, expected)
+        # 测试提取不存在元素的默认值
+        non_exist = self.response.extract_texts('.non-exist', default=['默认值'])
+        self.assertEqual(non_exist, ['默认值'])
+    def test_extract_texts_with_xpath(self):
+        """测试使用XPath选择器提取多个文本"""
+        # 测试提取多个li元素的文本
+        list_items = self.response.extract_texts('//ul[@class="list"]/li')
+        expected = ["项目1", "项目2", "项目3"]
+        self.assertEqual(list_items, expected)
+    def test_extract_attr(self):
+        """测试提取元素属性"""
+        # 测试提取链接的href属性
+        link_href = self.response.extract_attr('.link', 'href')
+        self.assertEqual(link_href, "https://example.com")
+        # 测试提取图片的alt属性
+        img_alt = self.response.extract_attr('.image', 'alt')
+        self.assertEqual(img_alt, "图片描述")
+        # 测试提取不存在属性的默认值
+        non_exist = self.response.extract_attr('.link', 'non-exist', default='默认值')
+        self.assertEqual(non_exist, '默认值')
+    def test_extract_attrs(self):
+        """测试提取多个元素的属性"""
+        # 测试提取所有li元素的属性（这里我们测试class属性）
+        list_classes = self.response.extract_attrs('.list li', 'class')
+        # 注意：在当前HTML中li元素没有class属性，所以应该返回空列表
+        self.assertEqual(list_classes, [])
+        # 测试提取所有图片元素的alt属性
+        img_alts = self.response.extract_attrs('.image', 'alt')
+        self.assertEqual(img_alts, ['图片描述'])
+        # 测试提取不存在元素时的默认值
+        non_exist = self.response.extract_attrs('.non-exist-elements', 'alt', default=['默认值'])
+        self.assertEqual(non_exist, ['默认值'])
+    def test_extract_text_from_elements(self):
+        """测试从复杂元素中提取文本"""
+        # 创建包含嵌套标签的HTML
+        complex_html = """
+        <div class="complex">
+            <p>段落文本 <strong>粗体文本</strong> 普通文本</p>
+            <p>第二段落 <em>斜体文本</em></p>
+        </div>
+        """
+        complex_response = Response(
+            url="https://example.com/complex",
+            body=complex_html.encode('utf-8')
+        )
+        # 测试提取复杂元素的文本
+        complex_text = complex_response.extract_text('.complex p', join_str=' ')
+        self.assertIn("段落文本", complex_text)
+        self.assertIn("粗体文本", complex_text)
+        self.assertIn("普通文本", complex_text)
+    def test_edge_cases(self):
+        """测试边界情况"""
+        # 测试空响应
+        empty_response = Response(url="https://example.com/empty", body=b"")
+        empty_text = empty_response.extract_text('title', default='默认标题')
+        self.assertEqual(empty_text, '默认标题')
+        # 测试只包含空白字符的元素
+        whitespace_html = "<div class='whitespace'>   </div>"
+        whitespace_response = Response(
+            url="https://example.com/whitespace",
+            body=whitespace_html.encode('utf-8')
+        )
+        whitespace_text = whitespace_response.extract_text('.whitespace')
+        self.assertEqual(whitespace_text, '')
+if __name__ == '__main__':
     unittest.main()

tests/test_retry_middleware.py ADDED Viewed

@@ -0,0 +1,242 @@
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+RetryMiddleware 测试文件
+用于测试重试中间件的功能
+"""
+import unittest
+from unittest.mock import Mock, patch
+from crawlo.middleware.retry import RetryMiddleware
+from crawlo.settings.setting_manager import SettingManager
+class MockStats:
+    """Mock Stats 类，用于测试统计信息"""
+    def __init__(self):
+        self.stats = {}
+    def inc_value(self, key, value=1):
+        if key in self.stats:
+            self.stats[key] += value
+        else:
+            self.stats[key] = value
+class TestRetryMiddleware(unittest.TestCase):
+    """RetryMiddleware 测试类"""
+    def setUp(self):
+        """测试前准备"""
+        # 创建设置管理器
+        self.settings = SettingManager()
+        # 创建爬虫模拟对象
+        self.crawler = Mock()
+        self.crawler.settings = self.settings
+        self.crawler.stats = MockStats()
+    def test_middleware_initialization(self):
+        """测试中间件初始化"""
+        # 设置重试配置
+        self.settings.set('RETRY_HTTP_CODES', [500, 502, 503, 504, 408])
+        self.settings.set('IGNORE_HTTP_CODES', [404])
+        self.settings.set('MAX_RETRY_TIMES', 3)
+        self.settings.set('RETRY_EXCEPTIONS', [])
+        self.settings.set('RETRY_PRIORITY', -10)
+        # 应该正常创建实例
+        middleware = RetryMiddleware.create_instance(self.crawler)
+        self.assertIsInstance(middleware, RetryMiddleware)
+        self.assertEqual(middleware.max_retry_times, 3)
+        self.assertEqual(middleware.retry_priority, -10)
+    def test_process_response_with_retry_code(self):
+        """测试处理需要重试的响应码"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和响应对象
+        request = Mock()
+        request.meta = {}
+        request.priority = 0  # 添加priority属性
+        response = Mock()
+        response.status_code = 500
+        spider = Mock()
+        # 处理响应
+        result = middleware.process_response(request, response, spider)
+        # 应该返回重试的请求
+        self.assertEqual(result, request)
+        self.assertEqual(request.meta['retry_times'], 1)
+        self.assertTrue(request.meta['dont_retry'])
+        self.assertEqual(request.priority, -10)
+    def test_process_response_with_ignore_code(self):
+        """测试处理需要忽略的响应码"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和响应对象
+        request = Mock()
+        request.meta = {}
+        request.priority = 0  # 添加priority属性
+        response = Mock()
+        response.status_code = 404
+        spider = Mock()
+        # 处理响应
+        result = middleware.process_response(request, response, spider)
+        # 应该返回原始响应
+        self.assertEqual(result, response)
+    def test_process_response_with_dont_retry(self):
+        """测试处理带有dont_retry标记的响应"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和响应对象
+        request = Mock()
+        request.meta = {'dont_retry': True}
+        request.priority = 0  # 添加priority属性
+        response = Mock()
+        response.status_code = 500
+        spider = Mock()
+        # 处理响应
+        result = middleware.process_response(request, response, spider)
+        # 应该返回原始响应
+        self.assertEqual(result, response)
+    def test_process_response_with_max_retries_exceeded(self):
+        """测试超过最大重试次数的响应"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和响应对象
+        request = Mock()
+        request.meta = {'retry_times': 3}  # 已达到最大重试次数
+        request.priority = 0  # 添加priority属性
+        response = Mock()
+        response.status_code = 500
+        spider = Mock()
+        # 处理响应
+        result = middleware.process_response(request, response, spider)
+        # 应该返回原始响应
+        self.assertEqual(result, response)
+    def test_process_exception_with_retry_exception(self):
+        """测试处理需要重试的异常"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[ValueError],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和异常对象
+        request = Mock()
+        request.meta = {}
+        request.priority = 0  # 添加priority属性
+        exc = ValueError("test error")
+        spider = Mock()
+        # 处理异常
+        result = middleware.process_exception(request, exc, spider)
+        # 应该返回重试的请求
+        self.assertEqual(result, request)
+        self.assertEqual(request.meta['retry_times'], 1)
+        self.assertTrue(request.meta['dont_retry'])
+        self.assertEqual(request.priority, -10)
+    def test_process_exception_with_dont_retry(self):
+        """测试处理带有dont_retry标记的异常"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[ValueError],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和异常对象
+        request = Mock()
+        request.meta = {'dont_retry': True}
+        request.priority = 0  # 添加priority属性
+        exc = ValueError("test error")
+        spider = Mock()
+        # 处理异常
+        result = middleware.process_exception(request, exc, spider)
+        # 应该返回None
+        self.assertIsNone(result)
+    def test_process_exception_with_non_retry_exception(self):
+        """测试处理不需要重试的异常"""
+        # 创建中间件实例
+        middleware = RetryMiddleware(
+            retry_http_codes=[500, 502, 503, 504, 408],
+            ignore_http_codes=[404],
+            max_retry_times=3,
+            retry_exceptions=[ValueError],
+            stats=MockStats(),
+            retry_priority=-10
+        )
+        # 创建请求和异常对象
+        request = Mock()
+        request.meta = {}
+        request.priority = 0  # 添加priority属性
+        exc = TypeError("test error")  # 不在重试异常列表中
+        spider = Mock()
+        # 处理异常
+        result = middleware.process_exception(request, exc, spider)
+        # 应该返回None
+        self.assertIsNone(result)
+if __name__ == '__main__':
+    unittest.main()

crawlo 1.1.9__py3-none-any.whl → 1.2.1__py3-none-any.whl

Potentially problematic release.

crawlo 1.1.9py3-none-any.whl → 1.2.1py3-none-any.whl