Issues (1656)

myems-api/reports/combinedequipmentincome.py (3 issues)

1
import re
2
from datetime import datetime, timedelta, timezone
3
from decimal import Decimal
4
import falcon
5
import mysql.connector
6
import simplejson as json
7
import config
8
import excelexporters.combinedequipmentincome
9
from core import utilities
10
from core.useractivity import access_control, api_key_control
11
12
13
class Reporting:
14
    def __init__(self):
15
        """"Initializes Reporting"""
16
        pass
17
18
    @staticmethod
19
    def on_options(req, resp):
20
        _ = req
21
        resp.status = falcon.HTTP_200
22
23
    ####################################################################################################################
24
    # PROCEDURES
25
    # Step 1: valid parameters
26
    # Step 2: query the combined equipment
27
    # Step 3: query energy categories
28
    # Step 4: query associated points
29
    # Step 5: query associated equipments
30
    # Step 6: query base period energy income
31
    # Step 7: query reporting period energy income
32
    # Step 8: query tariff data
33
    # Step 9: query associated points data
34
    # Step 10: query associated equipments energy income
35
    # Step 11: construct the report
36
    ####################################################################################################################
37
    @staticmethod
38
    def on_get(req, resp):
39
        if 'API-KEY' not in req.headers or \
40
                not isinstance(req.headers['API-KEY'], str) or \
41
                len(str.strip(req.headers['API-KEY'])) == 0:
42
            access_control(req)
43
        else:
44
            api_key_control(req)
45
        print(req.params)
46
        combined_equipment_id = req.params.get('combinedequipmentid')
47
        combined_equipment_uuid = req.params.get('combinedequipmentuuid')
48
        period_type = req.params.get('periodtype')
49
        base_period_start_datetime_local = req.params.get('baseperiodstartdatetime')
50
        base_period_end_datetime_local = req.params.get('baseperiodenddatetime')
51
        reporting_period_start_datetime_local = req.params.get('reportingperiodstartdatetime')
52
        reporting_period_end_datetime_local = req.params.get('reportingperiodenddatetime')
53
        language = req.params.get('language')
54
        quick_mode = req.params.get('quickmode')
55
56
        ################################################################################################################
57
        # Step 1: valid parameters
58
        ################################################################################################################
59
        if combined_equipment_id is None and combined_equipment_uuid is None:
60
            raise falcon.HTTPError(status=falcon.HTTP_400,
61
                                   title='API.BAD_REQUEST',
62
                                   description='API.INVALID_COMBINED_EQUIPMENT_ID')
63
64
        if combined_equipment_id is not None:
65
            combined_equipment_id = str.strip(combined_equipment_id)
66
            if not combined_equipment_id.isdigit() or int(combined_equipment_id) <= 0:
67
                raise falcon.HTTPError(status=falcon.HTTP_400,
68
                                       title='API.BAD_REQUEST',
69
                                       description='API.INVALID_COMBINED_EQUIPMENT_ID')
70
71
        if combined_equipment_uuid is not None:
72
            regex = re.compile(r'^[a-f0-9]{8}-?[a-f0-9]{4}-?4[a-f0-9]{3}-?[89ab][a-f0-9]{3}-?[a-f0-9]{12}\Z', re.I)
73
            match = regex.match(str.strip(combined_equipment_uuid))
74
            if not bool(match):
75
                raise falcon.HTTPError(status=falcon.HTTP_400,
76
                                       title='API.BAD_REQUEST',
77
                                       description='API.INVALID_COMBINED_EQUIPMENT_UUID')
78
79
        if period_type is None:
80
            raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
81
                                   description='API.INVALID_PERIOD_TYPE')
82
        else:
83
            period_type = str.strip(period_type)
84
            if period_type not in ['hourly', 'daily', 'weekly', 'monthly', 'yearly']:
85
                raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
86
                                       description='API.INVALID_PERIOD_TYPE')
87
88
        timezone_offset = int(config.utc_offset[1:3]) * 60 + int(config.utc_offset[4:6])
89
        if config.utc_offset[0] == '-':
90
            timezone_offset = -timezone_offset
