CyberSecAI / customization.py
invincible-jha's picture
Upload 8 files
de38b77 verified
class Customization:
def __init__(self):
self.alert_thresholds = {}
self.notification_methods = {}
self.custom_alert_rules = []
self.input_processing_settings = {}
self.model_settings = {}
def set_alert_threshold(self, alert_type, threshold):
self.alert_thresholds[alert_type] = threshold
def get_alert_threshold(self, alert_type):
return self.alert_thresholds.get(alert_type, 0.8)
def customize_notification_method(self, method, custom_function):
self.notification_methods[method] = custom_function
def get_notification_methods(self):
return list(self.notification_methods.keys())
def create_custom_alert_rule(self, rule):
self.custom_alert_rules.append(rule)
def get_custom_alert_rules(self):
return self.custom_alert_rules
def set_input_processing_setting(self, setting, value):
self.input_processing_settings[setting] = value
def get_input_processing_setting(self, setting):
return self.input_processing_settings.get(setting, None)
def enable_model(self, model_name):
self.model_settings[model_name] = {"enabled": True}
def disable_model(self, model_name):
self.model_settings[model_name] = {"enabled": False}
def set_model_setting(self, model_name, setting, value):
if model_name not in self.model_settings:
self.model_settings[model_name] = {}
self.model_settings[model_name][setting] = value
def get_model_setting(self, model_name, setting):
if model_name in self.model_settings:
return self.model_settings[model_name].get(setting, None)
return None