diff --git a/app/database.py b/app/database.py index b67e276..c547c36 100644 --- a/app/database.py +++ b/app/database.py @@ -5,9 +5,12 @@ def get_session(): # Initialize config_file = 'config.global.py' + db_params = configParser.getConfig('database_parameters', config_file) - url = configParser.getConfig('database_parameters', config_file)['database_url'] - engine = create_engine(url) + url = db_params['database_url'] + pool_size = int(db_params.get('pool_size', 25)) + max_overflow = int(db_params.get('max_overflow', 5)) + engine = create_engine(url, pool_size=pool_size, max_overflow=max_overflow) with Session(engine) as session: yield session