91
92
        base_start_datetime_utc = None
93
        if base_period_start_datetime_local is not None and len(str.strip(base_period_start_datetime_local)) > 0:
94
            base_period_start_datetime_local = str.strip(base_period_start_datetime_local)
95
            try:
96
                base_start_datetime_utc = datetime.strptime(base_period_start_datetime_local, '%Y-%m-%dT%H:%M:%S')
97
            except ValueError:
98
                raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
99
                                       description="API.INVALID_BASE_PERIOD_START_DATETIME")
100
            base_start_datetime_utc = \
101
                base_start_datetime_utc.replace(tzinfo=timezone.utc) - timedelta(minutes=timezone_offset)
102
            # nomalize the start datetime
103
            if config.minutes_to_count == 30 and base_start_datetime_utc.minute >= 30:
104
                base_start_datetime_utc = base_start_datetime_utc.replace(minute=30, second=0, microsecond=0)
105
            else:
106
                base_start_datetime_utc = base_start_datetime_utc.replace(minute=0, second=0, microsecond=0)
107
108
        base_end_datetime_utc = None
109
        if base_period_end_datetime_local is not None and len(str.strip(base_period_end_datetime_local)) > 0:
110
            base_period_end_datetime_local = str.strip(base_period_end_datetime_local)
111
            try:
112
                base_end_datetime_utc = datetime.strptime(base_period_end_datetime_local, '%Y-%m-%dT%H:%M:%S')
113
            except ValueError:
114
                raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
115
                                       description="API.INVALID_BASE_PERIOD_END_DATETIME")
116
            base_end_datetime_utc = \
117
                base_end_datetime_utc.replace(tzinfo=timezone.utc) - timedelta(minutes=timezone_offset)
118
119
        if base_start_datetime_utc is not None and base_end_datetime_utc is not None and \
120
                base_start_datetime_utc >= base_end_datetime_utc:
121
            raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
122
                                   description='API.INVALID_BASE_PERIOD_END_DATETIME')
123
124
        if reporting_period_start_datetime_local is None:
125
            raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
126
                                   description="API.INVALID_REPORTING_PERIOD_START_DATETIME")
127
        else:
128
            reporting_period_start_datetime_local = str.strip(reporting_period_start_datetime_local)
129
            try:
130
                reporting_start_datetime_utc = datetime.strptime(reporting_period_start_datetime_local,
131
                                                                 '%Y-%m-%dT%H:%M:%S')
132
            except ValueError:
133
                raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
134
                                       description="API.INVALID_REPORTING_PERIOD_START_DATETIME")
135
            reporting_start_datetime_utc = \
136
                reporting_start_datetime_utc.replace(tzinfo=timezone.utc) - timedelta(minutes=timezone_offset)
137
            # nomalize the start datetime
138
            if config.minutes_to_count == 30 and reporting_start_datetime_utc.minute >= 30:
139
                reporting_start_datetime_utc = reporting_start_datetime_utc.replace(minute=30, second=0, microsecond=0)
140
            else:
141
                reporting_start_datetime_utc = reporting_start_datetime_utc.replace(minute=0, second=0, microsecond=0)
142
143
        if reporting_period_end_datetime_local is None:
144
            raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
145
                                   description="API.INVALID_REPORTING_PERIOD_END_DATETIME")
146
        else:
147
            reporting_period_end_datetime_local = str.strip(reporting_period_end_datetime_local)
148
            try:
149
                reporting_end_datetime_utc = datetime.strptime(reporting_period_end_datetime_local,
150
                                                               '%Y-%m-%dT%H:%M:%S').replace(tzinfo=timezone.utc) - \
151
                                             timedelta(minutes=timezone_offset)
152
            except ValueError:
153
                raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
154
                                       description="API.INVALID_REPORTING_PERIOD_END_DATETIME")
155
156
        if reporting_start_datetime_utc >= reporting_end_datetime_utc:
157
            raise falcon.HTTPError(status=falcon.HTTP_400, title='API.BAD_REQUEST',
158
                                   description='API.INVALID_REPORTING_PERIOD_END_DATETIME')
