crawlo 1.1.1__py3-none-any.whl → 1.1.3__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 (128) hide show
  1. crawlo/__init__.py +34 -33
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +40 -40
  4. crawlo/commands/__init__.py +13 -13
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +152 -126
  7. crawlo/commands/list.py +156 -147
  8. crawlo/commands/run.py +285 -285
  9. crawlo/commands/startproject.py +196 -111
  10. crawlo/commands/stats.py +188 -188
  11. crawlo/commands/utils.py +187 -0
  12. crawlo/config.py +280 -0
  13. crawlo/core/__init__.py +2 -2
  14. crawlo/core/engine.py +171 -158
  15. crawlo/core/enhanced_engine.py +190 -0
  16. crawlo/core/processor.py +40 -40
  17. crawlo/core/scheduler.py +166 -57
  18. crawlo/crawler.py +1028 -495
  19. crawlo/downloader/__init__.py +242 -78
  20. crawlo/downloader/aiohttp_downloader.py +212 -199
  21. crawlo/downloader/cffi_downloader.py +251 -241
  22. crawlo/downloader/httpx_downloader.py +259 -246
  23. crawlo/event.py +11 -11
  24. crawlo/exceptions.py +82 -78
  25. crawlo/extension/__init__.py +31 -31
  26. crawlo/extension/log_interval.py +49 -49
  27. crawlo/extension/log_stats.py +44 -44
  28. crawlo/extension/logging_extension.py +34 -34
  29. crawlo/filters/__init__.py +154 -37
  30. crawlo/filters/aioredis_filter.py +242 -150
  31. crawlo/filters/memory_filter.py +269 -202
  32. crawlo/items/__init__.py +23 -23
  33. crawlo/items/base.py +21 -21
  34. crawlo/items/fields.py +53 -53
  35. crawlo/items/items.py +104 -104
  36. crawlo/middleware/__init__.py +21 -21
  37. crawlo/middleware/default_header.py +32 -32
  38. crawlo/middleware/download_delay.py +28 -28
  39. crawlo/middleware/middleware_manager.py +135 -135
  40. crawlo/middleware/proxy.py +248 -245
  41. crawlo/middleware/request_ignore.py +30 -30
  42. crawlo/middleware/response_code.py +18 -18
  43. crawlo/middleware/response_filter.py +26 -26
  44. crawlo/middleware/retry.py +125 -90
  45. crawlo/mode_manager.py +201 -0
  46. crawlo/network/__init__.py +21 -7
  47. crawlo/network/request.py +311 -203
  48. crawlo/network/response.py +271 -166
  49. crawlo/pipelines/__init__.py +22 -13
  50. crawlo/pipelines/bloom_dedup_pipeline.py +157 -0
  51. crawlo/pipelines/console_pipeline.py +39 -39
  52. crawlo/pipelines/csv_pipeline.py +317 -0
  53. crawlo/pipelines/database_dedup_pipeline.py +225 -0
  54. crawlo/pipelines/json_pipeline.py +219 -0
  55. crawlo/pipelines/memory_dedup_pipeline.py +116 -0
  56. crawlo/pipelines/mongo_pipeline.py +116 -116
  57. crawlo/pipelines/mysql_pipeline.py +195 -195
  58. crawlo/pipelines/pipeline_manager.py +56 -56
  59. crawlo/pipelines/redis_dedup_pipeline.py +163 -0
  60. crawlo/project.py +153 -153
  61. crawlo/queue/__init__.py +0 -0
  62. crawlo/queue/pqueue.py +37 -0
  63. crawlo/queue/queue_manager.py +308 -0
  64. crawlo/queue/redis_priority_queue.py +209 -0
  65. crawlo/settings/__init__.py +7 -7
  66. crawlo/settings/default_settings.py +245 -167
  67. crawlo/settings/setting_manager.py +99 -99
  68. crawlo/spider/__init__.py +639 -129
  69. crawlo/stats_collector.py +59 -59
  70. crawlo/subscriber.py +106 -106
  71. crawlo/task_manager.py +30 -27
  72. crawlo/templates/crawlo.cfg.tmpl +10 -10
  73. crawlo/templates/project/__init__.py.tmpl +3 -3
  74. crawlo/templates/project/items.py.tmpl +17 -17
  75. crawlo/templates/project/middlewares.py.tmpl +87 -76
  76. crawlo/templates/project/pipelines.py.tmpl +342 -64
  77. crawlo/templates/project/run.py.tmpl +252 -0
  78. crawlo/templates/project/settings.py.tmpl +251 -54
  79. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  80. crawlo/templates/spider/spider.py.tmpl +178 -32
  81. crawlo/utils/__init__.py +7 -7
  82. crawlo/utils/controlled_spider_mixin.py +440 -0
  83. crawlo/utils/date_tools.py +233 -233
  84. crawlo/utils/db_helper.py +343 -343
  85. crawlo/utils/func_tools.py +82 -82
  86. crawlo/utils/large_scale_config.py +287 -0
  87. crawlo/utils/large_scale_helper.py +344 -0
  88. crawlo/utils/log.py +128 -128
  89. crawlo/utils/queue_helper.py +176 -0
  90. crawlo/utils/request.py +267 -267
  91. crawlo/utils/request_serializer.py +220 -0
  92. crawlo/utils/spider_loader.py +62 -62
  93. crawlo/utils/system.py +11 -11
  94. crawlo/utils/tools.py +4 -4
  95. crawlo/utils/url.py +39 -39
  96. crawlo-1.1.3.dist-info/METADATA +635 -0
  97. crawlo-1.1.3.dist-info/RECORD +113 -0
  98. examples/__init__.py +7 -7
  99. examples/controlled_spider_example.py +205 -0
  100. tests/__init__.py +7 -7
  101. tests/test_final_validation.py +154 -0
  102. tests/test_proxy_health_check.py +32 -32
  103. tests/test_proxy_middleware_integration.py +136 -136
  104. tests/test_proxy_providers.py +56 -56
  105. tests/test_proxy_stats.py +19 -19
  106. tests/test_proxy_strategies.py +59 -59
  107. tests/test_redis_config.py +29 -0
  108. tests/test_redis_queue.py +225 -0
  109. tests/test_request_serialization.py +71 -0
  110. tests/test_scheduler.py +242 -0
  111. crawlo/pipelines/mysql_batch_pipline.py +0 -273
  112. crawlo/utils/pqueue.py +0 -174
  113. crawlo-1.1.1.dist-info/METADATA +0 -220
  114. crawlo-1.1.1.dist-info/RECORD +0 -100
  115. examples/baidu_spider/__init__.py +0 -7
  116. examples/baidu_spider/demo.py +0 -94
  117. examples/baidu_spider/items.py +0 -46
  118. examples/baidu_spider/middleware.py +0 -49
  119. examples/baidu_spider/pipeline.py +0 -55
  120. examples/baidu_spider/run.py +0 -27
  121. examples/baidu_spider/settings.py +0 -121
  122. examples/baidu_spider/spiders/__init__.py +0 -7
  123. examples/baidu_spider/spiders/bai_du.py +0 -61
  124. examples/baidu_spider/spiders/miit.py +0 -159
  125. examples/baidu_spider/spiders/sina.py +0 -79
  126. {crawlo-1.1.1.dist-info → crawlo-1.1.3.dist-info}/WHEEL +0 -0
  127. {crawlo-1.1.1.dist-info → crawlo-1.1.3.dist-info}/entry_points.txt +0 -0
  128. {crawlo-1.1.1.dist-info → crawlo-1.1.3.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,209 @@
1
+ import pickle
2
+ import time
3
+ import asyncio
4
+ from typing import Optional
5
+ import redis.asyncio as aioredis
6
+ import traceback
7
+ import os
8
+
9
+ from crawlo import Request
10
+ from crawlo.utils.log import get_logger
11
+ from crawlo.utils.request_serializer import RequestSerializer
12
+
13
+
14
+ logger = get_logger(__name__)
15
+
16
+
17
+ class RedisPriorityQueue:
18
+ """
19
+ 基于 Redis 的分布式异步优先级队列(生产级优化版)
20
+ """
21
+
22
+ def __init__(
23
+ self,
24
+ redis_url: str = None,
25
+ queue_name: str = "crawlo:requests",
26
+ processing_queue: str = "crawlo:processing",
27
+ failed_queue: str = "crawlo:failed",
28
+ max_retries: int = 3,
29
+ timeout: int = 300, # 任务处理超时时间(秒)
30
+ max_connections: int = 10, # 连接池大小
31
+ ):
32
+ # 如果没有提供 redis_url,则从环境变量构造
33
+ if redis_url is None:
34
+ redis_host = os.getenv('REDIS_HOST', 'localhost')
35
+ redis_port = os.getenv('REDIS_PORT', '6379')
36
+ redis_db = os.getenv('REDIS_DB', '0')
37
+ redis_password = os.getenv('REDIS_PASSWORD', '')
38
+
39
+ if redis_password:
40
+ redis_url = f"redis://:{redis_password}@{redis_host}:{redis_port}/{redis_db}"
41
+ else:
42
+ redis_url = f"redis://{redis_host}:{redis_port}/{redis_db}"
43
+
44
+ self.redis_url = redis_url
45
+ self.queue_name = queue_name
46
+ self.processing_queue = processing_queue
47
+ self.failed_queue = failed_queue
48
+ self.max_retries = max_retries
49
+ self.timeout = timeout
50
+ self.max_connections = max_connections
51
+ self._redis = None
52
+ self._lock = asyncio.Lock() # 用于连接初始化的锁
53
+ self.request_serializer = RequestSerializer() # 处理序列化
54
+
55
+ async def connect(self, max_retries=3, delay=1):
56
+ """异步连接 Redis,支持重试"""
57
+ async with self._lock:
58
+ if self._redis is not None:
59
+ return self._redis
60
+
61
+ for attempt in range(max_retries):
62
+ try:
63
+ self._redis = await aioredis.from_url(
64
+ self.redis_url,
65
+ decode_responses=False, # pickle 需要 bytes
66
+ max_connections=self.max_connections,
67
+ socket_connect_timeout=5,
68
+ socket_timeout=30,
69
+ )
70
+ # 测试连接
71
+ await self._redis.ping()
72
+ logger.info("✅ Redis 连接成功")
73
+ return self._redis
74
+ except Exception as e:
75
+ logger.warning(f"⚠️ Redis 连接失败 (尝试 {attempt + 1}/{max_retries}): {e}")
76
+ logger.warning(f"详细错误信息:\n{traceback.format_exc()}")
77
+ if attempt < max_retries - 1:
78
+ await asyncio.sleep(delay)
79
+ else:
80
+ raise ConnectionError(f"❌ 无法连接 Redis: {e}")
81
+
82
+ async def _ensure_connection(self):
83
+ """确保连接有效"""
84
+ if self._redis is None:
85
+ await self.connect()
86
+ try:
87
+ await self._redis.ping()
88
+ except Exception as e:
89
+ logger.warning(f"🔄 Redis 连接失效,尝试重连...: {e}")
90
+ self._redis = None
91
+ await self.connect()
92
+
93
+ async def put(self, request: Request, priority: int = 0) -> bool:
94
+ """放入请求到队列"""
95
+ await self._ensure_connection()
96
+ score = -priority
97
+ key = self._get_request_key(request)
98
+ try:
99
+ # 🔥 使用专用的序列化工具清理 Request
100
+ clean_request = self.request_serializer.prepare_for_serialization(request)
101
+
102
+ serialized = pickle.dumps(clean_request)
103
+ pipe = self._redis.pipeline()
104
+ pipe.zadd(self.queue_name, {key: score})
105
+ pipe.hset(f"{self.queue_name}:data", key, serialized)
106
+ result = await pipe.execute()
107
+
108
+ if result[0] > 0:
109
+ logger.debug(f"✅ 成功入队: {request.url}")
110
+ return result[0] > 0
111
+ except Exception as e:
112
+ logger.error(f"❌ 放入队列失败: {e}")
113
+ logger.error(f"详细错误信息:\n{traceback.format_exc()}")
114
+ return False
115
+
116
+ async def get(self, timeout: float = 5.0) -> Optional[Request]:
117
+ """
118
+ 获取请求(带超时)
119
+ :param timeout: 最大等待时间(秒),避免无限轮询
120
+ """
121
+ await self._ensure_connection()
122
+ start_time = asyncio.get_event_loop().time()
123
+
124
+ while True:
125
+ try:
126
+ # 尝试获取任务
127
+ result = await self._redis.zpopmin(self.queue_name, count=1)
128
+ if result:
129
+ key, score = result[0]
130
+ serialized = await self._redis.hget(f"{self.queue_name}:data", key)
131
+ if not serialized:
132
+ continue
133
+
134
+ # 移动到 processing
135
+ processing_key = f"{key}:{int(time.time())}"
136
+ pipe = self._redis.pipeline()
137
+ pipe.zadd(self.processing_queue, {processing_key: time.time() + self.timeout})
138
+ pipe.hset(f"{self.processing_queue}:data", processing_key, serialized)
139
+ pipe.hdel(f"{self.queue_name}:data", key)
140
+ await pipe.execute()
141
+
142
+ return pickle.loads(serialized)
143
+
144
+ # 检查是否超时
145
+ if asyncio.get_event_loop().time() - start_time > timeout:
146
+ return None
147
+
148
+ # 短暂等待,避免空轮询
149
+ await asyncio.sleep(0.1)
150
+
151
+ except Exception as e:
152
+ logger.error(f"❌ 获取队列任务失败: {e}")
153
+ logger.error(f"详细错误信息:\n{traceback.format_exc()}")
154
+ return None
155
+
156
+ async def ack(self, request: Request):
157
+ """确认任务完成"""
158
+ await self._ensure_connection()
159
+ key = self._get_request_key(request)
160
+ cursor = 0
161
+ while True:
162
+ cursor, keys = await self._redis.zscan(self.processing_queue, cursor, match=f"{key}:*")
163
+ if keys:
164
+ pipe = self._redis.pipeline()
165
+ for k in keys:
166
+ pipe.zrem(self.processing_queue, k)
167
+ pipe.hdel(f"{self.processing_queue}:data", k)
168
+ await pipe.execute()
169
+ if cursor == 0:
170
+ break
171
+
172
+ async def fail(self, request: Request, reason: str = ""):
173
+ """标记任务失败"""
174
+ await self._ensure_connection()
175
+ key = self._get_request_key(request)
176
+ await self.ack(request)
177
+
178
+ retry_key = f"{self.failed_queue}:retries:{key}"
179
+ retries = await self._redis.incr(retry_key)
180
+ await self._redis.expire(retry_key, 86400)
181
+
182
+ if retries <= self.max_retries:
183
+ await self.put(request, priority=request.priority + 1)
184
+ logger.info(f"🔁 任务重试 [{retries}/{self.max_retries}]: {request.url}")
185
+ else:
186
+ failed_data = {
187
+ "url": request.url,
188
+ "reason": reason,
189
+ "retries": retries,
190
+ "failed_at": time.time(),
191
+ "request_pickle": pickle.dumps(request).hex(), # 可选:保存完整请求
192
+ }
193
+ await self._redis.lpush(self.failed_queue, pickle.dumps(failed_data))
194
+ logger.error(f"❌ 任务彻底失败 [{retries}次]: {request.url}")
195
+
196
+ def _get_request_key(self, request: Request) -> str:
197
+ """生成请求唯一键"""
198
+ return f"url:{hash(request.url)}"
199
+
200
+ async def qsize(self) -> int:
201
+ """获取队列大小"""
202
+ await self._ensure_connection()
203
+ return await self._redis.zcard(self.queue_name)
204
+
205
+ async def close(self):
206
+ """关闭连接"""
207
+ if self._redis:
208
+ await self._redis.close()
209
+ self._redis = None
@@ -1,7 +1,7 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- """
4
- # @Time : 2025-05-11 11:08
5
- # @Author : oscar
6
- # @Desc : None
7
- """
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ """
4
+ # @Time : 2025-05-11 11:08
5
+ # @Author : oscar
6
+ # @Desc : None
7
+ """
@@ -1,167 +1,245 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- ==================================
5
- Crawlo 项目配置文件
6
- ==================================
7
- 说明:
8
- - 所有配置项均已按功能模块分类。
9
- - 支持通过环境变量覆盖部分敏感配置(如 Redis、MySQL 密码等)。
10
- - 可根据需求启用/禁用组件(如 MySQL、Redis、Proxy 等)。
11
- """
12
- import os
13
-
14
- # ============================== 核心信息 ==============================
15
- PROJECT_NAME = 'crawlo'
16
-
17
- # ============================== 网络请求配置 ==============================
18
-
19
- # 下载器选择(三选一)
20
- # DOWNLOADER = "crawlo.downloader.aiohttp_downloader.AioHttpDownloader"
21
- DOWNLOADER = "crawlo.downloader.cffi_downloader.CurlCffiDownloader" # 支持浏览器指纹
22
- # DOWNLOADER = "crawlo.downloader.httpx_downloader.HttpXDownloader"
23
-
24
- # 请求超时与安全
25
- DOWNLOAD_TIMEOUT = 30 # 下载超时时间(秒)
26
- VERIFY_SSL = True # 是否验证 SSL 证书
27
- USE_SESSION = True # 是否使用持久化会话(aiohttp 特有)
28
-
29
- # 请求延迟控制
30
- DOWNLOAD_DELAY = 1.0 # 基础延迟(秒)
31
- RANDOM_RANGE = (0.8, 1.2) # 随机延迟系数范围
32
- RANDOMNESS = True # 是否启用随机延迟
33
-
34
- # 重试策略
35
- MAX_RETRY_TIMES = 3 # 最大重试次数
36
- RETRY_PRIORITY = -1 # 重试请求的优先级调整
37
- RETRY_HTTP_CODES = [408, 429, 500, 502, 503, 504, 522, 524] # 触发重试的状态码
38
- IGNORE_HTTP_CODES = [403, 404] # 直接标记成功、不重试的状态码
39
- ALLOWED_CODES = [] # 允许的状态码(空表示不限制)
40
-
41
- # 连接与响应大小限制
42
- CONNECTION_POOL_LIMIT = 50 # 最大并发连接数(连接池大小)
43
- DOWNLOAD_MAXSIZE = 10 * 1024 * 1024 # 最大响应体大小(10MB)
44
- DOWNLOAD_WARN_SIZE = 1024 * 1024 # 响应体警告阈值(1MB)
45
- DOWNLOAD_RETRY_TIMES = MAX_RETRY_TIMES # 下载器内部重试次数(复用全局)
46
-
47
- # ============================== 并发与调度 ==============================
48
-
49
- CONCURRENCY = 8 # 单个爬虫的并发请求数
50
- INTERVAL = 5 # 日志统计输出间隔(秒)
51
- DEPTH_PRIORITY = 1 # 深度优先策略优先级
52
- MAX_RUNNING_SPIDERS = 3 # 最大同时运行的爬虫数
53
-
54
- # ============================== 数据存储配置 ==============================
55
-
56
- # --- MySQL 配置 ---
57
- MYSQL_HOST = '127.0.0.1'
58
- MYSQL_PORT = 3306
59
- MYSQL_USER = 'root'
60
- MYSQL_PASSWORD = '123456'
61
- MYSQL_DB = 'crawl'
62
- MYSQL_TABLE = 'crawlo'
63
- MYSQL_BATCH_SIZE = 100 # 批量插入条数
64
-
65
- # MySQL 连接池
66
- MYSQL_FLUSH_INTERVAL = 5 # 缓存刷新间隔(秒)
67
- MYSQL_POOL_MIN = 5
68
- MYSQL_POOL_MAX = 20
69
- MYSQL_ECHO = False # 是否打印 SQL 日志
70
-
71
- # --- MongoDB 配置 ---
72
- MONGO_URI = 'mongodb://user:password@host:27017'
73
- MONGO_DATABASE = 'scrapy_data'
74
- MONGO_COLLECTION = 'crawled_items'
75
- MONGO_MAX_POOL_SIZE = 200
76
- MONGO_MIN_POOL_SIZE = 20
77
-
78
- # ============================== 去重过滤配置 ==============================
79
-
80
- # 请求指纹存储目录(文件过滤器使用)
81
- REQUEST_DIR = '.'
82
-
83
- # 去重过滤器类(二选一)
84
- FILTER_CLASS = 'crawlo.filters.memory_filter.MemoryFilter'
85
- # FILTER_CLASS = 'crawlo.filters.redis_filter.AioRedisFilter' # 分布式去重
86
-
87
- # --- Redis 过滤器配置 ---
88
- REDIS_HOST = os.getenv('REDIS_HOST', '127.0.0.1')
89
- REDIS_PORT = int(os.getenv('REDIS_PORT', 6379))
90
- REDIS_PASSWORD = os.getenv('REDIS_PASSWORD', 'oscar&0503')
91
- REDIS_URL = f'redis://:{REDIS_PASSWORD or ""}@{REDIS_HOST}:{REDIS_PORT}/0'
92
- REDIS_KEY = 'request_fingerprint' # Redis 中存储指纹的键名
93
- REDIS_TTL = 0 # 指纹过期时间(0 表示永不过期)
94
- CLEANUP_FP = 0 # 程序结束时是否清理指纹(0=不清理)
95
- FILTER_DEBUG = True # 是否开启去重调试日志
96
- DECODE_RESPONSES = True # Redis 返回是否解码为字符串
97
-
98
- # ============================== 中间件配置 ==============================
99
-
100
- MIDDLEWARES = [
101
- # === 请求预处理阶段 ===
102
- 'crawlo.middleware.request_ignore.RequestIgnoreMiddleware', # 1. 忽略无效请求
103
- 'crawlo.middleware.download_delay.DownloadDelayMiddleware', # 2. 控制请求频率
104
- 'crawlo.middleware.default_header.DefaultHeaderMiddleware', # 3. 添加默认请求头
105
- 'crawlo.middleware.proxy.ProxyMiddleware', # 4. 设置代理
106
-
107
- # === 响应处理阶段 ===
108
- 'crawlo.middleware.retry.RetryMiddleware', # 5. 失败请求重试
109
- 'crawlo.middleware.response_code.ResponseCodeMiddleware', # 6. 处理特殊状态码
110
- 'crawlo.middleware.response_filter.ResponseFilterMiddleware', # 7. 响应内容过滤
111
- ]
112
-
113
- # ============================== 扩展与管道 ==============================
114
-
115
- # 数据处理管道(启用的存储方式)
116
- PIPELINES = [
117
- 'crawlo.pipelines.console_pipeline.ConsolePipeline', # 控制台输出
118
- # 'crawlo.pipelines.mysql_pipeline.AsyncmyMySQLPipeline', # MySQL 存储(可选)
119
- ]
120
-
121
- # 扩展组件(监控与日志)
122
- EXTENSIONS = [
123
- 'crawlo.extension.log_interval.LogIntervalExtension', # 定时日志
124
- 'crawlo.extension.log_stats.LogStats', # 统计信息
125
- 'crawlo.extension.logging_extension.CustomLoggerExtension', # 自定义日志
126
- ]
127
-
128
- # ============================== 日志与监控 ==============================
129
-
130
- LOG_LEVEL = 'INFO' # 日志级别: DEBUG/INFO/WARNING/ERROR
131
- STATS_DUMP = True # 是否周期性输出统计信息
132
- LOG_FILE = f'logs/{PROJECT_NAME}.log' # 日志文件路径
133
- LOG_FORMAT = '%(asctime)s - [%(name)s] - %(levelname)s: %(message)s'
134
- LOG_ENCODING = 'utf-8'
135
-
136
- # ============================== 代理配置 ==============================
137
-
138
- PROXY_ENABLED = False # 是否启用代理
139
- PROXY_API_URL = "https://api.proxyprovider.com/get" # 代理获取接口(请替换为真实地址)
140
-
141
- # 代理提取方式(支持字段路径或函数)
142
- PROXY_EXTRACTOR = "proxy" # 如返回 {"proxy": "http://1.1.1.1:8080"}
143
-
144
- # 代理刷新控制
145
- PROXY_REFRESH_INTERVAL = 60 # 代理刷新间隔(秒)
146
- PROXY_API_TIMEOUT = 10 # 请求代理 API 超时时间
147
-
148
- # ============================== Curl-Cffi 特有配置 ==============================
149
-
150
- # 浏览器指纹模拟(仅 CurlCffi 下载器有效)
151
- CURL_BROWSER_TYPE = "chrome" # 可选: chrome, edge, safari, firefox 或版本如 chrome136
152
-
153
- # 自定义浏览器版本映射(可覆盖默认行为)
154
- CURL_BROWSER_VERSION_MAP = {
155
- "chrome": "chrome136",
156
- "edge": "edge101",
157
- "safari": "safari184",
158
- "firefox": "firefox135",
159
- # 示例:旧版本测试
160
- # "chrome_legacy": "chrome110",
161
- }
162
-
163
- # 默认请求头(可被 Spider 覆盖)
164
- DEFAULT_REQUEST_HEADERS = {
165
- 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
166
- '(KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
167
- }
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ ==================================
5
+ Crawlo 项目配置文件
6
+ ==================================
7
+ 说明:
8
+ - 所有配置项均已按功能模块分类。
9
+ - 支持通过环境变量覆盖部分敏感配置(如 Redis、MySQL 密码等)。
10
+ - 可根据需求启用/禁用组件(如 MySQL、Redis、Proxy 等)。
11
+ """
12
+ import os
13
+
14
+ # ============================== 核心信息 ==============================
15
+ PROJECT_NAME = 'crawlo'
16
+
17
+ # ============================== 网络请求配置 ==============================
18
+
19
+ # 下载器选择(支持三种方式)
20
+ # 方式1: 直接指定类路径
21
+ DOWNLOADER = "crawlo.downloader.aiohttp_downloader.AioHttpDownloader"
22
+ # DOWNLOADER = "crawlo.downloader.cffi_downloader.CurlCffiDownloader" # 支持浏览器指纹
23
+ # DOWNLOADER = "crawlo.downloader.httpx_downloader.HttpXDownloader" # 支持HTTP/2
24
+
25
+ # 方式2: 使用简化名称(推荐)
26
+ # DOWNLOADER_TYPE = 'aiohttp' # 可选: aiohttp, httpx, curl_cffi, cffi
27
+
28
+ # 方式3: 在Spider中动态选择
29
+ # 可以在Spider类中设置 custom_settings = {'DOWNLOADER_TYPE': 'httpx'}
30
+
31
+ # 请求超时与安全
32
+ DOWNLOAD_TIMEOUT = 30 # 下载超时时间(秒)
33
+ VERIFY_SSL = True # 是否验证 SSL 证书
34
+ USE_SESSION = True # 是否使用持久化会话(aiohttp 特有)
35
+
36
+ # 请求延迟控制
37
+ DOWNLOAD_DELAY = 1.0 # 基础延迟(秒)
38
+ RANDOM_RANGE = (0.8, 1.2) # 随机延迟系数范围
39
+ RANDOMNESS = True # 是否启用随机延迟
40
+
41
+ # 重试策略
42
+ MAX_RETRY_TIMES = 3 # 最大重试次数
43
+ RETRY_PRIORITY = -1 # 重试请求的优先级调整
44
+ RETRY_HTTP_CODES = [408, 429, 500, 502, 503, 504, 522, 524] # 触发重试的状态码
45
+ IGNORE_HTTP_CODES = [403, 404] # 直接标记成功、不重试的状态码
46
+ ALLOWED_CODES = [] # 允许的状态码(空表示不限制)
47
+
48
+ # 连接与响应大小限制
49
+ CONNECTION_POOL_LIMIT = 50 # 最大并发连接数(连接池大小)
50
+ CONNECTION_POOL_LIMIT_PER_HOST = 20 # 每个主机的连接池大小
51
+ DOWNLOAD_MAXSIZE = 10 * 1024 * 1024 # 最大响应体大小(10MB)
52
+ DOWNLOAD_WARN_SIZE = 1024 * 1024 # 响应体警告阈值(1MB)
53
+ DOWNLOAD_RETRY_TIMES = MAX_RETRY_TIMES # 下载器内部重试次数(复用全局)
54
+
55
+ # 下载统计配置
56
+ DOWNLOADER_STATS = True # 是否启用下载器统计功能
57
+ DOWNLOAD_STATS = True # 是否记录下载时间和大小统计
58
+
59
+ # ============================== 并发与调度 ==============================
60
+
61
+ CONCURRENCY = 8 # 单个爬虫的并发请求数
62
+ INTERVAL = 5 # 日志统计输出间隔(秒)
63
+ DEPTH_PRIORITY = 1 # 深度优先策略优先级
64
+ MAX_RUNNING_SPIDERS = 3 # 最大同时运行的爬虫数
65
+
66
+ # ============================== 队列配置 ==============================
67
+
68
+ # 🎯 运行模式选择:'standalone'(单机), 'distributed'(分布式), 'auto'(自动检测)
69
+ RUN_MODE = 'standalone' # 默认单机模式,简单易用
70
+
71
+ # 队列类型选择:'memory'(内存), 'redis'(分布式), 'auto'(自动选择)
72
+ QUEUE_TYPE = 'memory' # 默认内存队列,无需外部依赖
73
+ SCHEDULER_MAX_QUEUE_SIZE = 2000 # 调度器队列最大容量
74
+ SCHEDULER_QUEUE_NAME = 'crawlo:requests' # Redis 队列名称
75
+ QUEUE_MAX_RETRIES = 3 # 队列操作最大重试次数
76
+ QUEUE_TIMEOUT = 300 # 队列操作超时时间(秒)
77
+
78
+ # 大规模爬取优化配置
79
+ LARGE_SCALE_BATCH_SIZE = 1000 # 批处理大小
80
+ LARGE_SCALE_CHECKPOINT_INTERVAL = 5000 # 进度保存间隔
81
+ LARGE_SCALE_MAX_MEMORY_USAGE = 500 # 最大内存使用量(MB)
82
+
83
+ # ============================== 数据存储配置 ==============================
84
+
85
+ # --- MySQL 配置 ---
86
+ MYSQL_HOST = '127.0.0.1'
87
+ MYSQL_PORT = 3306
88
+ MYSQL_USER = 'root'
89
+ MYSQL_PASSWORD = '123456'
90
+ MYSQL_DB = 'crawl'
91
+ MYSQL_TABLE = 'crawlo'
92
+ MYSQL_BATCH_SIZE = 100 # 批量插入条数
93
+
94
+ # MySQL 连接池
95
+ MYSQL_FLUSH_INTERVAL = 5 # 缓存刷新间隔(秒)
96
+ MYSQL_POOL_MIN = 5
97
+ MYSQL_POOL_MAX = 20
98
+ MYSQL_ECHO = False # 是否打印 SQL 日志
99
+
100
+ # --- MongoDB 配置 ---
101
+ MONGO_URI = 'mongodb://user:password@host:27017'
102
+ MONGO_DATABASE = 'scrapy_data'
103
+ MONGO_COLLECTION = 'crawled_items'
104
+ MONGO_MAX_POOL_SIZE = 200
105
+ MONGO_MIN_POOL_SIZE = 20
106
+
107
+ # ============================== 去重过滤配置 ==============================
108
+
109
+ # 请求指纹存储目录(文件过滤器使用)
110
+ REQUEST_DIR = '.'
111
+
112
+ # 根据运行模式自动选择去重管道
113
+ # 单机模式默认使用内存去重管道
114
+ # 分布式模式默认使用Redis去重管道
115
+ if RUN_MODE == 'distributed':
116
+ # 分布式模式下默认使用Redis去重管道
117
+ DEFAULT_DEDUP_PIPELINE = 'crawlo.pipelines.RedisDedupPipeline'
118
+ else:
119
+ # 单机模式下默认使用内存去重管道
120
+ DEFAULT_DEDUP_PIPELINE = 'crawlo.pipelines.MemoryDedupPipeline'
121
+
122
+ # 去重过滤器类(二选一)
123
+ FILTER_CLASS = 'crawlo.filters.memory_filter.MemoryFilter'
124
+ # FILTER_CLASS = 'crawlo.filters.aioredis_filter.AioRedisFilter' # 分布式去重
125
+
126
+ # --- Redis 过滤器配置 ---
127
+ REDIS_HOST = os.getenv('REDIS_HOST', '127.0.0.1')
128
+ REDIS_PORT = int(os.getenv('REDIS_PORT', 6379))
129
+ REDIS_PASSWORD = os.getenv('REDIS_PASSWORD', '') # 默认无密码
130
+ REDIS_DB = int(os.getenv('REDIS_DB', 0)) # Redis 数据库编号,默认为 0
131
+ # 🔧 根据是否有密码生成不同的 URL 格式
132
+ if REDIS_PASSWORD:
133
+ REDIS_URL = f'redis://:{REDIS_PASSWORD}@{REDIS_HOST}:{REDIS_PORT}/{REDIS_DB}'
134
+ else:
135
+ REDIS_URL = f'redis://{REDIS_HOST}:{REDIS_PORT}/{REDIS_DB}'
136
+ REDIS_KEY = 'request_fingerprint' # Redis 中存储指纹的键名
137
+ REDIS_TTL = 0 # 指纹过期时间(0 表示永不过期)
138
+ CLEANUP_FP = 0 # 程序结束时是否清理指纹(0=不清理)
139
+ FILTER_DEBUG = True # 是否开启去重调试日志
140
+ DECODE_RESPONSES = True # Redis 返回是否解码为字符串
141
+
142
+ # ============================== 中间件配置 ==============================
143
+
144
+ MIDDLEWARES = [
145
+ # === 请求预处理阶段 ===
146
+ 'crawlo.middleware.request_ignore.RequestIgnoreMiddleware', # 1. 忽略无效请求
147
+ 'crawlo.middleware.download_delay.DownloadDelayMiddleware', # 2. 控制请求频率
148
+ 'crawlo.middleware.default_header.DefaultHeaderMiddleware', # 3. 添加默认请求头
149
+ 'crawlo.middleware.proxy.ProxyMiddleware', # 4. 设置代理
150
+
151
+ # === 响应处理阶段 ===
152
+ 'crawlo.middleware.retry.RetryMiddleware', # 5. 失败请求重试
153
+ 'crawlo.middleware.response_code.ResponseCodeMiddleware', # 6. 处理特殊状态码
154
+ 'crawlo.middleware.response_filter.ResponseFilterMiddleware', # 7. 响应内容过滤
155
+ ]
156
+
157
+ # ============================== 扩展与管道 ==============================
158
+
159
+ # 数据处理管道(启用的存储方式)
160
+ PIPELINES = [
161
+ 'crawlo.pipelines.console_pipeline.ConsolePipeline', # 控制台输出
162
+ # 'crawlo.pipelines.mysql_pipeline.AsyncmyMySQLPipeline', # MySQL 存储(可选)
163
+ ]
164
+
165
+ # 根据运行模式自动配置默认去重管道
166
+ if RUN_MODE == 'distributed':
167
+ # 分布式模式下添加Redis去重管道
168
+ PIPELINES.insert(0, DEFAULT_DEDUP_PIPELINE)
169
+ else:
170
+ # 单机模式下添加内存去重管道
171
+ PIPELINES.insert(0, DEFAULT_DEDUP_PIPELINE)
172
+
173
+ # 扩展组件(监控与日志)
174
+ EXTENSIONS = [
175
+ 'crawlo.extension.log_interval.LogIntervalExtension', # 定时日志
176
+ 'crawlo.extension.log_stats.LogStats', # 统计信息
177
+ 'crawlo.extension.logging_extension.CustomLoggerExtension', # 自定义日志
178
+ ]
179
+
180
+ # ============================== 日志与监控 ==============================
181
+
182
+ LOG_LEVEL = 'INFO' # 日志级别: DEBUG/INFO/WARNING/ERROR
183
+ STATS_DUMP = True # 是否周期性输出统计信息
184
+ LOG_FILE = f'logs/{PROJECT_NAME}.log' # 日志文件路径
185
+ LOG_FORMAT = '%(asctime)s - [%(name)s] - %(levelname)s: %(message)s'
186
+ LOG_ENCODING = 'utf-8'
187
+
188
+ # ============================== 代理配置 ==============================
189
+
190
+ PROXY_ENABLED = False # 是否启用代理
191
+ PROXY_API_URL = "https://api.proxyprovider.com/get" # 代理获取接口(请替换为真实地址)
192
+
193
+ # 代理提取方式(支持字段路径或函数)
194
+ PROXY_EXTRACTOR = "proxy" # 如返回 {"proxy": "http://1.1.1.1:8080"}
195
+
196
+ # 代理刷新控制
197
+ PROXY_REFRESH_INTERVAL = 60 # 代理刷新间隔(秒)
198
+ PROXY_API_TIMEOUT = 10 # 请求代理 API 超时时间
199
+
200
+ # ============================== Curl-Cffi 特有配置 ==============================
201
+
202
+ # 浏览器指纹模拟(仅 CurlCffi 下载器有效)
203
+ CURL_BROWSER_TYPE = "chrome" # 可选: chrome, edge, safari, firefox 或版本如 chrome136
204
+
205
+ # 自定义浏览器版本映射(可覆盖默认行为)
206
+ CURL_BROWSER_VERSION_MAP = {
207
+ "chrome": "chrome136",
208
+ "edge": "edge101",
209
+ "safari": "safari184",
210
+ "firefox": "firefox135",
211
+ # 示例:旧版本测试
212
+ # "chrome_legacy": "chrome110",
213
+ }
214
+
215
+ # Curl-Cffi 优化配置
216
+ CURL_RANDOMIZE_DELAY = False # 是否启用随机延迟
217
+ CURL_RETRY_BACKOFF = True # 是否启用指数退避重试
218
+
219
+ # 默认请求头(可被 Spider 覆盖)
220
+ DEFAULT_REQUEST_HEADERS = {
221
+ 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
222
+ '(KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
223
+ }
224
+
225
+ # ============================== 下载器优化配置 ==============================
226
+
227
+ # 下载器健康检查
228
+ DOWNLOADER_HEALTH_CHECK = True # 是否启用下载器健康检查
229
+ HEALTH_CHECK_INTERVAL = 60 # 健康检查间隔(秒)
230
+
231
+ # 请求统计配置
232
+ REQUEST_STATS_ENABLED = True # 是否启用请求统计
233
+ STATS_RESET_ON_START = False # 启动时是否重置统计
234
+
235
+ # HttpX 下载器专用配置
236
+ HTTPX_HTTP2 = True # 是否启用HTTP/2支持
237
+ HTTPX_FOLLOW_REDIRECTS = True # 是否自动跟随重定向
238
+
239
+ # AioHttp 下载器专用配置
240
+ AIOHTTP_AUTO_DECOMPRESS = True # 是否自动解压响应
241
+ AIOHTTP_FORCE_CLOSE = False # 是否强制关闭连接
242
+
243
+ # 通用优化配置
244
+ CONNECTION_TTL_DNS_CACHE = 300 # DNS缓存TTL(秒)
245
+ CONNECTION_KEEPALIVE_TIMEOUT = 15 # Keep-Alive超时(秒)