hub/imports/physics_factory.py

"""
PhysicsFactory retrieve the specific physics module for the given region
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
"""
from imports.physics_feeders.us_new_york_city_physics_parameters import UsNewYorkCityPhysicsParameters
from imports.physics_feeders.us_physics_parameters import UsPhysicsParameters
from imports.physics_feeders.ca_physics_parameters import CaPhysicsParameters
from pathlib import Path


class PhysicsFactory:
  """
  PhysicsFactor class
  """
  def __init__(self, handler, city, base_path=Path(Path(__file__).parent.parent / 'data/physics')):
    self._handler = '_' + handler.lower().replace(' ', '_')
    self._city = city
    self._base_path = base_path

  def _us_new_york(self):
    UsNewYorkCityPhysicsParameters(self._city, self._base_path).enrich_buildings()

  def _us(self):
    UsPhysicsParameters(self._city, self._base_path).enrich_buildings()

  def _ca(self):
    CaPhysicsParameters(self._city, self._base_path).enrich_buildings()

  def _de(self):
    raise NotImplementedError

  def _es(self):
    raise NotImplementedError

  def enrich(self):
    """
    Enrich the city with the physics information
    :return: None
    """
    getattr(self, self._handler, lambda: None)()