159
160
        # if turn quick mode on, do not return parameters data and excel file
161
        is_quick_mode = False
162
        if quick_mode is not None and \
163
                len(str.strip(quick_mode)) > 0 and \
164
                str.lower(str.strip(quick_mode)) in ('true', 't', 'on', 'yes', 'y'):
165
            is_quick_mode = True
166
167
        trans = utilities.get_translation(language)
168
        trans.install()
169
        _ = trans.gettext
170
171
        ################################################################################################################
172
        # Step 2: query the combined equipment
173
        ################################################################################################################
174
        cnx_system = mysql.connector.connect(**config.myems_system_db)
175
        cursor_system = cnx_system.cursor()
176
177
        cnx_billing = mysql.connector.connect(**config.myems_billing_db)
178
        cursor_billing = cnx_billing.cursor()
179
180
        cnx_historical = mysql.connector.connect(**config.myems_historical_db)
181
        cursor_historical = cnx_historical.cursor()
182
183
        if combined_equipment_id is not None:
184
            cursor_system.execute(" SELECT id, name, cost_center_id "
185
                                  " FROM tbl_combined_equipments "
186
                                  " WHERE id = %s ", (combined_equipment_id,))
187
            row_combined_equipment = cursor_system.fetchone()
188
        elif combined_equipment_uuid is not None:
189
            cursor_system.execute(" SELECT id, name, cost_center_id "
190
                                  " FROM tbl_combined_equipments "
191
                                  " WHERE uuid = %s ", (combined_equipment_uuid,))
192
            row_combined_equipment = cursor_system.fetchone()
193
194
        if row_combined_equipment is None:
0 ignored issues
show
The variable row_combined_equipment does not seem to be defined for all execution paths.
Loading history...
195
            if cursor_system:
196
                cursor_system.close()
197
            if cnx_system:
198
                cnx_system.close()
199
200
            if cursor_billing:
201
                cursor_billing.close()
202
            if cnx_billing:
203
                cnx_billing.close()
204
205
            if cursor_historical:
206
                cursor_historical.close()
207
            if cnx_historical:
208
                cnx_historical.close()
209
            raise falcon.HTTPError(status=falcon.HTTP_404,
210
                                   title='API.NOT_FOUND',
211
                                   description='API.COMBINED_EQUIPMENT_NOT_FOUND')
212
213
        combined_equipment = dict()
214
        combined_equipment['id'] = row_combined_equipment[0]
215
        combined_equipment['name'] = row_combined_equipment[1]
216
        combined_equipment['cost_center_id'] = row_combined_equipment[2]
217
218
        ################################################################################################################
219
        # Step 3: query energy categories
220
        ################################################################################################################
221
        energy_category_set = set()
222
        # query energy categories in base period
223
        cursor_billing.execute(" SELECT DISTINCT(energy_category_id) "
224
                               " FROM tbl_combined_equipment_output_category_hourly "
225
                               " WHERE combined_equipment_id = %s "
226
                               "     AND start_datetime_utc >= %s "
227
                               "     AND start_datetime_utc < %s ",
228
                               (combined_equipment['id'], base_start_datetime_utc, base_end_datetime_utc))
229
        rows_energy_categories = cursor_billing.fetchall()
230
        if rows_energy_categories is not None and len(rows_energy_categories) > 0:
231
            for row_energy_category in rows_energy_categories:
232
                energy_category_set.add(row_energy_category[0])
233
234
        # query energy categories in reporting period
235
        cursor_billing.execute(" SELECT DISTINCT(energy_category_id) "
236
                               " FROM tbl_combined_equipment_output_category_hourly "
237
                               " WHERE combined_equipment_id = %s "
238
                               "     AND start_datetime_utc >= %s "
239
                               "     AND start_datetime_utc < %s ",
240
                               (combined_equipment['id'], reporting_start_datetime_utc, reporting_end_datetime_utc))
241
        rows_energy_categories = cursor_billing.fetchall()
242
        if rows_energy_categories is not None and len(rows_energy_categories) > 0:
