crawlo 1.1.8__py3-none-any.whl → 1.1.9__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 (190) hide show
  1. crawlo/__init__.py +61 -61
  2. crawlo/__version__.py +1 -1
  3. crawlo/cleaners/__init__.py +60 -60
  4. crawlo/cleaners/data_formatter.py +225 -225
  5. crawlo/cleaners/encoding_converter.py +125 -125
  6. crawlo/cleaners/text_cleaner.py +232 -232
  7. crawlo/cli.py +65 -65
  8. crawlo/commands/__init__.py +14 -14
  9. crawlo/commands/check.py +594 -594
  10. crawlo/commands/genspider.py +151 -151
  11. crawlo/commands/help.py +132 -132
  12. crawlo/commands/list.py +155 -155
  13. crawlo/commands/run.py +292 -292
  14. crawlo/commands/startproject.py +418 -418
  15. crawlo/commands/stats.py +188 -188
  16. crawlo/commands/utils.py +186 -186
  17. crawlo/config.py +312 -312
  18. crawlo/config_validator.py +252 -252
  19. crawlo/core/__init__.py +2 -2
  20. crawlo/core/engine.py +345 -345
  21. crawlo/core/processor.py +40 -40
  22. crawlo/core/scheduler.py +136 -136
  23. crawlo/crawler.py +1027 -1027
  24. crawlo/downloader/__init__.py +266 -266
  25. crawlo/downloader/aiohttp_downloader.py +220 -220
  26. crawlo/downloader/cffi_downloader.py +256 -256
  27. crawlo/downloader/httpx_downloader.py +259 -259
  28. crawlo/downloader/hybrid_downloader.py +213 -213
  29. crawlo/downloader/playwright_downloader.py +402 -402
  30. crawlo/downloader/selenium_downloader.py +472 -472
  31. crawlo/event.py +11 -11
  32. crawlo/exceptions.py +81 -81
  33. crawlo/extension/__init__.py +37 -37
  34. crawlo/extension/health_check.py +141 -141
  35. crawlo/extension/log_interval.py +57 -57
  36. crawlo/extension/log_stats.py +81 -81
  37. crawlo/extension/logging_extension.py +43 -43
  38. crawlo/extension/memory_monitor.py +104 -104
  39. crawlo/extension/performance_profiler.py +133 -133
  40. crawlo/extension/request_recorder.py +107 -107
  41. crawlo/filters/__init__.py +154 -154
  42. crawlo/filters/aioredis_filter.py +280 -280
  43. crawlo/filters/memory_filter.py +269 -269
  44. crawlo/items/__init__.py +23 -23
  45. crawlo/items/base.py +21 -21
  46. crawlo/items/fields.py +53 -53
  47. crawlo/items/items.py +104 -104
  48. crawlo/middleware/__init__.py +21 -21
  49. crawlo/middleware/default_header.py +32 -32
  50. crawlo/middleware/download_delay.py +28 -28
  51. crawlo/middleware/middleware_manager.py +135 -135
  52. crawlo/middleware/proxy.py +272 -272
  53. crawlo/middleware/request_ignore.py +30 -30
  54. crawlo/middleware/response_code.py +18 -18
  55. crawlo/middleware/response_filter.py +26 -26
  56. crawlo/middleware/retry.py +124 -124
  57. crawlo/mode_manager.py +211 -211
  58. crawlo/network/__init__.py +21 -21
  59. crawlo/network/request.py +338 -338
  60. crawlo/network/response.py +359 -359
  61. crawlo/pipelines/__init__.py +21 -21
  62. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  63. crawlo/pipelines/console_pipeline.py +39 -39
  64. crawlo/pipelines/csv_pipeline.py +316 -316
  65. crawlo/pipelines/database_dedup_pipeline.py +224 -224
  66. crawlo/pipelines/json_pipeline.py +218 -218
  67. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  68. crawlo/pipelines/mongo_pipeline.py +131 -131
  69. crawlo/pipelines/mysql_pipeline.py +316 -316
  70. crawlo/pipelines/pipeline_manager.py +61 -61
  71. crawlo/pipelines/redis_dedup_pipeline.py +167 -167
  72. crawlo/project.py +187 -187
  73. crawlo/queue/pqueue.py +37 -37
  74. crawlo/queue/queue_manager.py +334 -334
  75. crawlo/queue/redis_priority_queue.py +298 -298
  76. crawlo/settings/__init__.py +7 -7
  77. crawlo/settings/default_settings.py +219 -219
  78. crawlo/settings/setting_manager.py +122 -122
  79. crawlo/spider/__init__.py +639 -639
  80. crawlo/stats_collector.py +59 -59
  81. crawlo/subscriber.py +130 -130
  82. crawlo/task_manager.py +30 -30
  83. crawlo/templates/crawlo.cfg.tmpl +10 -10
  84. crawlo/templates/project/__init__.py.tmpl +3 -3
  85. crawlo/templates/project/items.py.tmpl +17 -17
  86. crawlo/templates/project/middlewares.py.tmpl +109 -109
  87. crawlo/templates/project/pipelines.py.tmpl +96 -96
  88. crawlo/templates/project/run.py.tmpl +45 -45
  89. crawlo/templates/project/settings.py.tmpl +326 -326
  90. crawlo/templates/project/settings_distributed.py.tmpl +119 -119
  91. crawlo/templates/project/settings_gentle.py.tmpl +94 -94
  92. crawlo/templates/project/settings_high_performance.py.tmpl +151 -151
  93. crawlo/templates/project/settings_simple.py.tmpl +68 -68
  94. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  95. crawlo/templates/spider/spider.py.tmpl +141 -141
  96. crawlo/tools/__init__.py +182 -182
  97. crawlo/tools/anti_crawler.py +268 -268
  98. crawlo/tools/authenticated_proxy.py +240 -240
  99. crawlo/tools/data_validator.py +180 -180
  100. crawlo/tools/date_tools.py +35 -35
  101. crawlo/tools/distributed_coordinator.py +386 -386
  102. crawlo/tools/retry_mechanism.py +220 -220
  103. crawlo/tools/scenario_adapter.py +262 -262
  104. crawlo/utils/__init__.py +35 -35
  105. crawlo/utils/batch_processor.py +260 -260
  106. crawlo/utils/controlled_spider_mixin.py +439 -439
  107. crawlo/utils/date_tools.py +290 -290
  108. crawlo/utils/db_helper.py +343 -343
  109. crawlo/utils/enhanced_error_handler.py +359 -359
  110. crawlo/utils/env_config.py +105 -105
  111. crawlo/utils/error_handler.py +125 -125
  112. crawlo/utils/func_tools.py +82 -82
  113. crawlo/utils/large_scale_config.py +286 -286
  114. crawlo/utils/large_scale_helper.py +343 -343
  115. crawlo/utils/log.py +128 -128
  116. crawlo/utils/performance_monitor.py +284 -284
  117. crawlo/utils/queue_helper.py +175 -175
  118. crawlo/utils/redis_connection_pool.py +334 -334
  119. crawlo/utils/redis_key_validator.py +199 -199
  120. crawlo/utils/request.py +267 -267
  121. crawlo/utils/request_serializer.py +219 -219
  122. crawlo/utils/spider_loader.py +62 -62
  123. crawlo/utils/system.py +11 -11
  124. crawlo/utils/tools.py +4 -4
  125. crawlo/utils/url.py +39 -39
  126. {crawlo-1.1.8.dist-info → crawlo-1.1.9.dist-info}/METADATA +626 -626
  127. crawlo-1.1.9.dist-info/RECORD +190 -0
  128. examples/__init__.py +7 -7
  129. tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +81 -81
  130. tests/__init__.py +7 -7
  131. tests/advanced_tools_example.py +275 -275
  132. tests/authenticated_proxy_example.py +236 -236
  133. tests/cleaners_example.py +160 -160
  134. tests/config_validation_demo.py +102 -102
  135. tests/controlled_spider_example.py +205 -205
  136. tests/date_tools_example.py +180 -180
  137. tests/dynamic_loading_example.py +523 -523
  138. tests/dynamic_loading_test.py +104 -104
  139. tests/env_config_example.py +133 -133
  140. tests/error_handling_example.py +171 -171
  141. tests/redis_key_validation_demo.py +130 -130
  142. tests/response_improvements_example.py +144 -144
  143. tests/test_advanced_tools.py +148 -148
  144. tests/test_all_redis_key_configs.py +145 -145
  145. tests/test_authenticated_proxy.py +141 -141
  146. tests/test_cleaners.py +54 -54
  147. tests/test_comprehensive.py +146 -146
  148. tests/test_config_validator.py +193 -193
  149. tests/test_date_tools.py +123 -123
  150. tests/test_double_crawlo_fix.py +207 -207
  151. tests/test_double_crawlo_fix_simple.py +124 -124
  152. tests/test_dynamic_downloaders_proxy.py +124 -124
  153. tests/test_dynamic_proxy.py +92 -92
  154. tests/test_dynamic_proxy_config.py +146 -146
  155. tests/test_dynamic_proxy_real.py +109 -109
  156. tests/test_edge_cases.py +303 -303
  157. tests/test_enhanced_error_handler.py +270 -270
  158. tests/test_env_config.py +121 -121
  159. tests/test_error_handler_compatibility.py +112 -112
  160. tests/test_final_validation.py +153 -153
  161. tests/test_framework_env_usage.py +103 -103
  162. tests/test_integration.py +356 -356
  163. tests/test_item_dedup_redis_key.py +122 -122
  164. tests/test_parsel.py +29 -29
  165. tests/test_performance.py +327 -327
  166. tests/test_proxy_health_check.py +32 -32
  167. tests/test_proxy_middleware_integration.py +136 -136
  168. tests/test_proxy_providers.py +56 -56
  169. tests/test_proxy_stats.py +19 -19
  170. tests/test_proxy_strategies.py +59 -59
  171. tests/test_queue_manager_double_crawlo.py +230 -230
  172. tests/test_queue_manager_redis_key.py +176 -176
  173. tests/test_redis_config.py +28 -28
  174. tests/test_redis_connection_pool.py +294 -294
  175. tests/test_redis_key_naming.py +181 -181
  176. tests/test_redis_key_validator.py +123 -123
  177. tests/test_redis_queue.py +224 -224
  178. tests/test_request_serialization.py +70 -70
  179. tests/test_response_improvements.py +152 -152
  180. tests/test_scheduler.py +241 -241
  181. tests/test_simple_response.py +61 -61
  182. tests/test_telecom_spider_redis_key.py +205 -205
  183. tests/test_template_content.py +87 -87
  184. tests/test_template_redis_key.py +134 -134
  185. tests/test_tools.py +153 -153
  186. tests/tools_example.py +257 -257
  187. crawlo-1.1.8.dist-info/RECORD +0 -190
  188. {crawlo-1.1.8.dist-info → crawlo-1.1.9.dist-info}/WHEEL +0 -0
  189. {crawlo-1.1.8.dist-info → crawlo-1.1.9.dist-info}/entry_points.txt +0 -0
  190. {crawlo-1.1.8.dist-info → crawlo-1.1.9.dist-info}/top_level.txt +0 -0
