def force_reset_config(): """ Resets the values for the CRIPTsConfig class by dropping the database collection and then saving a new default CRIPTsConfig. """ print "Resetting CRIPTs configuration settings." CRIPTsConfig.drop_collection(); cripts_config = CRIPTsConfig(); cripts_config.save();
def get_config(): """ Get the CRIPTs configuration. :returns: :class:`cripts.config.config.CRIPTsConfig` """ cripts_config = CRIPTsConfig.objects().first() if not cripts_config: cripts_config = CRIPTsConfig() cripts_config.save() return cripts_config
def create_config_if_not_exist(): """ If the CRIPTsConfig already exists then the CRIPTsConfig is returned, otherwise a new CRIPTsConfig will be created, saved, and returned. Returns: Returns the CRIPTsConfig if it already exists, otherwise a default CRIPTsConfig is returned. """ cripts_config = CRIPTsConfig.objects().first() if not cripts_config: print "Creating a new CRIPTs configuration." cripts_config = CRIPTsConfig() cripts_config.save() else: print "A CRPTs configuration already exists. Skipping default creation." return cripts_config
def modify_configuration(forms, analyst): """ Modify the configuration with the submitted changes. :param config_form: The form data. :type config_form: dict :param analyst: The user making the modifications. :type analyst: str :returns: dict with key "message" """ config = CRIPTsConfig.objects().first() if not config: config = CRIPTsConfig() data = None for form in forms: if not data: data = form.cleaned_data else: data.update(form.cleaned_data) # data = config_form.cleaned_data allowed_hosts_list = data['allowed_hosts'].split(',') allowed_hosts = () for allowed_host in allowed_hosts_list: allowed_hosts = allowed_hosts + (allowed_host.strip(), ) data['allowed_hosts'] = allowed_hosts service_dirs_list = data['service_dirs'].split(',') service_dirs = () for service_dir in service_dirs_list: service_dirs = service_dirs + (service_dir.strip(), ) data['service_dirs'] = service_dirs config.merge(data, overwrite=True) try: config.save(username=analyst) return {'message': "Success!"} except Exception, e: return {'message': "Failure: %s" % e}
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config cripts_config = CRIPTsConfig() cripts_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRIPTsUser.create_user( username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL, ) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TSRC, analyst=TUSER_NAME) obj.save() # Add another with Different source obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TUNKSRC, analyst=TRANDUSER) obj.save() # Add test non-source object obj = TestObject() obj.name = TOBJ_NAME obj.value = TOBJ_VALUE obj.save()
def prep_db(): """ Prep the DB for testing. """ clean_db() # Create a new default config cripts_config = CRIPTsConfig() cripts_config.save() # Add Source handlers.add_new_source(TSRC, TRANDUSER) # Add User user = CRIPTsUser.create_user(username=TUSER_NAME, password=TUSER_PASS, email=TUSER_EMAIL, ) user.first_name = TUSER_FNAME user.last_name = TUSER_LNAME user.save() # Add test source object obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TSRC, analyst=TUSER_NAME) obj.save() # Add another with Different source obj = TestSourceObject() obj.name = TOBJS_NAME obj.value = TOBJS_VALUE obj.add_source(source=TUNKSRC, analyst=TRANDUSER) obj.save() # Add test non-source object obj = TestObject() obj.name = TOBJ_NAME obj.value = TOBJ_VALUE obj.save()