diff --git a/src/database/prepare.py b/src/database/prepare.py index c805af1..94611ff 100644 --- a/src/database/prepare.py +++ b/src/database/prepare.py @@ -3,11 +3,13 @@ from sqlalchemy.future import Engine from . import models # noqa +HOUR_IN_SECONDS = 3600 + def get_engine(host: str, user: str, passwd: str, db: str) -> Engine: """Creates an engine for mariadb with pymysql as connector""" uri = f"mariadb+pymysql://{user}:{passwd}@{host}/{db}" - engine = sqlmodel.create_engine(uri) + engine = sqlmodel.create_engine(uri, pool_recycle=HOUR_IN_SECONDS) return engine