crawlo/items/fields.py CHANGED
@@ -1,54 +1,54 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- Field 类定义
5
- """
6
-
7
- from typing import Any, Optional, Type
8
-
9
-
10
- class Field:
11
- """
12
- 字段定义类,用于定义 Item 的字段属性和验证规则
13
- """
14
- def __init__(
15
- self,
16
- nullable: bool = True,
17
- *,
18
- default: Any = None,
19
- field_type: Optional[Type] = None,
20
- max_length: Optional[int] = None,
21
- description: str = ""
22
- ):
23
- self.nullable = nullable
24
- self.default = default
25
- self.field_type = field_type
26
- self.max_length = max_length
27
- self.description = description
28
-
29
- def validate(self, value: Any, field_name: str = "") -> Any:
30
- """
31
- 验证字段值是否符合规则
32
- """
33
- if value is None or (isinstance(value, str) and value.strip() == ""):
34
- if self.default is not None:
35
- return self.default
36
- elif not self.nullable:
37
- raise ValueError(
38
- f"字段 '{field_name}' 不允许为空。"
39
- )
40
-
41
- if value is not None and not (isinstance(value, str) and value.strip() == ""):
42
- if self.field_type and not isinstance(value, self.field_type):
43
- raise TypeError(
44
- f"字段 '{field_name}' 类型错误:期望类型 {self.field_type}, 得到 {type(value)},值:{value!r}"
45
- )
46
- if self.max_length and len(str(value)) > self.max_length:
47
- raise ValueError(
48
- f"字段 '{field_name}' 长度超限:最大长度 {self.max_length},当前长度 {len(str(value))},值:{value!r}"
49
- )
50
-
51
- return value
52
-
53
- def __repr__(self):
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ Field 类定义
5
+ """
6
+
7
+ from typing import Any, Optional, Type
8
+
9
+
10
+ class Field:
11
+ """
12
+ 字段定义类,用于定义 Item 的字段属性和验证规则
13
+ """
14
+ def __init__(
15
+ self,
16
+ nullable: bool = True,
17
+ *,
18
+ default: Any = None,
19
+ field_type: Optional[Type] = None,
20
+ max_length: Optional[int] = None,
21
+ description: str = ""
22
+ ):
23
+ self.nullable = nullable
24
+ self.default = default
25
+ self.field_type = field_type
26
+ self.max_length = max_length
27
+ self.description = description
28
+
29
+ def validate(self, value: Any, field_name: str = "") -> Any:
30
+ """
31
+ 验证字段值是否符合规则
32
+ """
33
+ if value is None or (isinstance(value, str) and value.strip() == ""):
34
+ if self.default is not None:
35
+ return self.default
36
+ elif not self.nullable:
37
+ raise ValueError(
38
+ f"字段 '{field_name}' 不允许为空。"
39
+ )
40
+
41
+ if value is not None and not (isinstance(value, str) and value.strip() == ""):
42
+ if self.field_type and not isinstance(value, self.field_type):
43
+ raise TypeError(
44
+ f"字段 '{field_name}' 类型错误:期望类型 {self.field_type}, 得到 {type(value)},值:{value!r}"
45
+ )
46
+ if self.max_length and len(str(value)) > self.max_length:
47
+ raise ValueError(
48
+ f"字段 '{field_name}' 长度超限:最大长度 {self.max_length},当前长度 {len(str(value))},值:{value!r}"
49
+ )
50
+
51
+ return value
52
+
53
+ def __repr__(self):
54
54
  return f"<Field nullable={self.nullable} type={self.field_type} default={self.default}>"
