hub/imports/occupancy_feeders/helpers/schedules_helper.py

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


class SchedulesHelper:
  comnet_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',
  }
  comnet_function = {
    'residential': 'C-12 Residential'
  }

  @staticmethod
  def comnet_pluto_schedules_function(building_pluto_function):
    """
    Get Comnet function from the given pluto function
    :param building_pluto_function: str
    :return: str
    """
    try:
      return SchedulesHelper.comnet_pluto_function[building_pluto_function]
    except KeyError:
      return SchedulesHelper.comnet_function[building_pluto_function]