s_ranjbar/summer_course_2024
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

persistence performance improvements

Browse Source
This commit is contained in:
Guille Gutierrez 2023-11-20 15:38:49 +01:00
parent 601058fc5e
commit fca74eeb08
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
hub/persistence/db_control.py
View File

@ -75,6 +75,7 @@ class DBControl:
: :
""" """
cities = self._city.get_by_user_id_application_id_and_scenario(user_id, application_id, scenario) cities = self._city.get_by_user_id_application_id_and_scenario(user_id, application_id, scenario)
c = [c[0].id for c in cities]
for city in cities: for city in cities:
result = self.building_info(name, city[0].id) result = self.building_info(name, city[0].id)
if result is not None: if result is not None:
@ -90,6 +91,15 @@ class DBControl:
""" """
return self._city_object.get_by_name_or_alias_and_city(name, city_id) return self._city_object.get_by_name_or_alias_and_city(name, city_id)
def building_info_in_cities(self, name, cities) -> CityObject:
"""
Retrieve the building info from the database
:param name: Building name
:param cities: [City ID]
:return: CityObject
"""
return self._city_object.get_by_name_or_alias_in_cities(name, cities)
def buildings_info(self, request_values, city_id) -> [CityObject]: def buildings_info(self, request_values, city_id) -> [CityObject]:
""" """
Retrieve the buildings info from the database Retrieve the buildings info from the database