crawlo/items/items.py CHANGED
@@ -1,105 +1,105 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- Item 类定义
5
- """
6
- from copy import deepcopy
7
- from pprint import pformat
8
- from typing import Any, Iterator, Dict
9
- from collections.abc import MutableMapping
10
-
11
- from .base import ItemMeta
12
- from crawlo.exceptions import ItemInitError, ItemAttributeError
13
-
14
-
15
- class Item(MutableMapping, metaclass=ItemMeta):
16
- """
17
- 数据项基类,用于定义结构化数据
18
- """
19
- FIELDS: Dict[str, Any] = {}
20
-
21
- def __init__(self, *args, **kwargs):
22
- if args:
23
- raise ItemInitError(f"{self.__class__.__name__} 不支持位置参数:{args},请使用关键字参数初始化。")
24
-
25
- self._values: Dict[str, Any] = {}
26
-
27
- # 初始化字段,默认值填充
28
- for field_name, field_obj in self.FIELDS.items():
29
- if field_obj.default is not None:
30
- self._values[field_name] = field_obj.default
31
-
32
- # 覆盖默认值或设置新值
33
- for key, value in kwargs.items():
34
- self[key] = value
35
-
36
- def __getitem__(self, item: str) -> Any:
37
- return self._values[item]
38
-
39
- def __setitem__(self, key: str, value: Any) -> None:
40
- if key not in self.FIELDS:
41
- raise KeyError(f"{self.__class__.__name__} 不包含字段:{key}")
42
-
43
- field = self.FIELDS[key]
44
- try:
45
- validated_value = field.validate(value, field_name=key)
46
- self._values[key] = validated_value
47
- except Exception as e:
48
- error_lines = [
49
- "",
50
- "【字段校验失败】",
51
- f"字段名称: {key}",
52
- f"数据类型: {type(value)}",
53
- f"原始值: {repr(value)}",
54
- f"是否允许空值: {field.nullable}",
55
- f"错误原因: {str(e)}",
56
- ""
57
- ]
58
- detailed_error = "\n".join(error_lines)
59
- raise type(e)(detailed_error) from e
60
-
61
- def __delitem__(self, key: str) -> None:
62
- del self._values[key]
63
-
64
- def __setattr__(self, key: str, value: Any) -> None:
65
- if not key.startswith("_"):
66
- raise AttributeError(
67
- f"设置字段值请使用 item[{key!r}] = {value!r}"
68
- )
69
- super().__setattr__(key, value)
70
-
71
- def __getattr__(self, item: str) -> Any:
72
- raise AttributeError(
73
- f"{self.__class__.__name__} 不支持字段:{item}。"
74
- f"请先在 `{self.__class__.__name__}` 中声明该字段,再通过 item[{item!r}] 获取。"
75
- )
76
-
77
- def __getattribute__(self, item: str) -> Any:
78
- try:
79
- field = super().__getattribute__("FIELDS")
80
- if isinstance(field, dict) and item in field:
81
- raise ItemAttributeError(
82
- f"获取字段值请使用 item[{item!r}]"
83
- )
84
- except AttributeError:
85
- pass # 如果 FIELDS 尚未定义,继续执行后续逻辑
86
- return super().__getattribute__(item)
87
-
88
- def __repr__(self) -> str:
89
- return pformat(dict(self))
90
-
91
- __str__ = __repr__
92
-
93
- def __iter__(self) -> Iterator[str]:
94
- return iter(self._values)
95
-
96
- def __len__(self) -> int:
97
- return len(self._values)
98
-
99
- def to_dict(self) -> Dict[str, Any]:
100
- """转换为字典"""
101
- return dict(self)
102
-
103
- def copy(self) -> "Item":
104
- """深拷贝当前 Item"""
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ Item 类定义
5
+ """
6
+ from copy import deepcopy
7
+ from pprint import pformat
8
+ from typing import Any, Iterator, Dict
9
+ from collections.abc import MutableMapping
10
+
11
+ from .base import ItemMeta
12
+ from crawlo.exceptions import ItemInitError, ItemAttributeError
13
+
14
+
15
+ class Item(MutableMapping, metaclass=ItemMeta):
16
+ """
17
+ 数据项基类,用于定义结构化数据
18
+ """
19
+ FIELDS: Dict[str, Any] = {}
20
+
21
+ def __init__(self, *args, **kwargs):
22
+ if args:
23
+ raise ItemInitError(f"{self.__class__.__name__} 不支持位置参数:{args},请使用关键字参数初始化。")
24
+
25
+ self._values: Dict[str, Any] = {}
26
+
27
+ # 初始化字段,默认值填充
28
+ for field_name, field_obj in self.FIELDS.items():
29
+ if field_obj.default is not None:
30
+ self._values[field_name] = field_obj.default
31
+
32
+ # 覆盖默认值或设置新值
33
+ for key, value in kwargs.items():
34
+ self[key] = value
35
+
36
+ def __getitem__(self, item: str) -> Any:
37
+ return self._values[item]
38
+
39
+ def __setitem__(self, key: str, value: Any) -> None:
40
+ if key not in self.FIELDS:
41
+ raise KeyError(f"{self.__class__.__name__} 不包含字段:{key}")
42
+
43
+ field = self.FIELDS[key]
44
+ try:
45
+ validated_value = field.validate(value, field_name=key)
46
+ self._values[key] = validated_value
47
+ except Exception as e:
48
+ error_lines = [
49
+ "",
50
+ "【字段校验失败】",
51
+ f"字段名称: {key}",
52
+ f"数据类型: {type(value)}",
53
+ f"原始值: {repr(value)}",
54
+ f"是否允许空值: {field.nullable}",
55
+ f"错误原因: {str(e)}",
56
+ ""
57
+ ]
58
+ detailed_error = "\n".join(error_lines)
59
+ raise type(e)(detailed_error) from e
60
+
61
+ def __delitem__(self, key: str) -> None:
62
+ del self._values[key]
63
+
64
+ def __setattr__(self, key: str, value: Any) -> None:
65
+ if not key.startswith("_"):
66
+ raise AttributeError(
67
+ f"设置字段值请使用 item[{key!r}] = {value!r}"
68
+ )
69
+ super().__setattr__(key, value)
70
+
71
+ def __getattr__(self, item: str) -> Any:
72
+ raise AttributeError(
73
+ f"{self.__class__.__name__} 不支持字段:{item}。"
74
+ f"请先在 `{self.__class__.__name__}` 中声明该字段,再通过 item[{item!r}] 获取。"
75
+ )
76
+
77
+ def __getattribute__(self, item: str) -> Any:
78
+ try:
79
+ field = super().__getattribute__("FIELDS")
80
+ if isinstance(field, dict) and item in field:
81
+ raise ItemAttributeError(
82
+ f"获取字段值请使用 item[{item!r}]"
83
+ )
84
+ except AttributeError:
85
+ pass # 如果 FIELDS 尚未定义,继续执行后续逻辑
86
+ return super().__getattribute__(item)
87
+
88
+ def __repr__(self) -> str:
89
+ return pformat(dict(self))
90
+
91
+ __str__ = __repr__
92
+
93
+ def __iter__(self) -> Iterator[str]:
94
+ return iter(self._values)
95
+
96
+ def __len__(self) -> int:
97
+ return len(self._values)
98
+
99
+ def to_dict(self) -> Dict[str, Any]:
100
+ """转换为字典"""
101
+ return dict(self)
102
+
103
+ def copy(self) -> "Item":
104
+ """深拷贝当前 Item"""
105
105
  return deepcopy(self)
