def setup(config=None): for document in json.load(open(config)): database = document.get("database", document["index"]) create_database(database) engine = pg_engine(database=database) Base.metadata.drop_all(engine) Base.metadata.create_all(engine)
def setup(config=None): for schema in json.load(open(config)): create_database(schema.get('index')) # create schema engine = pg_engine(database=schema.get('index')) Base.metadata.drop_all(engine) Base.metadata.create_all(engine)
def setup(config=None): for document in json.load(open(config)): database = document.get('database', document['index']) create_database(database) # create schema engine = pg_engine(database=database) Base.metadata.drop_all(engine) Base.metadata.create_all(engine)
def setup(config=None): for document in json.load(open(config)): database = document.get("database", document["index"]) schema = document.get("schema", SCHEMA) create_database(database) engine = pg_engine(database=database) create_schema(engine, schema) engine = engine.connect().execution_options( schema_translate_map={None: schema}) Base.metadata.drop_all(engine) Base.metadata.create_all(engine)
def setup(config=None): for document in json.load(open(config)): database = document.get('database', document['index']) schema = document.get('schema', SCHEMA) create_database(database) create_extension(database, 'POSTGIS', echo=True) engine = pg_engine(database=database) create_schema(engine, schema) engine = engine.connect().execution_options( schema_translate_map={None: schema}) Base.metadata.drop_all(engine) Base.metadata.create_all(engine)
def test_create_database( self, mock_logger, mock_pg_engine, mock_pg_execute, connection, ): database = connection.engine.url.database mock_pg_engine.return_value = connection.engine create_database(database, echo=True) assert mock_logger.debug.call_count == 2 mock_logger.debug.assert_any_call(f"Creating database: {database}") mock_logger.debug.assert_any_call(f"Created database: {database}") mock_pg_engine.assert_any_call(database="postgres", echo=True) mock_pg_execute.assert_any_call( connection.engine, f"CREATE DATABASE {database}", )
def engine(dns): engine = sa.create_engine(dns) drop_database("testdb") create_database("testdb") yield engine engine.dispose()