github-starred/burp-ui
2
0
Fork 0
mirror of https://github.com/ziirish/burp-ui.git synced 2026-05-21 06:45:24 -06:00
Code Issues 1 Projects Packages Wiki Activity Actions

use black to format the code

Browse source
This commit is contained in:
ziirish 2021-05-12 17:47:01 +02:00
parent 3dd7885bfb
commit 43253d8265
No known key found for this signature in database
GPG key ID: 72DB229A64B54E46
98 changed files with 10728 additions and 8599 deletions
Download patch file Download diff file Expand all files Collapse all files

61
burpui/misc/audit/handler.py
View file

@ -11,6 +11,7 @@ from flask_login import current_user
class BUIauditLoader(BUIaudit):
"""See :class:`burpui.misc.audit.interface.BUIaudit`"""
def __init__(self, app=None):
"""See :func:`burpui.misc.audit.interface.BUIaudit.__init__`
@ -20,37 +21,46 @@ class BUIauditLoader(BUIaudit):
self.app = app
backends = []
self.errors = {}
if self.app.audit_backends and 'none' not in self.app.audit_backends:
if self.app.audit_backends and "none" not in self.app.audit_backends:
me, _ = os.path.splitext(os.path.basename(__file__))
back = self.app.audit_backends
for au in back:
if au == me:
self.app.logger.critical('Recursive import not permitted!')
self.app.logger.critical("Recursive import not permitted!")
continue
try:
(modpath, _) = __name__.rsplit('.', 1)
mod = import_module('.' + au, modpath)
(modpath, _) = __name__.rsplit(".", 1)
mod = import_module("." + au, modpath)
obj = mod.BUIauditLoader(self.app)
backends.append(obj)
except:
import traceback
self.errors[au] = traceback.format_exc()
for name, plugin in self.app.plugin_manager.get_plugins_by_type('audit').items():
for name, plugin in self.app.plugin_manager.get_plugins_by_type(
"audit"
).items():
try:
obj = plugin.BUIauditLoader(self.app)
backends.append(obj)
except:
import traceback
self.errors[name] = traceback.format_exc()
backends.sort(key=lambda x: getattr(x, 'priority', -1), reverse=True)
if not backends and self.app.audit_backends and 'none' not in self.app.audit_backends:
backends.sort(key=lambda x: getattr(x, "priority", -1), reverse=True)
if (
not backends
and self.app.audit_backends
and "none" not in self.app.audit_backends
):
raise ImportError(
'No backend found for \'{}\':\n{}'.format(self.app.audit_backends,
self.errors)
"No backend found for '{}':\n{}".format(
self.app.audit_backends, self.errors
)
)
for name, err in self.errors.items():
self.app.logger.error(
'Unable to load module {}:\n{}'.format(repr(name), err)
"Unable to load module {}:\n{}".format(repr(name), err)
)
self.backends = OrderedDict()
for obj in backends:
@ -59,37 +69,34 @@ class BUIauditLoader(BUIaudit):
class BUIauditLogger(BUIauditLoggerInterface):
def __init__(self, loader):
self.loader = loader
def log(self, level, message, *args, **kwargs):
server_log = ''
if 'server' in kwargs:
server = kwargs['server']
del kwargs['server']
server_log = ""
if "server" in kwargs:
server = kwargs["server"]
del kwargs["server"]
if server:
server_log = f' on {server}'
server_log = f" on {server}"
if current_user and not current_user.is_anonymous:
msg = f'{current_user} -> {message}{server_log}'
msg = f"{current_user} -> {message}{server_log}"
else:
msg = f'{message}{server_log}'
caller = ''
msg = f"{message}{server_log}"
caller = ""
stack = inspect.stack()
exclude = [
'audit/interface.py',
'audit/handler.py',
"audit/interface.py",
"audit/handler.py",
]
for frame in stack:
if any(frame.filename.endswith(x) for x in exclude):
continue
caller = f'{frame.function} [{frame.filename}:{frame.lineno}]'
caller = f"{frame.function} [{frame.filename}:{frame.lineno}]"
break
if 'extra' in kwargs and isinstance(kwargs['extra'], dict):
kwargs['extra']['from'] = caller
if "extra" in kwargs and isinstance(kwargs["extra"], dict):
kwargs["extra"]["from"] = caller
else:
kwargs['extra'] = {
'from': caller
}
kwargs["extra"] = {"from": caller}
for back in self.loader.backends.values():
back.logger.log(level, msg, *args, **kwargs)