@@ -1,21 +1,21 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from crawlo import Request, Response
4
-
5
-
6
- class BaseMiddleware(object):
7
- def process_request(self, request, spider) -> None | Request | Response:
8
- # 请求预处理
9
- pass
10
-
11
- def process_response(self, request, response, spider) -> Request | Response:
12
- # 响应预处理
13
- pass
14
-
15
- def process_exception(self, request, exp, spider) -> None | Request | Response:
16
- # 异常预处理
17
- pass
18
-
19
- @classmethod
20
- def create_instance(cls, crawler):
21
- return cls()
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from crawlo import Request, Response
4
+
5
+
6
+ class BaseMiddleware(object):
7
+ def process_request(self, request, spider) -> None | Request | Response:
8
+ # 请求预处理
9
+ pass
10
+
11
+ def process_response(self, request, response, spider) -> Request | Response:
12
+ # 响应预处理
13
+ pass
14
+
15
+ def process_exception(self, request, exp, spider) -> None | Request | Response:
16
+ # 异常预处理
17
+ pass
18
+
19
+ @classmethod
20
+ def create_instance(cls, crawler):
21
+ return cls()
@@ -1,32 +1,32 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from crawlo.event import spider_opened
4
-
5
-
6
- class DefaultHeaderMiddleware(object):
7
-
8
- def __init__(self, user_agent, headers, spider):
9
- self.user_agent = user_agent
10
- self.headers = headers
11
- self.spider = spider
12
-
13
- @classmethod
14
- def create_instance(cls, crawler):
15
- o = cls(
16
- user_agent=crawler.settings.get('USER_AGENT'),
17
- headers=crawler.settings.get_dict('DEFAULT_HEADERS'),
18
- spider=crawler.spider
19
- )
20
- crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
21
- return o
22
-
23
- async def spider_opened(self):
24
- self.user_agent = getattr(self.spider, 'user_agent', self.user_agent)
25
- self.headers = getattr(self.spider, 'headers', self.headers)
26
- if self.user_agent:
27
- self.headers.setdefault('User-Agent', self.user_agent)
28
-
29
- def process_request(self, request, _spider):
30
- if self.headers:
31
- for key, value in self.headers.items():
32
- request.headers.setdefault(key, value)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from crawlo.event import spider_opened
4
+
5
+
6
+ class DefaultHeaderMiddleware(object):
7
+
8
+ def __init__(self, user_agent, headers, spider):
9
+ self.user_agent = user_agent
10
+ self.headers = headers
11
+ self.spider = spider
12
+
13
+ @classmethod
14
+ def create_instance(cls, crawler):
15
+ o = cls(
16
+ user_agent=crawler.settings.get('USER_AGENT'),
17
+ headers=crawler.settings.get_dict('DEFAULT_HEADERS'),
18
+ spider=crawler.spider
19
+ )
20
+ crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
21
+ return o
22
+
23
+ async def spider_opened(self):
24
+ self.user_agent = getattr(self.spider, 'user_agent', self.user_agent)
25
+ self.headers = getattr(self.spider, 'headers', self.headers)
26
+ if self.user_agent:
27
+ self.headers.setdefault('User-Agent', self.user_agent)
28
+
29
+ def process_request(self, request, _spider):
30
+ if self.headers:
31
+ for key, value in self.headers.items():
32
+ request.headers.setdefault(key, value)
@@ -1,28 +1,28 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from asyncio import sleep
4
- from random import uniform
5
- from crawlo.utils.log import get_logger
6
- from crawlo.exceptions import NotConfiguredError
7
-
8
-
9
- class DownloadDelayMiddleware(object):
10
-
11
- def __init__(self, settings, log_level):
12
- self.delay = settings.get_float("DOWNLOAD_DELAY")
13
- if not self.delay:
14
- raise NotConfiguredError
15
- self.randomness = settings.get_bool("RANDOMNESS")
16
- self.floor, self.upper = settings.get_list("RANDOM_RANGE")
17
- self.logger = get_logger(self.__class__.__name__, log_level)
18
-
19
- @classmethod
20
- def create_instance(cls, crawler):
21
- o = cls(settings=crawler.settings, log_level=crawler.settings.get('LOG_LEVEL'))
22
- return o
23
-
24
- async def process_request(self, _request, _spider):
25
- if self.randomness:
26
- await sleep(uniform(self.delay * self.floor, self.delay * self.upper))
27
- else:
28
- await sleep(self.delay)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from asyncio import sleep
4
+ from random import uniform
5
+ from crawlo.utils.log import get_logger
6
+ from crawlo.exceptions import NotConfiguredError
7
+
8
+
9
+ class DownloadDelayMiddleware(object):
10
+
11
+ def __init__(self, settings, log_level):
12
+ self.delay = settings.get_float("DOWNLOAD_DELAY")
13
+ if not self.delay:
14
+ raise NotConfiguredError
15
+ self.randomness = settings.get_bool("RANDOMNESS")
16
+ self.floor, self.upper = settings.get_list("RANDOM_RANGE")
17
+ self.logger = get_logger(self.__class__.__name__, log_level)
18
+
19
+ @classmethod
20
+ def create_instance(cls, crawler):
21
+ o = cls(settings=crawler.settings, log_level=crawler.settings.get('LOG_LEVEL'))
22
+ return o
23
+
24
+ async def process_request(self, _request, _spider):
25
+ if self.randomness:
26
+ await sleep(uniform(self.delay * self.floor, self.delay * self.upper))
27
+ else:
28
+ await sleep(self.delay)