2021-01-04 18:50:26 +01:00
|
|
|
"""Provide connection utils for the ORM layer.
|
|
|
|
|
|
|
|
This module defines fully configured `engine`, `connection`, and `session`
|
|
|
|
objects to be used as globals within the `urban_meal_delivery` package.
|
|
|
|
|
|
|
|
If a database is not guaranteed to be available, they are set to `None`.
|
|
|
|
That is the case on the CI server.
|
|
|
|
"""
|
|
|
|
|
|
|
|
import os
|
2020-08-09 03:45:19 +02:00
|
|
|
|
|
|
|
import sqlalchemy as sa
|
2021-01-06 16:17:05 +01:00
|
|
|
from sqlalchemy import engine as engine_mod
|
2020-08-09 03:45:19 +02:00
|
|
|
from sqlalchemy import orm
|
|
|
|
|
|
|
|
import urban_meal_delivery
|
|
|
|
|
|
|
|
|
2021-01-04 18:50:26 +01:00
|
|
|
if os.getenv('TESTING'):
|
2021-01-06 16:17:05 +01:00
|
|
|
# Specify the types explicitly to make mypy happy.
|
|
|
|
engine: engine_mod.Engine = None
|
|
|
|
connection: engine_mod.Connection = None
|
|
|
|
session: orm.Session = None
|
2020-08-09 03:45:19 +02:00
|
|
|
|
2021-01-04 18:50:26 +01:00
|
|
|
else: # pragma: no cover
|
|
|
|
engine = sa.create_engine(urban_meal_delivery.config.DATABASE_URI)
|
|
|
|
connection = engine.connect()
|
|
|
|
session = orm.sessionmaker(bind=connection)()
|