243
            for row_energy_category in rows_energy_categories:
244
                energy_category_set.add(row_energy_category[0])
245
246
        # query all energy categories in base period and reporting period
247
        cursor_system.execute(" SELECT id, name, unit_of_measure, kgce, kgco2e "
248
                              " FROM tbl_energy_categories "
249
                              " ORDER BY id ", )
250
        rows_energy_categories = cursor_system.fetchall()
251
        if rows_energy_categories is None or len(rows_energy_categories) == 0:
252
            if cursor_system:
253
                cursor_system.close()
254
            if cnx_system:
255
                cnx_system.close()
256
257
            if cursor_billing:
258
                cursor_billing.close()
259
            if cnx_billing:
260
                cnx_billing.close()
261
262
            if cursor_historical:
263
                cursor_historical.close()
264
            if cnx_historical:
265
                cnx_historical.close()
266
            raise falcon.HTTPError(status=falcon.HTTP_404,
267
                                   title='API.NOT_FOUND',
268
                                   description='API.ENERGY_CATEGORY_NOT_FOUND')
269
        energy_category_dict = dict()
270
        for row_energy_category in rows_energy_categories:
271
            if row_energy_category[0] in energy_category_set:
272
                energy_category_dict[row_energy_category[0]] = {"name": row_energy_category[1],
273
                                                                "unit_of_measure": row_energy_category[2],
274
                                                                "kgce": row_energy_category[3],
275
                                                                "kgco2e": row_energy_category[4]}
276
277
        ################################################################################################################
278
        # Step 4: query associated points
279
        ################################################################################################################
280
        point_list = list()
281
        cursor_system.execute(" SELECT p.id, ep.name, p.units, p.object_type  "
282
                              " FROM tbl_combined_equipments e, tbl_combined_equipments_parameters ep, tbl_points p "
283
                              " WHERE e.id = %s AND e.id = ep.combined_equipment_id AND ep.parameter_type = 'point' "
284
                              "       AND ep.point_id = p.id "
285
                              " ORDER BY p.id ", (combined_equipment['id'],))
286
        rows_points = cursor_system.fetchall()
287
        if rows_points is not None and len(rows_points) > 0:
288
            for row in rows_points:
289
                point_list.append({"id": row[0], "name": row[1], "units": row[2], "object_type": row[3]})
290
291
        ################################################################################################################
292
        # Step 5: query associated equipments
293
        ################################################################################################################
294
        associated_equipment_list = list()
295
        cursor_system.execute(" SELECT e.id, e.name "
296
                              " FROM tbl_equipments e,tbl_combined_equipments_equipments ee"
297
                              " WHERE ee.combined_equipment_id = %s AND e.id = ee.equipment_id"
298
                              " ORDER BY id ", (combined_equipment['id'],))
299
        rows_associated_equipments = cursor_system.fetchall()
300
        if rows_associated_equipments is not None and len(rows_associated_equipments) > 0:
301
            for row in rows_associated_equipments:
302
                associated_equipment_list.append({"id": row[0], "name": row[1]})
303
304
        ################################################################################################################
305
        # Step 6: query base period energy income
306
        ################################################################################################################
307
        base = dict()
308
        if energy_category_set is not None and len(energy_category_set) > 0:
309
            for energy_category_id in energy_category_set:
310
                base[energy_category_id] = dict()
311
                base[energy_category_id]['timestamps'] = list()
312
                base[energy_category_id]['values'] = list()
313
                base[energy_category_id]['subtotal'] = Decimal(0.0)
314
315
                cursor_billing.execute(" SELECT start_datetime_utc, actual_value "
316
                                       " FROM tbl_combined_equipment_output_category_hourly "
317
                                       " WHERE combined_equipment_id = %s "
318
                                       "     AND energy_category_id = %s "
319
                                       "     AND start_datetime_utc >= %s "
320
                                       "     AND start_datetime_utc < %s "
321
                                       " ORDER BY start_datetime_utc ",
322
                                       (combined_equipment['id'],
323
                                        energy_category_id,
324
                                        base_start_datetime_utc,
325
                                        base_end_datetime_utc))
