Restrict blocked site addition when using mergerSiteAdd

This commit is contained in:
shortcutme 2019-09-28 17:02:27 +02:00
parent b21b885aa9
commit 43c366d2fb
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
3 changed files with 18 additions and 2 deletions

View file

@ -25,6 +25,13 @@ class SiteManagerPlugin(object):
super(SiteManagerPlugin, self).load(*args, **kwargs)
filter_storage = ContentFilterStorage(site_manager=self)
def add(self, address, *args, **kwargs):
if filter_storage.isSiteblocked(address):
details = filter_storage.getSiteblockDetails(address)
raise Exception("Site blocked: %s" % html.escape(details.get("reason", "unknown reason")))
else:
return super(SiteManagerPlugin, self).add(address, *args, **kwargs)
@PluginManager.registerTo("UiWebsocket")
class UiWebsocketPlugin(object):