api_v1.4/hub_api/buildings/meb.py

import json
from flask import Response, request
from flask_restful import Resource

from hub_api.helpers.session_helper import active_session, refresh_session
from hub_api.config import Config

class Meb(Resource, Config):
  def __init__(self):
    super().__init__()

  def get(self):
    """
    API call for requesting a specified list of enriched buildings
    """
    session_id = request.headers.get('session_id', None)
    token = request.headers.get('token', None)
    application_uuid = request.headers.get('application_uuid', None)

    if active_session(session_id, token, application_uuid):
      refresh_session(session_id, token, application_uuid)
      payload = request.get_json()
      buildings = {}

      for city in payload['cities']:
        buildings[city] = {}
        for building in payload['cities'][city]:
          buildings[city][building] = self.export_db_factory.building_info(building, city)

      # TODO: finish formatting buildings to match swagger documentation
      #      and change export_db_factory.building_info to use MEB database
      #      call when ready

      return Response(json.dumps({'result': 'succeed', 'results': buildings}), status=200)
    return Response(json.dumps({'error': 'unauthorized'}), status=403)