326
                rows_combined_equipment_hourly = cursor_billing.fetchall()
327
328
                rows_combined_equipment_periodically = \
329
                    utilities.aggregate_hourly_data_by_period(rows_combined_equipment_hourly,
330
                                                              base_start_datetime_utc,
331
                                                              base_end_datetime_utc,
332
                                                              period_type)
333
                for row_combined_equipment_periodically in rows_combined_equipment_periodically:
334
                    current_datetime_local = row_combined_equipment_periodically[0].replace(tzinfo=timezone.utc) + \
335
                                             timedelta(minutes=timezone_offset)
336
                    if period_type == 'hourly':
337
                        current_datetime = current_datetime_local.isoformat()[0:19]
338
                    elif period_type == 'daily':
339
                        current_datetime = current_datetime_local.isoformat()[0:10]
340
                    elif period_type == 'weekly':
341
                        current_datetime = current_datetime_local.isoformat()[0:10]
342
                    elif period_type == 'monthly':
343
                        current_datetime = current_datetime_local.isoformat()[0:7]
344
                    elif period_type == 'yearly':
345
                        current_datetime = current_datetime_local.isoformat()[0:4]
346
347
                    actual_value = Decimal(0.0) if row_combined_equipment_periodically[1] is None \
348
                        else row_combined_equipment_periodically[1]
349
                    base[energy_category_id]['timestamps'].append(current_datetime)
0 ignored issues
show
The variable current_datetime does not seem to be defined for all execution paths.
Loading history...
350
                    base[energy_category_id]['values'].append(actual_value)
351
                    base[energy_category_id]['subtotal'] += actual_value
352
353
        ################################################################################################################
354
        # Step 7: query reporting period energy income
355
        ################################################################################################################
356
        reporting = dict()
357
        if energy_category_set is not None and len(energy_category_set) > 0:
358
            for energy_category_id in energy_category_set:
359
                reporting[energy_category_id] = dict()
360
                reporting[energy_category_id]['timestamps'] = list()
361
                reporting[energy_category_id]['values'] = list()
362
                reporting[energy_category_id]['subtotal'] = Decimal(0.0)
363
364
                cursor_billing.execute(" SELECT start_datetime_utc, actual_value "
365
                                       " FROM tbl_combined_equipment_output_category_hourly "
366
                                       " WHERE combined_equipment_id = %s "
367
                                       "     AND energy_category_id = %s "
368
                                       "     AND start_datetime_utc >= %s "
369
                                       "     AND start_datetime_utc < %s "
370
                                       " ORDER BY start_datetime_utc ",
371
                                       (combined_equipment['id'],
372
                                        energy_category_id,
373
                                        reporting_start_datetime_utc,
374
                                        reporting_end_datetime_utc))
375
                rows_combined_equipment_hourly = cursor_billing.fetchall()
376
377
                rows_combined_equipment_periodically = \
378
                    utilities.aggregate_hourly_data_by_period(rows_combined_equipment_hourly,
379
                                                              reporting_start_datetime_utc,
380
                                                              reporting_end_datetime_utc,
381
                                                              period_type)
382
                for row_combined_equipment_periodically in rows_combined_equipment_periodically:
383
                    current_datetime_local = row_combined_equipment_periodically[0].replace(tzinfo=timezone.utc) + \
384
                                             timedelta(minutes=timezone_offset)
385
                    if period_type == 'hourly':
386
                        current_datetime = current_datetime_local.isoformat()[0:19]
387
                    elif period_type == 'daily':
388
                        current_datetime = current_datetime_local.isoformat()[0:10]
389
                    elif period_type == 'weekly':
390
                        current_datetime = current_datetime_local.isoformat()[0:10]
391
                    elif period_type == 'monthly':
392
                        current_datetime = current_datetime_local.isoformat()[0:7]
393
                    elif period_type == 'yearly':
394
                        current_datetime = current_datetime_local.isoformat()[0:4]
395
396
                    actual_value = Decimal(0.0) if row_combined_equipment_periodically[1] is None \
397
                        else row_combined_equipment_periodically[1]
