system_assignation/hub/imports/usage_factory.py

"""
UsageFactory retrieve the specific usage module for the given region
This factory can only be called after calling the construction factory so the thermal zones are created.
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2022 Concordia CERC group
Project Coder Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
Code contributors: Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
"""
import logging
from hub.imports.usage.comnet_usage_parameters import ComnetUsageParameters
from hub.imports.usage.nrcan_usage_parameters import NrcanUsageParameters
from hub.helpers.utils import validate_import_export_type


class UsageFactory:
  """
  UsageFactory class
  """
  def __init__(self, handler, city):
    self._handler = '_' + handler.lower().replace(' ', '_')
    class_funcs = validate_import_export_type(UsageFactory)
    if self._handler not in class_funcs:
      error_message = f"Wrong import type [{self._handler}]. Valid functions include {class_funcs}"
      logging.error(error_message)
      raise Exception(error_message)
    self._city = city

  def _comnet(self):
    """
    Enrich the city with COMNET usage library
    """
    ComnetUsageParameters(self._city).enrich_buildings()
    self._city.level_of_detail.usage = 2
    for building in self._city.buildings:
      building.level_of_detail.usage = 2

  def _nrcan(self):
    """
    Enrich the city with NRCAN usage library
    """
    NrcanUsageParameters(self._city).enrich_buildings()
    self._city.level_of_detail.usage = 2
    for building in self._city.buildings:
      building.level_of_detail.usage = 2

  def enrich(self):
    """
    Enrich the city given to the class using the usage factory given handler
    :return: None
    """
    getattr(self, self._handler, lambda: None)()