From 1a8148845d75da8626f57d7fd69c39ef15e87e3e Mon Sep 17 00:00:00 2001 From: YangZhang-GitHub <2533471770@qq.com> Date: Mon, 17 May 2021 16:59:42 +0800 Subject: [PATCH] The storebatch API has been completed --- myems-api/reports/storebatch.py | 243 ++++++++++++++++++++++++++++++++ 1 file changed, 243 insertions(+) create mode 100644 myems-api/reports/storebatch.py diff --git a/myems-api/reports/storebatch.py b/myems-api/reports/storebatch.py new file mode 100644 index 00000000..074722f5 --- /dev/null +++ b/myems-api/reports/storebatch.py @@ -0,0 +1,243 @@ +import falcon +import simplejson as json +import mysql.connector +import config +from anytree import Node, AnyNode, LevelOrderIter +from datetime import datetime, timedelta, timezone +from decimal import Decimal +import excelexporters.tenantbatch + + +class Reporting: + @staticmethod + def __init__(): + pass + + @staticmethod + def on_options(req, resp): + resp.status = falcon.HTTP_200 + + #################################################################################################################### + # PROCEDURES + # Step 1: valid parameters + # Step 2: build a space tree + # Step 3: query all tenants in the space tree + # Step 4: query energy categories + # Step 5: query reporting period energy input + # Step 6: construct the report + #################################################################################################################### + @staticmethod + def on_get(req, resp): + print(req.params) + space_id = req.params.get('spaceid') + reporting_period_start_datetime_local = req.params.get('reportingperiodstartdatetime') + reporting_period_end_datetime_local = req.params.get('reportingperiodenddatetime') + + ################################################################################################################ + # Step 1: valid parameters + ################################################################################################################ + if space_id is None: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_SPACE_ID') + else: + space_id = str.strip(space_id) + if not space_id.isdigit() or int(space_id) <= 0: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_SPACE_ID') + else: + space_id = int(space_id) + + timezone_offset = int(config.utc_offset[1:3]) * 60 + int(config.utc_offset[4:6]) + if config.utc_offset[0] == '-': + timezone_offset = -timezone_offset + + if reporting_period_start_datetime_local is None: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + description="API.INVALID_REPORTING_PERIOD_START_DATETIME") + else: + reporting_period_start_datetime_local = str.strip(reporting_period_start_datetime_local) + try: + reporting_start_datetime_utc = datetime.strptime(reporting_period_start_datetime_local, + '%Y-%m-%dT%H:%M:%S') + except ValueError: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + description="API.INVALID_REPORTING_PERIOD_START_DATETIME") + reporting_start_datetime_utc = reporting_start_datetime_utc.replace(tzinfo=timezone.utc) - \ + timedelta(minutes=timezone_offset) + + if reporting_period_end_datetime_local is None: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + description="API.INVALID_REPORTING_PERIOD_END_DATETIME") + else: + reporting_period_end_datetime_local = str.strip(reporting_period_end_datetime_local) + try: + reporting_end_datetime_utc = datetime.strptime(reporting_period_end_datetime_local, + '%Y-%m-%dT%H:%M:%S') + except ValueError: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + description="API.INVALID_REPORTING_PERIOD_END_DATETIME") + reporting_end_datetime_utc = reporting_end_datetime_utc.replace(tzinfo=timezone.utc) - \ + timedelta(minutes=timezone_offset) + + if reporting_start_datetime_utc >= reporting_end_datetime_utc: + raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + description='API.INVALID_REPORTING_PERIOD_END_DATETIME') + + # if reporting_start_datetime_utc + timedelta(minutes=15) >= reporting_end_datetime_utc: + # raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', + # description='API.THE_REPORTING_PERIOD_MUST_BE_LONGER_THAN_15_MINUTES') + + cnx_system_db = mysql.connector.connect(**config.myems_system_db) + cursor_system_db = cnx_system_db.cursor(dictionary=True) + + cursor_system_db.execute(" SELECT name " + " FROM tbl_spaces " + " WHERE id = %s ", (space_id,)) + row = cursor_system_db.fetchone() + + if row is None: + if cursor_system_db: + cursor_system_db.close() + if cnx_system_db: + cnx_system_db.disconnect() + raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', + description='API.SPACE_NOT_FOUND') + else: + space_name = row['name'] + + ################################################################################################################ + # Step 2: build a space tree + ################################################################################################################ + + query = (" SELECT id, name, parent_space_id " + " FROM tbl_spaces " + " ORDER BY id ") + cursor_system_db.execute(query) + rows_spaces = cursor_system_db.fetchall() + node_dict = dict() + if rows_spaces is not None and len(rows_spaces) > 0: + for row in rows_spaces: + parent_node = node_dict[row['parent_space_id']] if row['parent_space_id'] is not None else None + node_dict[row['id']] = AnyNode(id=row['id'], parent=parent_node, name=row['name']) + + ################################################################################################################ + # Step 3: query all tenants in the space tree + ################################################################################################################ + store_dict = dict() + space_dict = dict() + + for node in LevelOrderIter(node_dict[space_id]): + space_dict[node.id] = node.name + + cursor_system_db.execute(" SELECT store.id, store.name AS store_name, s.name AS space_name, " + " cc.name AS cost_center_name, store.description " + " FROM tbl_spaces s, tbl_spaces_stores ss, tbl_stores store, tbl_cost_centers cc " + " WHERE s.id IN ( " + ', '.join(map(str, space_dict.keys())) + ") " + " AND ss.space_id = s.id AND ss.store_id = store.id " + " AND store.cost_center_id = cc.id ", ) + rows_stores = cursor_system_db.fetchall() + if rows_stores is not None and len(rows_stores) > 0: + for row in rows_stores: + store_dict[row['id']] = {"store_name": row['store_name'], + "space_name": row['space_name'], + "cost_center_name": row['cost_center_name'], + "description": row['description'], + "values": list()} + + ################################################################################################################ + # Step 4: query energy categories + ################################################################################################################ + cnx_energy_db = mysql.connector.connect(**config.myems_energy_db) + cursor_energy_db = cnx_energy_db.cursor() + + # query energy categories in reporting period + energy_category_set = set() + cursor_energy_db.execute(" SELECT DISTINCT(energy_category_id) " + " FROM tbl_store_input_category_hourly " + " WHERE start_datetime_utc >= %s AND start_datetime_utc < %s ", + (reporting_start_datetime_utc, reporting_end_datetime_utc)) + rows_energy_categories = cursor_energy_db.fetchall() + if rows_energy_categories is not None or len(rows_energy_categories) > 0: + for row_energy_category in rows_energy_categories: + energy_category_set.add(row_energy_category[0]) + + # query all energy categories + cursor_system_db.execute(" SELECT id, name, unit_of_measure " + " FROM tbl_energy_categories " + " ORDER BY id ", ) + rows_energy_categories = cursor_system_db.fetchall() + if rows_energy_categories is None or len(rows_energy_categories) == 0: + if cursor_system_db: + cursor_system_db.close() + if cnx_system_db: + cnx_system_db.disconnect() + + if cursor_energy_db: + cursor_energy_db.close() + if cnx_energy_db: + cnx_energy_db.disconnect() + + raise falcon.HTTPError(falcon.HTTP_404, + title='API.NOT_FOUND', + description='API.ENERGY_CATEGORY_NOT_FOUND') + energy_category_list = list() + for row_energy_category in rows_energy_categories: + if row_energy_category['id'] in energy_category_set: + energy_category_list.append({"id": row_energy_category['id'], + "name": row_energy_category['name'], + "unit_of_measure": row_energy_category['unit_of_measure']}) + + ################################################################################################################ + # Step 5: query reporting period energy input + ################################################################################################################ + for store_id in store_dict: + + cursor_energy_db.execute(" SELECT energy_category_id, SUM(actual_value) " + " FROM tbl_store_input_category_hourly " + " WHERE store_id = %s " + " AND start_datetime_utc >= %s " + " AND start_datetime_utc < %s " + " GROUP BY energy_category_id ", + (store_id, + reporting_start_datetime_utc, + reporting_end_datetime_utc)) + rows_tenant_energy = cursor_energy_db.fetchall() + for energy_category in energy_category_list: + subtotal = Decimal(0.0) + for row_tenant_energy in rows_tenant_energy: + if energy_category['id'] == row_tenant_energy[0]: + subtotal = row_tenant_energy[1] + break + store_dict[store_id]['values'].append(subtotal) + + if cursor_system_db: + cursor_system_db.close() + if cnx_system_db: + cnx_system_db.disconnect() + + if cursor_energy_db: + cursor_energy_db.close() + if cnx_energy_db: + cnx_energy_db.disconnect() + + ################################################################################################################ + # Step 6: construct the report + ################################################################################################################ + store_list = list() + for store_id, store in store_dict.items(): + store_list.append({ + "id": store_id, + "store_name": store['store_name'], + "space_name": store['space_name'], + "cost_center_name": store['cost_center_name'], + "description": store['description'], + "values": store['values'], + }) + + result = {'stores': store_list, + 'energycategories': energy_category_list} + + # export result to Excel file and then encode the file to base64 string + # result['excel_bytes_base64'] = excelexporters.storebatch.export(result, + # space_name, + # reporting_period_start_datetime_local, + # reporting_period_end_datetime_local) + resp.body = json.dumps(result)