398
                    reporting[energy_category_id]['timestamps'].append(current_datetime)
399
                    reporting[energy_category_id]['values'].append(actual_value)
400
                    reporting[energy_category_id]['subtotal'] += actual_value
401
402
        ################################################################################################################
403
        # Step 8: query tariff data
404
        ################################################################################################################
405
        parameters_data = dict()
406
        parameters_data['names'] = list()
407
        parameters_data['timestamps'] = list()
408
        parameters_data['values'] = list()
409
        if not is_quick_mode:
410
            if config.is_tariff_appended and energy_category_set is not None and len(energy_category_set) > 0:
411
                for energy_category_id in energy_category_set:
412
                    energy_category_tariff_dict = \
413
                        utilities.get_energy_category_tariffs(combined_equipment['cost_center_id'],
414
                                                              energy_category_id,
415
                                                              reporting_start_datetime_utc,
416
                                                              reporting_end_datetime_utc)
417
                    tariff_timestamp_list = list()
418
                    tariff_value_list = list()
419
                    for k, v in energy_category_tariff_dict.items():
420
                        # convert k from utc to local
421
                        k = k + timedelta(minutes=timezone_offset)
422
                        tariff_timestamp_list.append(k.isoformat()[0:19])
423
                        tariff_value_list.append(v)
424
425
                    parameters_data['names'].append(
426
                        _('Tariff') + '-' + energy_category_dict[energy_category_id]['name'])
427
                    parameters_data['timestamps'].append(tariff_timestamp_list)
428
                    parameters_data['values'].append(tariff_value_list)
429
430
        ################################################################################################################
431
        # Step 9: query associated points data
432
        ################################################################################################################
433
        if not is_quick_mode:
434
            for point in point_list:
435
                point_values = []
436
                point_timestamps = []
437
                if point['object_type'] == 'ENERGY_VALUE':
438
                    query = (" SELECT utc_date_time, actual_value "
439
                             " FROM tbl_energy_value "
440
                             " WHERE point_id = %s "
441
                             "       AND utc_date_time BETWEEN %s AND %s "
442
                             " ORDER BY utc_date_time ")
443
                    cursor_historical.execute(query, (point['id'],
444
                                                      reporting_start_datetime_utc,
445
                                                      reporting_end_datetime_utc))
446
                    rows = cursor_historical.fetchall()
447
448
                    if rows is not None and len(rows) > 0:
449
                        for row in rows:
450
                            current_datetime_local = row[0].replace(tzinfo=timezone.utc) + \
451
                                                     timedelta(minutes=timezone_offset)
452
                            current_datetime = current_datetime_local.isoformat()[0:19]
453
                            point_timestamps.append(current_datetime)
454
                            point_values.append(row[1])
455
                elif point['object_type'] == 'ANALOG_VALUE':
456
                    query = (" SELECT utc_date_time, actual_value "
457
                             " FROM tbl_analog_value "
458
                             " WHERE point_id = %s "
459
                             "       AND utc_date_time BETWEEN %s AND %s "
460
                             " ORDER BY utc_date_time ")
461
                    cursor_historical.execute(query, (point['id'],
462
                                                      reporting_start_datetime_utc,
463
                                                      reporting_end_datetime_utc))
464
                    rows = cursor_historical.fetchall()
465
466
                    if rows is not None and len(rows) > 0:
467
                        for row in rows:
468
                            current_datetime_local = row[0].replace(tzinfo=timezone.utc) + \
469
                                                     timedelta(minutes=timezone_offset)
470
                            current_datetime = current_datetime_local.isoformat()[0:19]
471
                            point_timestamps.append(current_datetime)
472
                            point_values.append(row[1])
473
                elif point['object_type'] == 'DIGITAL_VALUE':
474
                    query = (" SELECT utc_date_time, actual_value "
475
                             " FROM tbl_digital_value "
476
                             " WHERE point_id = %s "
477
                             "       AND utc_date_time BETWEEN %s AND %s "
478
                             " ORDER BY utc_date_time ")
479
                    cursor_historical.execute(query, (point['id'],
480
                                                      reporting_start_datetime_utc,
481
                                                      reporting_end_datetime_utc))
