hub/factories/usage/usage_factory.py

"""
UsageFactory retrieve the specific usage module for the given region
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
"""
from factories.usage.usage_feeders.de_usage_parameters import DeUsageParameters
from factories.usage.usage_feeders.us_new_york_city_usage_parameters import UsNewYorkCityUsageParameters


class UsageFactory:
  """
  UsageFactory class
  """
  def __init__(self, handler, city):
    self._handler = '_' + handler.lower().replace(' ', '_')
    self._city = city
    self.factory()

  def _us_new_york(self):
    UsNewYorkCityUsageParameters(self._city)

  def _ca(self):
    raise Exception('Not implemented')

  def _de(self):
    DeUsageParameters(self._city)

  def _es(self):
    raise Exception('Not implemented')

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