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 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()