482
                    rows = cursor_historical.fetchall()
483
484
                    if rows is not None and len(rows) > 0:
485
                        for row in rows:
486
                            current_datetime_local = row[0].replace(tzinfo=timezone.utc) + \
487
                                                     timedelta(minutes=timezone_offset)
488
                            current_datetime = current_datetime_local.isoformat()[0:19]
489
                            point_timestamps.append(current_datetime)
490
                            point_values.append(row[1])
491
492
                parameters_data['names'].append(point['name'] + ' (' + point['units'] + ')')
493
                parameters_data['timestamps'].append(point_timestamps)
494
                parameters_data['values'].append(point_values)
495
496
        ################################################################################################################
497
        # Step 10: query associated equipments energy income
498
        ################################################################################################################
499
        associated_equipment_data = dict()
500
501
        if energy_category_set is not None and len(energy_category_set) > 0:
502
            for energy_category_id in energy_category_set:
503
                associated_equipment_data[energy_category_id] = dict()
504
                associated_equipment_data[energy_category_id]['associated_equipment_names'] = list()
505
                associated_equipment_data[energy_category_id]['subtotals'] = list()
506
                for associated_equipment in associated_equipment_list:
507
                    associated_equipment_data[energy_category_id]['associated_equipment_names'].append(
508
                        associated_equipment['name'])
509
510
                    cursor_billing.execute(" SELECT SUM(actual_value) "
511
                                           " FROM tbl_equipment_output_category_hourly "
512
                                           " WHERE equipment_id = %s "
513
                                           "     AND energy_category_id = %s "
514
                                           "     AND start_datetime_utc >= %s "
515
                                           "     AND start_datetime_utc < %s ",
516
                                           (associated_equipment['id'],
517
                                            energy_category_id,
518
                                            reporting_start_datetime_utc,
519
                                            reporting_end_datetime_utc))
520
                    row_subtotal = cursor_billing.fetchone()
521
522
                    subtotal = Decimal(0.0) if (row_subtotal is None or row_subtotal[0] is None) else row_subtotal[0]
523
                    associated_equipment_data[energy_category_id]['subtotals'].append(subtotal)
524
525
        ################################################################################################################
526
        # Step 11: construct the report
527
        ################################################################################################################
528
        if cursor_system:
529
            cursor_system.close()
530
        if cnx_system:
531
            cnx_system.close()
532
533
        if cursor_billing:
534
            cursor_billing.close()
535
        if cnx_billing:
536
            cnx_billing.close()
537
538
        if cursor_historical:
539
            cursor_historical.close()
540
        if cnx_historical:
541
            cnx_historical.close()
542
543
        result = dict()
544
545
        result['combined_equipment'] = dict()
546
        result['combined_equipment']['name'] = combined_equipment['name']
547
548
        result['base_period'] = dict()
549
        result['base_period']['names'] = list()
550
        result['base_period']['units'] = list()
551
        result['base_period']['timestamps'] = list()
552
        result['base_period']['values'] = list()
553
        result['base_period']['subtotals'] = list()
554
        result['base_period']['total'] = Decimal(0.0)
555
        if energy_category_set is not None and len(energy_category_set) > 0:
556
            for energy_category_id in energy_category_set:
557
                result['base_period']['names'].append(energy_category_dict[energy_category_id]['name'])
558
                result['base_period']['units'].append(config.currency_unit)
559
                result['base_period']['timestamps'].append(base[energy_category_id]['timestamps'])
560
                result['base_period']['values'].append(base[energy_category_id]['values'])
561
                result['base_period']['subtotals'].append(base[energy_category_id]['subtotal'])
562
                result['base_period']['total'] += base[energy_category_id]['subtotal']
563
564
        result['reporting_period'] = dict()
565
        result['reporting_period']['names'] = list()
566
        result['reporting_period']['energy_category_ids'] = list()
567
        result['reporting_period']['units'] = list()
568
        result['reporting_period']['timestamps'] = list()
569
        result['reporting_period']['values'] = list()
