def spike_init():
logging.info("Initializing Spike")
timestamp = int(time())
app = create_app(__get_config_file())
db.init_app(app)
with app.app_context():
db.create_all()
for r in rulesets_seeds:
logging.info("Adding ruleset: %s", r)
rmks = "Ruleset for %s / auto-created %s" % (
r, strftime("%F - %H:%M", localtime(time())))
db.session.add(NaxsiRuleSets(r, rmks, timestamp))
for w in whitelists_seeds:
logging.info("Adding whitelistset: %s", w)
rmks = "Ruleset for %s / auto-created %s" % (
w, strftime("%F - %H:%M", localtime(time())))
db.session.add(NaxsiWhitelistSets(w, rmks, timestamp))
try:
db.session.commit()
except SQLAlchemyError:
logging.error(
'It seems that the database was already initialized. Did you meant to run `%s run` instead?',
sys.argv[0])
logging.info('Spike initialization completed')
def create_app(config_filename=''):
logging.info("Spike app.init()")
app = Flask(__name__)
if config_filename:
app.config.from_pyfile(config_filename)
if not app.config["SECRET_KEY"]:
app.config["SECRET_KEY"] = base64.b64encode(os.urandom(128))
app.config["SQLALCHEMY_BINDS"] = {'rules': 'sqlite:///rules.db'}
db.init_app(app)
db.app = app
Bootstrap(app) # add bootstrap templates and css
# add blueprints
app.register_blueprint(default.default)
app.register_blueprint(rules.rules, url_prefix='/rules')
app.register_blueprint(rulesets.rulesets, url_prefix='/rulesets')
app.register_blueprint(sandbox.sandbox, url_prefix='/sandbox')
app.register_blueprint(whitelists.whitelists, url_prefix='/whitelists')
app.register_blueprint(whitelistsets.whitelistsets, url_prefix='/whitelistsets')
# register filters
app.jinja_env.globals['version'] = version
return app
def create_app(config_filename=''):
logging.info("Spike app.init()")
app = Flask(__name__)
if config_filename:
app.config.from_pyfile(config_filename)
if not app.config["SECRET_KEY"]:
app.config["SECRET_KEY"] = base64.b64encode(os.urandom(128))
app.config["SQLALCHEMY_BINDS"] = {'rules': 'sqlite:///rules.db'}
db.init_app(app)
db.app = app
Bootstrap(app) # add bootstrap templates and css
# add blueprints
app.register_blueprint(default.default)
app.register_blueprint(rules.rules, url_prefix='/rules')
app.register_blueprint(rulesets.rulesets, url_prefix='/rulesets')
app.register_blueprint(sandbox.sandbox, url_prefix='/sandbox')
app.register_blueprint(whitelists.whitelists, url_prefix='/whitelists')
app.register_blueprint(whitelistsets.whitelistsets,
url_prefix='/whitelistsets')
# register filters
app.jinja_env.globals['version'] = version
return app
def setUp(self):
app = create_app()
db.init_app(app)
app.config['TESTING'] = True
self.app = app.test_client()
self.created_rules = list()
self.__create_rule()
def run(debug=False):
app = create_app(__get_config_file())
db.init_app(app)
if debug:
app.test_request_context().push()
try:
host = app.config["APP_HOST"]
except KeyError:
host = '0.0.0.0'
try:
port = int(app.config["APP_PORT"])
except KeyError:
port = 5555
app.run(debug=debug, host=host, port=port)
def run(debug=False):
app = create_app(__get_config_file())
db.init_app(app)
if debug:
app.test_request_context().push()
try:
host = app.config["APP_HOST"]
except KeyError:
host = '127.0.0.1'
try:
port = int(app.config["APP_PORT"])
except KeyError:
port = 5555
app.run(debug=debug, host=host, port=port)
def spike_init():
logging.info("Initializing Spike")
timestamp = int(time())
app = create_app(__get_config_file())
db.init_app(app)
with app.app_context():
db.create_all()
for r in rulesets_seeds:
logging.info("Adding ruleset: %s", r)
rmks = "Ruleset for %s / auto-created %s" % (r, strftime("%F - %H:%M", localtime(time())))
db.session.add(NaxsiRuleSets(r, rmks, timestamp))
for w in whitelists_seeds:
logging.info("Adding whitelistset: %s", w)
rmks = "Ruleset for %s / auto-created %s" % (w, strftime("%F - %H:%M", localtime(time())))
db.session.add(NaxsiWhitelistSets(w, rmks, timestamp))
try:
db.session.commit()
except SQLAlchemyError:
logging.error('It seems that the database was already initialized. Did you meant to run `%s run` instead?',
sys.argv[0])
logging.info('Spike initialization completed')
def setUp(self):
app = create_app()
db.init_app(app)
app.config['TESTING'] = True
self.app = app.test_client()
self.wid = self.__create_whitelist()
def setUp(self):
app = create_app()
db.init_app(app)
app.config['TESTING'] = True
self.app = app.test_client()
self.wid = self.__create_whitelist()
def setUp(self):
app = create_app()
db.init_app(app)
app.config['TESTING'] = True
self.app = app.test_client()
def setUp(self):
app = create_app('../config.cfg')
db.init_app(app)
app.config['TESTING'] = True
self.app = app.test_client()