mirror of https://github.com/Kkevsterrr/geneva
fix handlers exception
This commit is contained in:
parent
01828ddf97
commit
0abac72ddb
|
@ -135,6 +135,7 @@ class CustomAdapter(logging.LoggerAdapter):
|
||||||
|
|
||||||
def __init__(self, logger, extras):
|
def __init__(self, logger, extras):
|
||||||
super().__init__(logger, extras)
|
super().__init__(logger, extras)
|
||||||
|
self.handlers = logger.handlers
|
||||||
self.ips = {}
|
self.ips = {}
|
||||||
|
|
||||||
def debug(self, msg, *args, **kwargs):
|
def debug(self, msg, *args, **kwargs):
|
||||||
|
|
|
@ -30,7 +30,7 @@ BASEPATH = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
|
||||||
|
|
||||||
class Engine():
|
class Engine():
|
||||||
def __init__(self, server_port, string_strategy, server_side=False, environment_id=None, output_directory="trials", log_level="info"):
|
def __init__(self, server_port, string_strategy, server_side=False, environment_id=None, output_directory="trials", log_level="info", demo_mode=False):
|
||||||
self.server_port = server_port
|
self.server_port = server_port
|
||||||
self.seen_packets = []
|
self.seen_packets = []
|
||||||
# Set up the directory and ID for logging
|
# Set up the directory and ID for logging
|
||||||
|
@ -45,7 +45,8 @@ class Engine():
|
||||||
__name__,
|
__name__,
|
||||||
"engine",
|
"engine",
|
||||||
environment_id,
|
environment_id,
|
||||||
log_level=log_level)
|
log_level=log_level,
|
||||||
|
demo_mode=demo_mode)
|
||||||
self.output_directory = output_directory
|
self.output_directory = output_directory
|
||||||
self.server_side = server_side
|
self.server_side = server_side
|
||||||
|
|
||||||
|
@ -313,6 +314,7 @@ def get_args():
|
||||||
parser.add_argument('--log', action='store', default="debug",
|
parser.add_argument('--log', action='store', default="debug",
|
||||||
choices=("debug", "info", "warning", "critical", "error"),
|
choices=("debug", "info", "warning", "critical", "error"),
|
||||||
help="Sets the log level")
|
help="Sets the log level")
|
||||||
|
parser.add_argument("--demo-mode", action='store_true', help="Replaces all IPs with dummy IPs in log messages so as not to reveal sensitive IP addresses")
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
return args
|
return args
|
||||||
|
|
Loading…
Reference in New Issue