570
        result['reporting_period']['rates'] = list()
571
        result['reporting_period']['subtotals'] = list()
572
        result['reporting_period']['subtotals_per_unit_area'] = list()
573
        result['reporting_period']['increment_rates'] = list()
574
        result['reporting_period']['total'] = Decimal(0.0)
575
        result['reporting_period']['total_per_unit_area'] = Decimal(0.0)
576
        result['reporting_period']['total_increment_rate'] = Decimal(0.0)
577
        result['reporting_period']['total_unit'] = config.currency_unit
578
579 View Code Duplication
        if energy_category_set is not None and len(energy_category_set) > 0:
0 ignored issues
show
This code seems to be duplicated in your project.
Loading history...
580
            for energy_category_id in energy_category_set:
581
                result['reporting_period']['names'].append(energy_category_dict[energy_category_id]['name'])
582
                result['reporting_period']['energy_category_ids'].append(energy_category_id)
583
                result['reporting_period']['units'].append(config.currency_unit)
584
                result['reporting_period']['timestamps'].append(reporting[energy_category_id]['timestamps'])
585
                result['reporting_period']['values'].append(reporting[energy_category_id]['values'])
586
                result['reporting_period']['subtotals'].append(reporting[energy_category_id]['subtotal'])
587
                result['reporting_period']['increment_rates'].append(
588
                    (reporting[energy_category_id]['subtotal'] - base[energy_category_id]['subtotal']) /
589
                    base[energy_category_id]['subtotal']
590
                    if base[energy_category_id]['subtotal'] > 0.0 else None)
591
                result['reporting_period']['total'] += reporting[energy_category_id]['subtotal']
592
593
                rate = list()
594
                for index, value in enumerate(reporting[energy_category_id]['values']):
595
                    if index < len(base[energy_category_id]['values']) \
596
                            and base[energy_category_id]['values'][index] != 0 and value != 0:
597
                        rate.append((value - base[energy_category_id]['values'][index])
598
                                    / base[energy_category_id]['values'][index])
599
                    else:
600
                        rate.append(None)
601
                result['reporting_period']['rates'].append(rate)
602
603
        result['reporting_period']['total_increment_rate'] = \
604
            (result['reporting_period']['total'] - result['base_period']['total']) / \
605
            result['base_period']['total'] \
606
            if result['base_period']['total'] > Decimal(0.0) else None
607
608
        result['parameters'] = {
609
            "names": parameters_data['names'],
610
            "timestamps": parameters_data['timestamps'],
611
            "values": parameters_data['values']
612
        }
613
614
        result['associated_equipment'] = dict()
615
        result['associated_equipment']['energy_category_names'] = list()
616
        result['associated_equipment']['units'] = list()
617
        result['associated_equipment']['associated_equipment_names_array'] = list()
618
        result['associated_equipment']['subtotals_array'] = list()
619
        result['associated_equipment']['total_unit'] = config.currency_unit
620
        if energy_category_set is not None and len(energy_category_set) > 0:
621
            for energy_category_id in energy_category_set:
622
                result['associated_equipment']['energy_category_names'].append(
623
                    energy_category_dict[energy_category_id]['name'])
624
                result['associated_equipment']['units'].append(config.currency_unit)
625
                result['associated_equipment']['associated_equipment_names_array'].append(
626
                    associated_equipment_data[energy_category_id]['associated_equipment_names'])
627
                result['associated_equipment']['subtotals_array'].append(
628
                    associated_equipment_data[energy_category_id]['subtotals'])
629
630
        # export result to Excel file and then encode the file to base64 string
631
        result['excel_bytes_base64'] = None
632
        if not is_quick_mode:
633
            result['excel_bytes_base64'] = \
634
                excelexporters.combinedequipmentincome.export(result,
635
                                                              combined_equipment['name'],
636
                                                              base_period_start_datetime_local,
637
                                                              base_period_end_datetime_local,
638
                                                              reporting_period_start_datetime_local,
639
                                                              reporting_period_end_datetime_local,
640
                                                              period_type,
641
                                                              language)
642
        resp.text = json.dumps(result)
643