crawlo 1.0.9__py3-none-any.whl → 1.1.1__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 (111) hide show
  1. crawlo/__init__.py +33 -24
  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 -106
  6. crawlo/commands/genspider.py +125 -110
  7. crawlo/commands/list.py +147 -92
  8. crawlo/commands/run.py +286 -181
  9. crawlo/commands/startproject.py +111 -101
  10. crawlo/commands/stats.py +188 -59
  11. crawlo/core/__init__.py +2 -2
  12. crawlo/core/engine.py +158 -158
  13. crawlo/core/processor.py +40 -40
  14. crawlo/core/scheduler.py +57 -57
  15. crawlo/crawler.py +494 -492
  16. crawlo/downloader/__init__.py +78 -78
  17. crawlo/downloader/aiohttp_downloader.py +199 -199
  18. crawlo/downloader/cffi_downloader.py +242 -277
  19. crawlo/downloader/httpx_downloader.py +246 -246
  20. crawlo/event.py +11 -11
  21. crawlo/exceptions.py +78 -78
  22. crawlo/extension/__init__.py +31 -31
  23. crawlo/extension/log_interval.py +49 -49
  24. crawlo/extension/log_stats.py +44 -44
  25. crawlo/extension/logging_extension.py +34 -34
  26. crawlo/filters/__init__.py +37 -37
  27. crawlo/filters/aioredis_filter.py +150 -150
  28. crawlo/filters/memory_filter.py +202 -202
  29. crawlo/items/__init__.py +23 -23
  30. crawlo/items/base.py +21 -21
  31. crawlo/items/fields.py +53 -53
  32. crawlo/items/items.py +104 -104
  33. crawlo/middleware/__init__.py +21 -21
  34. crawlo/middleware/default_header.py +32 -32
  35. crawlo/middleware/download_delay.py +28 -28
  36. crawlo/middleware/middleware_manager.py +135 -135
  37. crawlo/middleware/proxy.py +245 -245
  38. crawlo/middleware/request_ignore.py +30 -30
  39. crawlo/middleware/response_code.py +18 -18
  40. crawlo/middleware/response_filter.py +26 -26
  41. crawlo/middleware/retry.py +90 -90
  42. crawlo/network/__init__.py +7 -7
  43. crawlo/network/request.py +203 -203
  44. crawlo/network/response.py +166 -166
  45. crawlo/pipelines/__init__.py +13 -13
  46. crawlo/pipelines/console_pipeline.py +39 -39
  47. crawlo/pipelines/mongo_pipeline.py +116 -116
  48. crawlo/pipelines/mysql_batch_pipline.py +272 -272
  49. crawlo/pipelines/mysql_pipeline.py +195 -195
  50. crawlo/pipelines/pipeline_manager.py +56 -56
  51. crawlo/project.py +153 -0
  52. crawlo/settings/__init__.py +7 -7
  53. crawlo/settings/default_settings.py +166 -168
  54. crawlo/settings/setting_manager.py +99 -99
  55. crawlo/spider/__init__.py +129 -129
  56. crawlo/stats_collector.py +59 -59
  57. crawlo/subscriber.py +106 -106
  58. crawlo/task_manager.py +27 -27
  59. crawlo/templates/crawlo.cfg.tmpl +10 -10
  60. crawlo/templates/project/__init__.py.tmpl +3 -3
  61. crawlo/templates/project/items.py.tmpl +17 -17
  62. crawlo/templates/project/middlewares.py.tmpl +75 -75
  63. crawlo/templates/project/pipelines.py.tmpl +63 -63
  64. crawlo/templates/project/settings.py.tmpl +54 -54
  65. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  66. crawlo/templates/spider/spider.py.tmpl +31 -31
  67. crawlo/utils/__init__.py +7 -7
  68. crawlo/utils/date_tools.py +233 -233
  69. crawlo/utils/db_helper.py +343 -343
  70. crawlo/utils/func_tools.py +82 -82
  71. crawlo/utils/log.py +128 -128
  72. crawlo/utils/pqueue.py +173 -173
  73. crawlo/utils/request.py +267 -267
  74. crawlo/utils/spider_loader.py +62 -62
  75. crawlo/utils/system.py +11 -11
  76. crawlo/utils/tools.py +4 -4
  77. crawlo/utils/url.py +39 -39
  78. crawlo-1.1.1.dist-info/METADATA +220 -0
  79. crawlo-1.1.1.dist-info/RECORD +100 -0
  80. examples/__init__.py +7 -0
  81. examples/baidu_spider/__init__.py +7 -0
  82. examples/baidu_spider/demo.py +94 -0
  83. examples/baidu_spider/items.py +46 -0
  84. examples/baidu_spider/middleware.py +49 -0
  85. examples/baidu_spider/pipeline.py +55 -0
  86. examples/baidu_spider/run.py +27 -0
  87. examples/baidu_spider/settings.py +121 -0
  88. examples/baidu_spider/spiders/__init__.py +7 -0
  89. examples/baidu_spider/spiders/bai_du.py +61 -0
  90. examples/baidu_spider/spiders/miit.py +159 -0
  91. examples/baidu_spider/spiders/sina.py +79 -0
  92. tests/__init__.py +7 -7
  93. tests/test_proxy_health_check.py +32 -32
  94. tests/test_proxy_middleware_integration.py +136 -136
  95. tests/test_proxy_providers.py +56 -56
  96. tests/test_proxy_stats.py +19 -19
  97. tests/test_proxy_strategies.py +59 -59
  98. crawlo/utils/concurrency_manager.py +0 -125
  99. crawlo/utils/project.py +0 -197
  100. crawlo-1.0.9.dist-info/METADATA +0 -49
  101. crawlo-1.0.9.dist-info/RECORD +0 -97
  102. examples/gxb/__init__.py +0 -0
  103. examples/gxb/items.py +0 -36
  104. examples/gxb/run.py +0 -16
  105. examples/gxb/settings.py +0 -72
  106. examples/gxb/spider/__init__.py +0 -0
  107. examples/gxb/spider/miit_spider.py +0 -180
  108. examples/gxb/spider/telecom_device.py +0 -129
  109. {crawlo-1.0.9.dist-info → crawlo-1.1.1.dist-info}/WHEEL +0 -0
  110. {crawlo-1.0.9.dist-info → crawlo-1.1.1.dist-info}/entry_points.txt +0 -0
  111. {crawlo-1.0.9.dist-info → crawlo-1.1.1.dist-info}/top_level.txt +0 -0
