Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
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)
Ejemplo n.º 3
0
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)