2021-11-28 02:36:52 +03:00

24 lines
610 B
Python

import model.postgres_model
import utils
from EDMCLogging import get_main_logger
import os
logger = get_main_logger()
env_choose = os.getenv('DB_NAME')
if env_choose == 'postgres':
logger.info('Using postgres DB')
from .import postgres_model as model
elif env_choose == 'sqlite':
logger.info('Using sqlite DB')
from .import sqlite_model as model
else:
logger.info('Using sqlite DB')
from . import sqlite_model as model
model.get_diff_action_id = utils.measure(model.get_diff_action_id)
model.get_activity_changes = utils.measure(model.get_activity_changes)