simpleml.utils.initialization module

Util module to initialize SimpleML and configure database management

class simpleml.utils.initialization.Database(database='SimpleML', username='simpleml', password='simpleml', drivername='postgresql', host='localhost', port=5432, **kwargs)

Bases: sqlalchemy.engine.url.URL

Basic configuration to interact with database

alembic_config

create_tables(base, drop_tables=False, ignore_errors=False)

Creates database tables (and potentially drops existing ones). Assumes to be running under a sufficiently privileged user

Parameters:drop_tables – Whether or not to drop the existing tables first. Returns:None

downgrade(revision)

Proxy Method to invoke alembic downgrade command to specified revision

engine

initialize(base_list=None, **kwargs)

Initialization method to set up database connection and inject session manager

Parameters:drop_tables – Bool, whether to drop existing tables in database Returns:None

upgrade(revision='head')

Proxy Method to invoke alembic upgrade command to specified revision