city_retrofit/hub/exports/db_factory.py

"""
DBFactory performs read related operations
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2022 Concordia CERC group
Project CoderPeter Yefi peteryefi@gmail.com
"""
from hub.persistence import City
from hub.persistence import Application
from hub.persistence import User
from hub.persistence import CityObject
from hub.persistence import SimulationResults


class DBFactory:
  """
  DBFactory class
  """

  def __init__(self, db_name, app_env, dotenv_path):
    self._city = City(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._application = Application(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._user = User(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._city_object = CityObject(db_name=db_name, app_env=app_env, dotenv_path=dotenv_path)
    self._simulation_results = SimulationResults(db_name=db_name, dotenv_path=dotenv_path, app_env=app_env)

  def get_city(self, city_id):
    """
    Retrieve a single city from postgres
    :param city_id: the id of the city to get
    """
    return self._city.get_by_id(city_id)

  def get_city_by_name(self, city_name):
    """
    Retrieve a single city from postgres
    :param city_name: the name of the city to get
    """
    return self._city.get_by_name(city_name)

  def get_city_by_user(self, user_id):
    """
    Retrieve cities created by user
    :param user_id: the id of the user
    """
    return self._city.get_by_user(user_id)

  def get_simulation_results_by_id(self, sim_id):
    """
    Retrieve a single simulation results
    :param sim_id: the id of the simulation results to get
    """
    return self._simulation_results.get_simulation_results_by_id(sim_id)

  def get_simulation_results_by_name(self, name):
    """
    Retrieve a single simulation results
    :param name: the name of the simulation results to get
    """
    return self._simulation_results.get_simulation_results_by_name(name)

  def get_simulation_results_by_city_id(self, city_id):
    """
    Retrieve a single simulation results
    :param city_id: the city id of the simulation results to get
    """
    return self._simulation_results.get_simulation_results_by_city_id(city_id)

  def get_simulation_results_by_city_object_id(self, city_object_id):
    """
    Retrieve a single simulation results
    :param city_object_id: the city object id of the simulation results to get
    """
    return self._simulation_results.get_simulation_results_by_city_object_id(city_object_id)

  def application_info(self, application_uuid):
    return self._application.get_by_uuid(application_uuid)

  def user_info(self, name, password, application_id):
    return self._user.get_by_name_application_id_and_password(name, password, application_id)

  def user_login(self, name, password, application_uuid):
    return self._user.get_by_name_application_uuid_and_password(name, password, application_uuid)

  def building_info(self, name, city_id):
    return self._city_object.get_by_name_and_city(name, city_id)