hub/factories/occupancy_feeders/helpers/occupancy_helper.py

"""
Geometry helper
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
"""


class OccupancyHelper:
  occupancy_pluto_function = {
    'C1': 'C-12 Residential',
    'C5': 'C-12 Residential',
    'D3': 'C-12 Residential',
    'D6': 'C-12 Residential',
    'I1': 'C-2 Health',
    'RI': 'C-12 Residential',
    'RM': 'C-12 Residential',
    'U0': 'C-10 Warehouse',
    'W4': 'C-9 School',
  }
  occupancy_function = {
    'residential': 'C-12 Residential'
  }

  @staticmethod
  def pluto_occupancy_function(building_pluto_function):
    """
    Get nrel function from the given pluto function
    :param building_pluto_function: str
    :return: str
    """
    try:
      return OccupancyHelper.occupancy_pluto_function[building_pluto_function]
    except KeyError:
      return OccupancyHelper.occupancy_function[building_pluto_function]