def activity_repository(app: Flask) -> ActivityCosmosDBRepository:
with app.app_context():
from commons.data_access_layer.cosmos_db import init_app, cosmos_helper
if cosmos_helper is None:
init_app(app)
return ActivityCosmosDBRepository()
def time_entry_repository(app: Flask) -> TimeEntryCosmosDBRepository:
with app.app_context():
from commons.data_access_layer.cosmos_db import init_app, cosmos_helper
if cosmos_helper is None:
init_app(app)
return TimeEntryCosmosDBRepository()
def cosmos_db_repository(app: Flask, cosmos_db_model) -> CosmosDBRepository:
with app.app_context():
from commons.data_access_layer.cosmos_db import init_app, cosmos_helper
if cosmos_helper is None:
init_app(app)
from commons.data_access_layer.cosmos_db import cosmos_helper
app.logger.info("Creating Cosmos DB test models...")
cosmos_helper.create_container_if_not_exists(cosmos_db_model)
app.logger.info("Cosmos DB test models created!")
yield CosmosDBRepository.from_definition(cosmos_db_model)
app.logger.info("Removing Cosmos DB test models...")
cosmos_helper.delete_container(cosmos_db_model["id"])
app.logger.info("Cosmos DB test models removed!")
def sql_repository(app: Flask, sql_model_class):
with app.app_context():
from commons.data_access_layer.sql import init_app, db
if db is None:
init_app(app)
from commons.data_access_layer.sql import db
db.metadata.create_all(bind=db.engine,
tables=[sql_model_class.__table__])
app.logger.info("SQl test models created!")
from commons.data_access_layer.sql import SQLRepository
yield SQLRepository(sql_model_class)
db.metadata.drop_all(bind=db.engine,
tables=[sql_model_class.__table__])
app.logger.info("SQL test models removed!")
def init_cosmos_db(app: Flask) -> None:
from commons.data_access_layer.cosmos_db import init_app
init_app(app)
def init_sql(app: Flask) -> None:
from commons.data_access_layer.sql import init_app
init_app(app)