def get_engine(database='all'): global engines if engines == {}: from avconfig.ossimsetupconfig import AVOssimSetupConfigHandler config_file = "/etc/ossim/ossim_setup.conf" ossim_setup = AVOssimSetupConfigHandler(config_file) uri = "mysql://%s:%s@%s/" % (ossim_setup.get_database_user(), ossim_setup.get_database_pass(), ossim_setup.get_database_db_ip()) kwargs = { 'echo': False, 'poolclass': NullPool, 'connect_args': { 'cursorclass': MySQLdb.cursors.SSCursor } } engines = { 'alienvault': create_engine(uri + 'alienvault', **kwargs), 'alienvault_siem': create_engine(uri + 'alienvault_siem', **kwargs), 'avcenter': create_engine(uri + 'avcenter', **kwargs), 'alienvault_api': create_engine(uri + 'alienvault_api', **kwargs) } if database == 'all': return engines return engines.get(database)
def get_engine(database='any'): global engines if engines == {}: from avconfig.ossimsetupconfig import AVOssimSetupConfigHandler config_file = "/etc/ossim/ossim_setup.conf" ossim_setup = AVOssimSetupConfigHandler(config_file) uri = "mysql://%s:%s@%s/" % (ossim_setup.get_database_user(), ossim_setup.get_database_pass(), ossim_setup.get_database_db_ip()) kwargs = {'echo': False, 'poolclass': NullPool, 'pool_reset_on_return': None, 'connect_args': {'cursorclass': MySQLdb.cursors.SSCursor, 'autocommit': True}} engines = {'alienvault': create_engine(uri + 'alienvault', **kwargs), 'alienvault_siem': create_engine(uri + 'alienvault_siem', **kwargs), 'alienvault_api': create_engine(uri + 'alienvault_api', **kwargs)} if database == 'any': # Return any engine, as it is not really needed to specify one. return engines.get('alienvault') return engines.get(database)
def get_engine (database='all'): global engines if engines == {}: from avconfig.ossimsetupconfig import AVOssimSetupConfigHandler config_file = "/etc/ossim/ossim_setup.conf" ossim_setup = AVOssimSetupConfigHandler(config_file) uri = "mysql://%s:%s@%s/" % (ossim_setup.get_database_user(), ossim_setup.get_database_pass(), ossim_setup.get_database_db_ip()) kwargs = {'echo': False, 'poolclass': NullPool, 'connect_args': {'cursorclass': MySQLdb.cursors.SSCursor}} engines = {'alienvault': create_engine(uri + 'alienvault', **kwargs), 'alienvault_siem': create_engine(uri + 'alienvault_siem', **kwargs), 'avcenter': create_engine(uri + 'avcenter', **kwargs), 'alienvault_api': create_engine(uri + 'alienvault_api', **kwargs)} if database == 'all': return engines return engines.get(database)