added associated equipment data to combinedeuqipmentincome report in API

myems-api/excelexporters/combinedequipmentincome.py

@@ -1,6 +1,8 @@
 import base64
 import uuid
 import os
+from decimal import Decimal
+
 from openpyxl.chart import (
     PieChart,
     LineChart,
@@ -276,7 +278,7 @@ def generate_excel(report,
         current_row_number += 1
 
         ca_len = len(reporting_period_data['names'])
-        wssum = 0
+        wssum = Decimal(0.0)
         for i in range(0, ca_len):
             wssum = round(reporting_period_data['subtotals'][i], 2) + wssum
         for i in range(0, ca_len):
@@ -293,7 +295,9 @@ def generate_excel(report,
             ws['D' + str(current_row_number)].font = title_font
             ws['D' + str(current_row_number)].alignment = c_c_alignment
             ws['D' + str(current_row_number)].border = f_border
-            ws['D' + str(current_row_number)] = '{:.2%}'.format(round(reporting_period_data['subtotals'][i], 2) / wssum)
+            ws['D' + str(current_row_number)] = '{:.2%}'.format(round(
+                reporting_period_data['subtotals'][i], 2) / wssum) if wssum is not None and wssum != Decimal(0.0) \
+                else " "
 
             current_row_number += 1
 
@@ -449,6 +453,63 @@ def generate_excel(report,
         for i in range(37, 69 + 1):
             ws.row_dimensions[i].height = 0.1
 
+    #####################################
+
+    has_associated_equipment_flag = True
+
+    current_row_number += 2
+    if "associated_equipment" not in report.keys() or \
+            "energy_category_names" not in report['associated_equipment'].keys() or \
+            len(report['associated_equipment']["energy_category_names"]) == 0 \
+            or 'associated_equipment_names_array' not in report['associated_equipment'].keys() \
+            or report['associated_equipment']['associated_equipment_names_array'] is None \
+            or len(report['associated_equipment']['associated_equipment_names_array']) == 0 \
+            or len(report['associated_equipment']['associated_equipment_names_array'][0]) == 0:
+        has_associated_equipment_flag = False
+
+    if has_associated_equipment_flag:
+        associated_equipment = report['associated_equipment']
+
+        ws['B' + str(current_row_number)].font = title_font
+        ws['B' + str(current_row_number)] = name + ' 相关设备数据'
+
+        current_row_number += 1
+
+        ws.row_dimensions[current_row_number].height = 60
+        ws['B' + str(current_row_number)].fill = table_fill
+        ws['B' + str(current_row_number)].font = name_font
+        ws['B' + str(current_row_number)].alignment = c_c_alignment
+        ws['B' + str(current_row_number)].border = f_border
+        ws['B' + str(current_row_number)] = '相关设备'
+        ca_len = len(associated_equipment['energy_category_names'])
+
+        for i in range(0, ca_len):
+            row = chr(ord('C') + i)
+            ws[row + str(current_row_number)].fill = table_fill
+            ws[row + str(current_row_number)].font = name_font
+            ws[row + str(current_row_number)].alignment = c_c_alignment
+            ws[row + str(current_row_number)].border = f_border
+            ws[row + str(current_row_number)] = \
+                reporting_period_data['names'][i] + " (" + reporting_period_data['units'][i] + ")"
+
+        associated_equipment_len = len(associated_equipment['associated_equipment_names_array'][0])
+
+        for i in range(0, associated_equipment_len):
+            current_row_number += 1
+            row = str(current_row_number)
+
+            ws['B' + row].font = title_font
+            ws['B' + row].alignment = c_c_alignment
+            ws['B' + row] = associated_equipment['associated_equipment_names_array'][0][i]
+            ws['B' + row].border = f_border
+
+            for j in range(0, ca_len):
+                col = chr(ord('C') + j)
+                ws[col + row].font = title_font
+                ws[col + row].alignment = c_c_alignment
+                ws[col + row] = round(associated_equipment['subtotals_array'][j][i], 2)
+                ws[col + row].border = f_border
+
     filename = str(uuid.uuid4()) + '.xlsx'
     wb.save(filename)
 

myems-api/reports/combinedequipmentincome.py

@@ -23,11 +23,13 @@ class Reporting:
     # Step 2: query the combined equipment
     # Step 3: query energy categories
     # Step 4: query associated points
-    # Step 5: query base period energy income
+    # Step 5: query associated equipments
-    # Step 6: query reporting period energy income
+    # Step 6: query base period energy income
-    # Step 7: query tariff data
+    # Step 7: query reporting period energy income
-    # Step 8: query associated points data
+    # Step 8: query tariff data
-    # Step 9: construct the report
+    # Step 9: query associated points data
+    # Step 10: query associated equipments energy income
+    # Step 11: construct the report
     ####################################################################################################################
     @staticmethod
     def on_get(req, resp):
@@ -235,7 +237,20 @@ class Reporting:
                 point_list.append({"id": row[0], "name": row[1], "units": row[2], "object_type": row[3]})
 
         ################################################################################################################
-        # Step 5: query base period energy income
+        # Step 5: query associated equipments
+        ################################################################################################################
+        associated_equipment_list = list()
+        cursor_system.execute(" SELECT e.id, e.name "
+                              " FROM tbl_equipments e,tbl_combined_equipments_equipments ee"
+                              " WHERE ee.combined_equipment_id = %s AND e.id = ee.equipment_id"
+                              " ORDER BY id ", (combined_equipment['id'],))
+        rows_associated_equipments = cursor_system.fetchall()
+        if rows_associated_equipments is not None and len(rows_associated_equipments) > 0:
+            for row in rows_associated_equipments:
+                associated_equipment_list.append({"id": row[0], "name": row[1]})
+
+        ################################################################################################################
+        # Step 6: query base period energy income
         ################################################################################################################
         base = dict()
         if energy_category_set is not None and len(energy_category_set) > 0:
@@ -282,7 +297,7 @@ class Reporting:
                     base[energy_category_id]['subtotal'] += actual_value
 
         ################################################################################################################
-        # Step 8: query reporting period energy income
+        # Step 7: query reporting period energy income
         ################################################################################################################
         reporting = dict()
         if energy_category_set is not None and len(energy_category_set) > 0:
@@ -329,7 +344,7 @@ class Reporting:
                     reporting[energy_category_id]['subtotal'] += actual_value
 
         ################################################################################################################
-        # Step 9: query tariff data
+        # Step 8: query tariff data
         ################################################################################################################
         parameters_data = dict()
         parameters_data['names'] = list()
@@ -355,7 +370,7 @@ class Reporting:
                 parameters_data['values'].append(tariff_value_list)
 
         ################################################################################################################
-        # Step 10: query associated points data
+        # Step 9: query associated points data
         ################################################################################################################
         for point in point_list:
             point_values = []
@@ -420,7 +435,37 @@ class Reporting:
             parameters_data['values'].append(point_values)
 
         ################################################################################################################
-        # Step 12: construct the report
+        # Step 10: query associated equipments energy income
+        ################################################################################################################
+        associated_equipment_data = dict()
+
+        if energy_category_set is not None and len(energy_category_set) > 0:
+            for energy_category_id in energy_category_set:
+                associated_equipment_data[energy_category_id] = dict()
+                associated_equipment_data[energy_category_id]['associated_equipment_names'] = list()
+                associated_equipment_data[energy_category_id]['subtotals'] = list()
+                for associated_equipment in associated_equipment_list:
+                    associated_equipment_data[energy_category_id]['associated_equipment_names'].append(
+                        associated_equipment['name'])
+
+                    cursor_billing.execute(" SELECT SUM(actual_value) "
+                                           " FROM tbl_equipment_output_category_hourly "
+                                           " WHERE equipment_id = %s "
+                                           "     AND energy_category_id = %s "
+                                           "     AND start_datetime_utc >= %s "
+                                           "     AND start_datetime_utc < %s "
+                                           " ORDER BY start_datetime_utc ",
+                                           (associated_equipment['id'],
+                                            energy_category_id,
+                                            reporting_start_datetime_utc,
+                                            reporting_end_datetime_utc))
+                    row_subtotal = cursor_billing.fetchone()
+
+                    subtotal = Decimal(0.0) if (row_subtotal is None or row_subtotal[0] is None) else row_subtotal[0]
+                    associated_equipment_data[energy_category_id]['subtotals'].append(subtotal)
+
+        ################################################################################################################
+        # Step 11: construct the report
         ################################################################################################################
         if cursor_system:
             cursor_system.close()
@@ -497,6 +542,22 @@ class Reporting:
             "values": parameters_data['values']
         }
 
+        result['associated_equipment'] = dict()
+        result['associated_equipment']['energy_category_names'] = list()
+        result['associated_equipment']['units'] = list()
+        result['associated_equipment']['associated_equipment_names_array'] = list()
+        result['associated_equipment']['subtotals_array'] = list()
+        result['associated_equipment']['total_unit'] = config.currency_unit
+        if energy_category_set is not None and len(energy_category_set) > 0:
+            for energy_category_id in energy_category_set:
+                result['associated_equipment']['energy_category_names'].append(
+                    energy_category_dict[energy_category_id]['name'])
+                result['associated_equipment']['units'].append(config.currency_unit)
+                result['associated_equipment']['associated_equipment_names_array'].append(
+                    associated_equipment_data[energy_category_id]['associated_equipment_names'])
+                result['associated_equipment']['subtotals_array'].append(
+                    associated_equipment_data[energy_category_id]['subtotals'])
+
         # export result to Excel file and then encode the file to base64 string
         result['excel_bytes_base64'] = excelexporters.combinedequipmentincome.export(result,
                                                                                      combined_equipment['name'],