mirror of
https://github.com/ilri/dspace-statistics-api.git
synced 2024-11-24 23:30:18 +01:00
Alan Orth
0650c5985e
All checks were successful
continuous-integration/drone/push Build is passing
See: https://spdx.github.io/spdx-spec/appendix-V-using-SPDX-short-identifiers-in-source-files/
39 lines
1007 B
Python
39 lines
1007 B
Python
# SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
import falcon
|
|
import psycopg2
|
|
import psycopg2.extras
|
|
|
|
from .config import (
|
|
DATABASE_HOST,
|
|
DATABASE_NAME,
|
|
DATABASE_PASS,
|
|
DATABASE_PORT,
|
|
DATABASE_USER,
|
|
)
|
|
|
|
|
|
class DatabaseManager:
|
|
"""Manage database connection."""
|
|
|
|
def __init__(self):
|
|
self._connection_uri = f"dbname={DATABASE_NAME} user={DATABASE_USER} password={DATABASE_PASS} host={DATABASE_HOST} port={DATABASE_PORT}"
|
|
|
|
def __enter__(self):
|
|
try:
|
|
self._connection = psycopg2.connect(
|
|
self._connection_uri, cursor_factory=psycopg2.extras.DictCursor
|
|
)
|
|
except psycopg2.OperationalError:
|
|
title = "500 Internal Server Error"
|
|
description = "Could not connect to database"
|
|
raise falcon.HTTPInternalServerError(title, description)
|
|
|
|
return self._connection
|
|
|
|
def __exit__(self, exc_type, exc_value, exc_traceback):
|
|
self._connection.close()
|
|
|
|
|
|
# vim: set sw=4 ts=4 expandtab:
|