"""Accountingflat connector."""

import functools

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

from ows_accounting import config
from ows_accounting.connectors import mysql


# please don't use the following private variables directly;
# use db_session
db_engine = create_engine(
    config.ACCOUNTINGFLAT_DB_CONNECTION_URL,
    connect_args={},
    poolclass=config.POOLCLASS)
_db_session = sessionmaker(bind=db_engine)
_session = _db_session()

db_session = functools.partial(mysql.db_session, _session)

BaseModel = declarative_base()