hub/imports/customized_imports_factory.py

"""
CustomizedImportsFactory is used to import any information using user customized formats
This factory can only be called after calling the construction factory.
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
"""
from pathlib import Path


class CustomizedImportsFactory:
  """
  CustomizedImportsFactory class
  """
  def __init__(self, importer_class, city, base_path=None):
    if base_path is None:
      base_path = Path(Path(__file__).parent.parent / 'data/customized_imports')
    self._importer_class = importer_class
    self._city = city
    self._base_path = base_path
    for building in city.buildings:
      if len(building.thermal_zones) == 0:
        raise Exception('It seems that the customized imports factory is being called before the construction factory. '
                        'Please ensure that the construction factory is called first.')

  def enrich(self):
    """
    Returns the class that will enrich the city given
    :return: Class
    """
    importer = self._importer_class(self._city, self._base_path)
    return importer.enrich_buildings()