crawlo/exceptions.py CHANGED
@@ -1,78 +1,78 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- class TransformTypeError(TypeError):
4
- pass
5
-
6
-
7
- class OutputError(Exception):
8
- pass
9
-
10
-
11
- class SpiderTypeError(TypeError):
12
- pass
13
-
14
-
15
- class ItemInitError(Exception):
16
- pass
17
-
18
-
19
- class ItemAttributeError(Exception):
20
- pass
21
-
22
-
23
- class DecodeError(Exception):
24
- pass
25
-
26
-
27
- class MiddlewareInitError(Exception):
28
- pass
29
-
30
-
31
- class PipelineInitError(Exception):
32
- pass
33
-
34
-
35
- class InvalidOutputError(Exception):
36
- pass
37
-
38
-
39
- class RequestMethodError(Exception):
40
- pass
41
-
42
-
43
- class IgnoreRequestError(Exception):
44
- def __init__(self, msg):
45
- self.msg = msg
46
- super(IgnoreRequestError, self).__init__(msg)
47
-
48
-
49
- class ItemDiscard(Exception):
50
- def __init__(self, msg):
51
- self.msg = msg
52
- super(ItemDiscard, self).__init__(msg)
53
-
54
-
55
- class NotConfigured(Exception):
56
- pass
57
-
58
-
59
- class NotConfiguredError(Exception):
60
- pass
61
-
62
-
63
- class ExtensionInitError(Exception):
64
- pass
65
-
66
-
67
- class ReceiverTypeError(Exception):
68
- pass
69
-
70
-
71
- class SpiderCreationError(Exception):
72
- """爬虫实例化失败异常"""
73
- pass
74
-
75
-
76
- class ItemValidationError(Exception):
77
- """Item 字段验证错误"""
78
- pass
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ class TransformTypeError(TypeError):
4
+ pass
5
+
6
+
7
+ class OutputError(Exception):
8
+ pass
9
+
10
+
11
+ class SpiderTypeError(TypeError):
12
+ pass
13
+
14
+
15
+ class ItemInitError(Exception):
16
+ pass
17
+
18
+
19
+ class ItemAttributeError(Exception):
20
+ pass
21
+
22
+
23
+ class DecodeError(Exception):
24
+ pass
25
+
26
+
27
+ class MiddlewareInitError(Exception):
28
+ pass
29
+
30
+
31
+ class PipelineInitError(Exception):
32
+ pass
33
+
34
+
35
+ class InvalidOutputError(Exception):
36
+ pass
37
+
38
+
39
+ class RequestMethodError(Exception):
40
+ pass
41
+
42
+
43
+ class IgnoreRequestError(Exception):
44
+ def __init__(self, msg):
45
+ self.msg = msg
46
+ super(IgnoreRequestError, self).__init__(msg)
47
+
48
+
49
+ class ItemDiscard(Exception):
50
+ def __init__(self, msg):
51
+ self.msg = msg
52
+ super(ItemDiscard, self).__init__(msg)
53
+
54
+
55
+ class NotConfigured(Exception):
56
+ pass
57
+
58
+
59
+ class NotConfiguredError(Exception):
60
+ pass
61
+
62
+
63
+ class ExtensionInitError(Exception):
64
+ pass
65
+
66
+
67
+ class ReceiverTypeError(Exception):
68
+ pass
69
+
70
+
71
+ class SpiderCreationError(Exception):
72
+ """爬虫实例化失败异常"""
73
+ pass
74
+
75
+
76
+ class ItemValidationError(Exception):
77
+ """Item 字段验证错误"""
78
+ pass
@@ -1,31 +1,31 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from typing import List
4
- from pprint import pformat
5
-
6
- from crawlo.utils.log import get_logger
7
- from crawlo.utils.project import load_class
8
- from crawlo.exceptions import ExtensionInitError
9
-
10
-
11
- class ExtensionManager(object):
12
-
13
- def __init__(self, crawler):
14
- self.crawler = crawler
15
- self.extensions: List = []
16
- extensions = self.crawler.settings.get_list('EXTENSIONS')
17
- self.logger = get_logger(self.__class__.__name__, crawler.settings.get('LOG_LEVEL'))
18
- self._add_extensions(extensions)
19
-
20
- @classmethod
21
- def create_instance(cls, *args, **kwargs):
22
- return cls(*args, **kwargs)
23
-
24
- def _add_extensions(self, extensions):
25
- for extension in extensions:
26
- extension_cls = load_class(extension)
27
- if not hasattr(extension_cls, 'create_instance'):
28
- raise ExtensionInitError(f"extension init failed, Must have method 'create_instance()")
29
- self.extensions.append(extension_cls.create_instance(self.crawler))
30
- if extensions:
31
- self.logger.info(f"enabled extensions: \n {pformat(extensions)}")
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from typing import List
4
+ from pprint import pformat
5
+
6
+ from crawlo.utils.log import get_logger
7
+ from crawlo.project import load_class
8
+ from crawlo.exceptions import ExtensionInitError
9
+
10
+
11
+ class ExtensionManager(object):
12
+
13
+ def __init__(self, crawler):
14
+ self.crawler = crawler
15
+ self.extensions: List = []
16
+ extensions = self.crawler.settings.get_list('EXTENSIONS')
17
+ self.logger = get_logger(self.__class__.__name__, crawler.settings.get('LOG_LEVEL'))
18
+ self._add_extensions(extensions)
19
+
20
+ @classmethod
21
+ def create_instance(cls, *args, **kwargs):
22
+ return cls(*args, **kwargs)
23
+
24
+ def _add_extensions(self, extensions):
25
+ for extension in extensions:
26
+ extension_cls = load_class(extension)
27
+ if not hasattr(extension_cls, 'create_instance'):
28
+ raise ExtensionInitError(f"extension init failed, Must have method 'create_instance()")
29
+ self.extensions.append(extension_cls.create_instance(self.crawler))
30
+ if extensions:
31
+ self.logger.info(f"enabled extensions: \n {pformat(extensions)}")
@@ -1,49 +1,49 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- import asyncio
4
-
5
- from crawlo.utils.log import get_logger
6
- from crawlo.event import spider_opened, spider_closed
7
-
8
-
9
- class LogIntervalExtension(object):
10
-
11
- def __init__(self, crawler):
12
- self.task = None
13
- self.stats = crawler.stats
14
- self.item_count = 0
15
- self.response_count = 0
16
- self.seconds = crawler.settings.get('INTERVAL')
17
- self.interval = int(self.seconds / 60) if self.seconds % 60 == 0 else self.seconds
18
- self.interval = "" if self.interval == 1 else self.interval
19
- self.unit = 'min' if self.seconds % 60 == 0 else 's'
20
-
21
- self.logger = get_logger(self.__class__.__name__, crawler.settings.get('LOG_LEVEL'))
22
-
23
- @classmethod
24
- def create_instance(cls, crawler):
25
- o = cls(crawler)
26
- crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
27
- crawler.subscriber.subscribe(o.spider_closed, event=spider_closed)
28
- return o
29
-
30
- async def spider_opened(self):
31
- self.task = asyncio.create_task(self.interval_log())
32
- await self.task
33
-
34
- async def spider_closed(self):
35
- if self.task:
36
- self.task.cancel()
37
-
38
- async def interval_log(self):
39
- while True:
40
- last_item_count = self.stats.get_value('item_successful_count', default=0)
41
- last_response_count = self.stats.get_value('response_received_count', default=0)
42
- item_rate = last_item_count - self.item_count
43
- response_rate = last_response_count - self.response_count
44
- self.item_count, self.response_count = last_item_count, last_response_count
45
- self.logger.info(
46
- f'Crawled {last_response_count} pages (at {response_rate} pages/{self.interval}{self.unit}),'
47
- f' Got {last_item_count} items (at {item_rate} items/{self.interval}{self.unit}).'
48
- )
49
- await asyncio.sleep(self.seconds)
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ import asyncio
4
+
5
+ from crawlo.utils.log import get_logger
6
+ from crawlo.event import spider_opened, spider_closed
7
+
8
+
9
+ class LogIntervalExtension(object):
10
+
11
+ def __init__(self, crawler):
12
+ self.task = None
13
+ self.stats = crawler.stats
14
+ self.item_count = 0
15
+ self.response_count = 0
16
+ self.seconds = crawler.settings.get('INTERVAL')
17
+ self.interval = int(self.seconds / 60) if self.seconds % 60 == 0 else self.seconds
18
+ self.interval = "" if self.interval == 1 else self.interval
19
+ self.unit = 'min' if self.seconds % 60 == 0 else 's'
20
+
21
+ self.logger = get_logger(self.__class__.__name__, crawler.settings.get('LOG_LEVEL'))
22
+
23
+ @classmethod
24
+ def create_instance(cls, crawler):
25
+ o = cls(crawler)
26
+ crawler.subscriber.subscribe(o.spider_opened, event=spider_opened)
27
+ crawler.subscriber.subscribe(o.spider_closed, event=spider_closed)
28
+ return o
29
+
30
+ async def spider_opened(self):
31
+ self.task = asyncio.create_task(self.interval_log())
32
+ await self.task
33
+
34
+ async def spider_closed(self):
35
+ if self.task:
36
+ self.task.cancel()
37
+
38
+ async def interval_log(self):
39
+ while True:
40
+ last_item_count = self.stats.get_value('item_successful_count', default=0)
41
+ last_response_count = self.stats.get_value('response_received_count', default=0)
42
+ item_rate = last_item_count - self.item_count
43
+ response_rate = last_response_count - self.response_count
44
+ self.item_count, self.response_count = last_item_count, last_response_count
45
+ self.logger.info(
46
+ f'Crawled {last_response_count} pages (at {response_rate} pages/{self.interval}{self.unit}),'
47
+ f' Got {last_item_count} items (at {item_rate} items/{self.interval}{self.unit}).'
48
+ )
49
+ await asyncio.sleep(self.seconds)
@@ -1,44 +1,44 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from crawlo import event
4
- from crawlo.utils.date_tools import now, time_diff
5
-
6
-
7
- class LogStats(object):
8
-
9
- def __init__(self, stats):
10
- self._stats = stats
11
-
12
- @classmethod
13
- def create_instance(cls, crawler):
14
- o = cls(crawler.stats)
15
- crawler.subscriber.subscribe(o.spider_opened, event=event.spider_opened)
16
- crawler.subscriber.subscribe(o.spider_closed, event=event.spider_closed)
17
- crawler.subscriber.subscribe(o.item_successful, event=event.item_successful)
18
- crawler.subscriber.subscribe(o.item_discard, event=event.item_discard)
19
- crawler.subscriber.subscribe(o.response_received, event=event.response_received)
20
- crawler.subscriber.subscribe(o.request_scheduled, event=event.request_scheduled)
21
-
22
- return o
23
-
24
- async def spider_opened(self):
25
- self._stats['start_time'] = now(fmt='%Y-%m-%d %H:%M:%S')
26
-
27
- async def spider_closed(self):
28
- self._stats['end_time'] = now(fmt='%Y-%m-%d %H:%M:%S')
29
- self._stats['cost_time(s)'] = time_diff(start=self._stats['start_time'], end=self._stats['end_time'])
30
-
31
- async def item_successful(self, _item, _spider):
32
- self._stats.inc_value('item_successful_count')
33
-
34
- async def item_discard(self, _item, exc, _spider):
35
- self._stats.inc_value('item_discard_count')
36
- reason = exc.msg
37
- if reason:
38
- self._stats.inc_value(f"item_discard/{reason}")
39
-
40
- async def response_received(self, _response, _spider):
41
- self._stats.inc_value('response_received_count')
42
-
43
- async def request_scheduled(self, _request, _spider):
44
- self._stats.inc_value('request_scheduler_count')
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from crawlo import event
4
+ from crawlo.utils.date_tools import now, time_diff
5
+
6
+
7
+ class LogStats(object):
8
+
9
+ def __init__(self, stats):
10
+ self._stats = stats
11
+
12
+ @classmethod
13
+ def create_instance(cls, crawler):
14
+ o = cls(crawler.stats)
15
+ crawler.subscriber.subscribe(o.spider_opened, event=event.spider_opened)
16
+ crawler.subscriber.subscribe(o.spider_closed, event=event.spider_closed)
17
+ crawler.subscriber.subscribe(o.item_successful, event=event.item_successful)
18
+ crawler.subscriber.subscribe(o.item_discard, event=event.item_discard)
19
+ crawler.subscriber.subscribe(o.response_received, event=event.response_received)
20
+ crawler.subscriber.subscribe(o.request_scheduled, event=event.request_scheduled)
21
+
22
+ return o
23
+
24
+ async def spider_opened(self):
25
+ self._stats['start_time'] = now(fmt='%Y-%m-%d %H:%M:%S')
26
+
27
+ async def spider_closed(self):
28
+ self._stats['end_time'] = now(fmt='%Y-%m-%d %H:%M:%S')
29
+ self._stats['cost_time(s)'] = time_diff(start=self._stats['start_time'], end=self._stats['end_time'])
30
+
31
+ async def item_successful(self, _item, _spider):
32
+ self._stats.inc_value('item_successful_count')
33
+
34
+ async def item_discard(self, _item, exc, _spider):
35
+ self._stats.inc_value('item_discard_count')
36
+ reason = exc.msg
37
+ if reason:
38
+ self._stats.inc_value(f"item_discard/{reason}")
39
+
40
+ async def response_received(self, _response, _spider):
41
+ self._stats.inc_value('response_received_count')
42
+
43
+ async def request_scheduled(self, _request, _spider):
44
+ self._stats.inc_value('request_scheduler_count')
@@ -1,35 +1,35 @@
1
- from crawlo.exceptions import NotConfigured
2
- from crawlo.utils.log import get_logger
3
- from crawlo.utils.log import LoggerManager
4
-
5
-
6
- class CustomLoggerExtension:
7
- """
8
- 日志系统初始化扩展
9
- 遵循与 ExtensionManager 一致的接口规范:使用 create_instance
10
- """
11
-
12
- def __init__(self, settings):
13
- self.settings = settings
14
- # 初始化全局日志配置
15
- LoggerManager.configure(settings)
16
-
17
- @classmethod
18
- def create_instance(cls, crawler, *args, **kwargs):
19
- """
20
- 工厂方法:兼容 ExtensionManager 的创建方式
21
- 被 ExtensionManager 调用
22
- """
23
- # 可以通过 settings 控制是否启用
24
- if not crawler.settings.get('LOG_FILE') and not crawler.settings.get('LOG_ENABLE_CUSTOM'):
25
- raise NotConfigured("CustomLoggerExtension: LOG_FILE not set and LOG_ENABLE_CUSTOM=False")
26
-
27
- return cls(crawler.settings)
28
-
29
- def spider_opened(self, spider):
30
- logger = get_logger(__name__)
31
- logger.info(
32
- f"CustomLoggerExtension: Logging initialized. "
33
- f"LOG_FILE={self.settings.get('LOG_FILE')}, "
34
- f"LOG_LEVEL={self.settings.get('LOG_LEVEL')}"
1
+ from crawlo.exceptions import NotConfigured
2
+ from crawlo.utils.log import get_logger
3
+ from crawlo.utils.log import LoggerManager
4
+
5
+
6
+ class CustomLoggerExtension:
7
+ """
8
+ 日志系统初始化扩展
9
+ 遵循与 ExtensionManager 一致的接口规范:使用 create_instance
10
+ """
11
+
12
+ def __init__(self, settings):
13
+ self.settings = settings
14
+ # 初始化全局日志配置
15
+ LoggerManager.configure(settings)
16
+
17
+ @classmethod
18
+ def create_instance(cls, crawler, *args, **kwargs):
19
+ """
20
+ 工厂方法:兼容 ExtensionManager 的创建方式
21
+ 被 ExtensionManager 调用
22
+ """
23
+ # 可以通过 settings 控制是否启用
24
+ if not crawler.settings.get('LOG_FILE') and not crawler.settings.get('LOG_ENABLE_CUSTOM'):
25
+ raise NotConfigured("CustomLoggerExtension: LOG_FILE not set and LOG_ENABLE_CUSTOM=False")
26
+
27
+ return cls(crawler.settings)
28
+
29
+ def spider_opened(self, spider):
30
+ logger = get_logger(__name__)
31
+ logger.info(
32
+ f"CustomLoggerExtension: Logging initialized. "
33
+ f"LOG_FILE={self.settings.get('LOG_FILE')}, "
34
+ f"LOG_LEVEL={self.settings.get('LOG_LEVEL')}"
35
35
  )
@@ -1,37 +1,37 @@
1
- #!/usr/bin/python
2
- # -*- coding:UTF-8 -*-
3
- from abc import ABC, abstractmethod
4
-
5
- from crawlo import Request
6
- from crawlo.utils.request import request_fingerprint
7
-
8
-
9
- class BaseFilter(ABC):
10
-
11
- def __init__(self, logger, stats, debug: bool):
12
- self.logger = logger
13
- self.stats = stats
14
- self.debug = debug
15
-
16
- @classmethod
17
- def create_instance(cls, *args, **kwargs) -> 'BaseFilter':
18
- return cls(*args, **kwargs)
19
-
20
- def requested(self, request: Request):
21
- fp = request_fingerprint(request)
22
- if fp in self:
23
- return True
24
- self.add_fingerprint(fp)
25
- return False
26
-
27
- @abstractmethod
28
- def add_fingerprint(self, fp) -> None:
29
- pass
30
-
31
- def log_stats(self, request: Request) -> None:
32
- if self.debug:
33
- self.logger.debug(f'Filtered duplicate request: {request}')
34
- self.stats.inc_value(f'{self}/filtered_count')
35
-
36
- def __str__(self) -> str:
37
- return f'{self.__class__.__name__}'
1
+ #!/usr/bin/python
2
+ # -*- coding:UTF-8 -*-
3
+ from abc import ABC, abstractmethod
4
+
5
+ from crawlo import Request
6
+ from crawlo.utils.request import request_fingerprint
7
+
8
+
9
+ class BaseFilter(ABC):
10
+
11
+ def __init__(self, logger, stats, debug: bool):
12
+ self.logger = logger
13
+ self.stats = stats
14
+ self.debug = debug
15
+
16
+ @classmethod
17
+ def create_instance(cls, *args, **kwargs) -> 'BaseFilter':
18
+ return cls(*args, **kwargs)
19
+
20
+ def requested(self, request: Request):
21
+ fp = request_fingerprint(request)
22
+ if fp in self:
23
+ return True
24
+ self.add_fingerprint(fp)
25
+ return False
26
+
27
+ @abstractmethod
28
+ def add_fingerprint(self, fp) -> None:
29
+ pass
30
+
31
+ def log_stats(self, request: Request) -> None:
32
+ if self.debug:
33
+ self.logger.debug(f'Filtered duplicate request: {request}')
34
+ self.stats.inc_value(f'{self}/filtered_count')
35
+
36
+ def __str__(self) -> str:
37
+ return f'{self.__class__.__name__}'