From 0908e64782337f902be258da3e3a0d9ecfa738f1 Mon Sep 17 00:00:00 2001 From: Guille Date: Fri, 24 May 2024 11:18:58 +0200 Subject: [PATCH] Correct cerc_persistence autodocumentation --- cerc_persistence/repositories/city.py | 4 ++++ unittests/test_db_retrieve.py | 9 --------- 2 files changed, 4 insertions(+), 9 deletions(-) diff --git a/cerc_persistence/repositories/city.py b/cerc_persistence/repositories/city.py index 5737b69..9c9a2e9 100644 --- a/cerc_persistence/repositories/city.py +++ b/cerc_persistence/repositories/city.py @@ -133,3 +133,7 @@ class City(Repository): except SQLAlchemyError as err: logging.error('Error while fetching city by name %s', err) raise SQLAlchemyError from err + + def get_by_id(self, city_id) -> Model: + with Session(self.engine) as session: + return session.execute(select(Model).where(Model.id == city_id)).first()[0] \ No newline at end of file diff --git a/unittests/test_db_retrieve.py b/unittests/test_db_retrieve.py index 6dda2b0..f10a8cf 100644 --- a/unittests/test_db_retrieve.py +++ b/unittests/test_db_retrieve.py @@ -5,18 +5,9 @@ Copyright © 2022 Concordia CERC group Project Coder Ruben Sanchez ruben.sanchez@mail.concordia.ca """ import datetime -import distutils.spawn -import os import unittest -from pathlib import Path from unittest import TestCase -from sqlalchemy import create_engine -from sqlalchemy_utils import database_exists, create_database, drop_database - -from cerc_persistence.db_control import DBControl -from cerc_persistence.repository import Repository -from cerc_persistence.models import City, Application, CityObject, SimulationResults, User from unittests.control import Control control = Control()