zaclys/searxng
1
0
Fork 1
mirror of https://github.com/searxng/searxng synced 2024-01-01 19:24:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

add rate limiting per engine

Browse source
This commit is contained in:
Marc Abonce Seguin 2022-03-20 19:22:44 -06:00
parent dec04c0ed6
commit 93982f6d8e
3 changed files with 97 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
searx/engines/__init__.py
View file

@ -44,6 +44,7 @@ ENGINE_DEFAULT_ARGS = {
"enable_http": False,
"using_tor_proxy": False,
"display_error_messages": True,
"rate_limit": { "max_requests": float('inf'), "interval": 1 },
"tokens": [],
"about": {},
}
@ -168,6 +169,11 @@ def update_engine_attributes(engine: Engine, engine_data):
engine.categories = param_value
elif hasattr(engine, 'about') and param_name == 'about':
engine.about = {**engine.about, **engine_data['about']}
elif hasattr(engine, 'rate_limit') and param_name == 'rate_limit':
engine.rate_limit = {
'max_requests': int(param_value.get('max_requests')),
'interval': int(param_value.get('interval', 1))
}
else:
setattr(engine, param_name, param_value)