Implement more methods in CustomStatsCollector.

This commit is contained in:
Andrey Rakhmatullin 2026-06-20 16:49:08 +05:00
parent 5531ced9fb
commit 31af88a81f
1 changed files with 9 additions and 0 deletions

View File

@ -141,10 +141,19 @@ class TestStatsCollector:
def get_stats(self, spider=None):
return self._stats
def get_value(self, key, default=None, spider=None):
return self._stats.get(key, default)
def set_value(self, key, value, spider=None):
self._stats[key] = value
def inc_value(self, key, count=1, start=0, spider=None):
d = self._stats
d[key] = d.setdefault(key, start) + count
def max_value(self, key, value, spider=None) -> None:
self._stats[key] = max(self._stats.setdefault(key, value), value)
def close_spider(self, spider, reason):
pass