Passed
Push — master ( 88a5d8...7de4c1 )
by Guangyu
01:49 queued 11s
created

offlinemetercost   C

Complexity

Total Complexity 54

Size/Duplication

Total Lines 392
Duplicated Lines 97.45 %

Importance

Changes 0
Metric Value
wmc 54
eloc 280
dl 382
loc 392
rs 6.4799
c 0
b 0
f 0

3 Methods

Rating   Name   Duplication   Size   Complexity  
A Reporting.__init__() 3 3 1
A Reporting.on_options() 3 3 1
F Reporting.on_get() 362 362 52

How to fix   Duplicated Code    Complexity   

Duplicated Code

Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.

Common duplication problems, and corresponding solutions are:

Complexity

 Tip:   Before tackling complexity, make sure that you eliminate any duplication first. This often can reduce the size of classes significantly.

Complex classes like offlinemetercost often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.

Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.

1
import falcon
2
import simplejson as json
3
import mysql.connector
4
import config
5
from datetime import datetime, timedelta, timezone
6
import utilities
7
from decimal import *
8
9
10 View Code Duplication
class Reporting:
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
11
    @staticmethod
12
    def __init__():
13
        pass
14
15
    @staticmethod
16
    def on_options(req, resp):
17
        resp.status = falcon.HTTP_200
18
19
    ####################################################################################################################
20
    # PROCEDURES
21
    # Step 1: valid parameters
22
    # Step 2: query the offline meter and energy category
23
    # Step 3: query base period energy consumption
24
    # Step 4: query base period energy cost
25
    # Step 5: query reporting period energy consumption
26
    # Step 6: query reporting period energy cost
27
    # Step 7: query tariff data
28
    # Step 8: construct the report
29
    ####################################################################################################################
30
    @staticmethod
31
    def on_get(req, resp):
32
        print(req.params)
33
        offline_meter_id = req.params.get('offlinemeterid')
34
        period_type = req.params.get('periodtype')
35
        base_period_start_datetime = req.params.get('baseperiodstartdatetime')
36
        base_period_end_datetime = req.params.get('baseperiodenddatetime')
37
        reporting_period_start_datetime = req.params.get('reportingperiodstartdatetime')
38
        reporting_period_end_datetime = req.params.get('reportingperiodenddatetime')
39
40
        ################################################################################################################
41
        # Step 1: valid parameters
42
        ################################################################################################################
43
        if offline_meter_id is None:
44
            raise falcon.HTTPError(falcon.HTTP_400,
45
                                   title='API.BAD_REQUEST',
46
                                   description='API.INVALID_OFFLINE_METER_ID')
47
        else:
48
            offline_meter_id = str.strip(offline_meter_id)
49
            if not offline_meter_id.isdigit() or int(offline_meter_id) <= 0:
50
                raise falcon.HTTPError(falcon.HTTP_400,
51
                                       title='API.BAD_REQUEST',
52
                                       description='API.INVALID_OFFLINE_METER_ID')
53
54
        if period_type is None:
55
            raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_PERIOD_TYPE')
56
        else:
57
            period_type = str.strip(period_type)
58
            if period_type not in ['hourly', 'daily', 'monthly', 'yearly']:
59
                raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', description='API.INVALID_PERIOD_TYPE')
60
61
        timezone_offset = int(config.utc_offset[1:3]) * 60 + int(config.utc_offset[4:6])
62
        if config.utc_offset[0] == '-':
63
            timezone_offset = -timezone_offset
64
65
        base_start_datetime_utc = None
66
        if base_period_end_datetime is not None and len(str.strip(base_period_end_datetime)) > 0:
67
            base_period_start_datetime = str.strip(base_period_start_datetime)
68
            try:
69
                base_start_datetime_utc = datetime.strptime(base_period_start_datetime, '%Y-%m-%dT%H:%M:%S')
70
            except ValueError:
71
                raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
72
                                       description="API.INVALID_BASE_PERIOD_BEGINS_DATETIME")
73
            base_start_datetime_utc = base_start_datetime_utc.replace(tzinfo=timezone.utc) - \
74
                timedelta(minutes=timezone_offset)
75
76
        base_end_datetime_utc = None
77
        if base_period_end_datetime is not None and len(str.strip(base_period_end_datetime)) > 0:
78
            base_period_end_datetime = str.strip(base_period_end_datetime)
79
            try:
80
                base_end_datetime_utc = datetime.strptime(base_period_end_datetime, '%Y-%m-%dT%H:%M:%S')
81
            except ValueError:
82
                raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
83
                                       description="API.INVALID_BASE_PERIOD_ENDS_DATETIME")
84
            base_end_datetime_utc = base_end_datetime_utc.replace(tzinfo=timezone.utc) - \
85
                timedelta(minutes=timezone_offset)
86
87
        if base_start_datetime_utc is not None and base_end_datetime_utc is not None and \
88
                base_start_datetime_utc >= base_end_datetime_utc:
89
            raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
90
                                   description='API.INVALID_BASE_PERIOD_ENDS_DATETIME')
91
92
        if reporting_period_start_datetime is None:
93
            raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
94
                                   description="API.INVALID_REPORTING_PERIOD_BEGINS_DATETIME")
95
        else:
96
            reporting_period_start_datetime = str.strip(reporting_period_start_datetime)
97
            try:
98
                reporting_start_datetime_utc = datetime.strptime(reporting_period_start_datetime, '%Y-%m-%dT%H:%M:%S')
99
            except ValueError:
100
                raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
101
                                       description="API.INVALID_REPORTING_PERIOD_BEGINS_DATETIME")
102
            reporting_start_datetime_utc = reporting_start_datetime_utc.replace(tzinfo=timezone.utc) - \
103
                timedelta(minutes=timezone_offset)
104
105
        if reporting_period_end_datetime is None:
106
            raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
107
                                   description="API.INVALID_REPORTING_PERIOD_ENDS_DATETIME")
108
        else:
109
            reporting_period_end_datetime = str.strip(reporting_period_end_datetime)
110
            try:
111
                reporting_end_datetime_utc = datetime.strptime(reporting_period_end_datetime, '%Y-%m-%dT%H:%M:%S')
112
            except ValueError:
113
                raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
114
                                       description="API.INVALID_REPORTING_PERIOD_ENDS_DATETIME")
115
            reporting_end_datetime_utc = reporting_end_datetime_utc.replace(tzinfo=timezone.utc) - \
116
                timedelta(minutes=timezone_offset)
117
118
        if reporting_start_datetime_utc >= reporting_end_datetime_utc:
119
            raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
120
                                   description='API.INVALID_REPORTING_PERIOD_ENDS_DATETIME')
121
122
        ################################################################################################################
123
        # Step 2: query the offline meter and energy category
124
        ################################################################################################################
125
        cnx_system = mysql.connector.connect(**config.myems_system_db)
126
        cursor_system = cnx_system.cursor()
127
128
        cnx_energy = mysql.connector.connect(**config.myems_energy_db)
129
        cursor_energy = cnx_energy.cursor()
130
131
        cnx_billing = mysql.connector.connect(**config.myems_billing_db)
132
        cursor_billing = cnx_billing.cursor()
133
134
        cursor_system.execute(" SELECT m.id, m.name, m.cost_center_id, m.energy_category_id, "
135
                              "        ec.name, ec.unit_of_measure, ec.kgce, ec.kgco2e "
136
                              " FROM tbl_offline_meters m, tbl_energy_categories ec "
137
                              " WHERE m.id = %s AND m.energy_category_id = ec.id ", (offline_meter_id,))
138
        row_offline_meter = cursor_system.fetchone()
139
        if row_offline_meter is None:
140
            if cursor_system:
141
                cursor_system.close()
142
            if cnx_system:
143
                cnx_system.disconnect()
144
145
            if cursor_energy:
146
                cursor_energy.close()
147
            if cnx_energy:
148
                cnx_energy.disconnect()
149
150
            if cursor_billing:
151
                cursor_billing.close()
152
            if cnx_billing:
153
                cnx_billing.disconnect()
154
155
            raise falcon.HTTPError(falcon.HTTP_404, title='API.NOT_FOUND', description='API.OFFLINE_METER_NOT_FOUND')
156
157
        offline_meter = dict()
158
        offline_meter['id'] = row_offline_meter[0]
159
        offline_meter['name'] = row_offline_meter[1]
160
        offline_meter['cost_center_id'] = row_offline_meter[2]
161
        offline_meter['energy_category_id'] = row_offline_meter[3]
162
        offline_meter['energy_category_name'] = row_offline_meter[4]
163
        offline_meter['unit_of_measure'] = config.currency_unit
164
        offline_meter['kgce'] = row_offline_meter[6]
165
        offline_meter['kgco2e'] = row_offline_meter[7]
166
167
        ################################################################################################################
168
        # Step 3: query base period energy consumption
169
        ################################################################################################################
170
        query = (" SELECT start_datetime_utc, actual_value "
171
                 " FROM tbl_offline_meter_hourly "
172
                 " WHERE offline_meter_id = %s "
173
                 " AND start_datetime_utc >= %s "
174
                 " AND start_datetime_utc < %s "
175
                 " ORDER BY start_datetime_utc ")
176
        cursor_energy.execute(query, (offline_meter['id'], base_start_datetime_utc, base_end_datetime_utc))
177
        rows_offline_meter_hourly = cursor_energy.fetchall()
178
179
        rows_offline_meter_periodically = utilities.aggregate_hourly_data_by_period(rows_offline_meter_hourly,
180
                                                                                    base_start_datetime_utc,
181
                                                                                    base_end_datetime_utc,
182
                                                                                    period_type)
183
        base = dict()
184
        base['timestamps'] = list()
185
        base['values_in_category'] = list()
186
        base['total_in_category'] = Decimal(0.0)
187
        base['values_in_kgce'] = list()
188
        base['total_in_kgce'] = Decimal(0.0)
189
        base['values_in_kgco2e'] = list()
190
        base['total_in_kgco2e'] = Decimal(0.0)
191
192
        for row_offline_meter_periodically in rows_offline_meter_periodically:
193
            current_datetime_local = row_offline_meter_periodically[0].replace(tzinfo=timezone.utc) + \
194
                                     timedelta(minutes=timezone_offset)
195
            if period_type == 'hourly':
196
                current_datetime = current_datetime_local.strftime('%Y-%m-%dT%H:%M:%S')
197
            elif period_type == 'daily':
198
                current_datetime = current_datetime_local.strftime('%Y-%m-%d')
199
            elif period_type == 'monthly':
200
                current_datetime = current_datetime_local.strftime('%Y-%m')
201
            elif period_type == 'yearly':
202
                current_datetime = current_datetime_local.strftime('%Y')
203
204
            actual_value = Decimal(0.0) if row_offline_meter_periodically[1] is None \
205
                else row_offline_meter_periodically[1]
206
            base['timestamps'].append(current_datetime)
0 ignored issues
show
introduced by
The variable current_datetime does not seem to be defined for all execution paths.
Loading history...
207
            base['values_in_kgce'].append(actual_value * offline_meter['kgce'])
208
            base['total_in_kgce'] += actual_value * offline_meter['kgce']
209
            base['values_in_kgco2e'].append(actual_value * offline_meter['kgco2e'])
210
            base['total_in_kgco2e'] += actual_value * offline_meter['kgco2e']
211
212
        ################################################################################################################
213
        # Step 4: query base period energy cost
214
        ################################################################################################################
215
        query = (" SELECT start_datetime_utc, actual_value "
216
                 " FROM tbl_offline_meter_hourly "
217
                 " WHERE offline_meter_id = %s "
218
                 " AND start_datetime_utc >= %s "
219
                 " AND start_datetime_utc < %s "
220
                 " ORDER BY start_datetime_utc ")
221
        cursor_billing.execute(query, (offline_meter['id'], base_start_datetime_utc, base_end_datetime_utc))
222
        rows_offline_meter_hourly = cursor_billing.fetchall()
223
224
        rows_offline_meter_periodically = utilities.aggregate_hourly_data_by_period(rows_offline_meter_hourly,
225
                                                                                    base_start_datetime_utc,
226
                                                                                    base_end_datetime_utc,
227
                                                                                    period_type)
228
229
        base['values_in_category'] = list()
230
        base['total_in_category'] = Decimal(0.0)
231
232
        for row_offline_meter_periodically in rows_offline_meter_periodically:
233
            actual_value = Decimal(0.0) if row_offline_meter_periodically[1] is None \
234
                else row_offline_meter_periodically[1]
235
            base['values_in_category'].append(actual_value)
236
            base['total_in_category'] += actual_value
237
238
        ################################################################################################################
239
        # Step 5: query reporting period energy consumption
240
        ################################################################################################################
241
        query = (" SELECT start_datetime_utc, actual_value "
242
                 " FROM tbl_offline_meter_hourly "
243
                 " WHERE offline_meter_id = %s "
244
                 " AND start_datetime_utc >= %s "
245
                 " AND start_datetime_utc < %s "
246
                 " ORDER BY start_datetime_utc ")
247
        cursor_energy.execute(query, (offline_meter['id'], reporting_start_datetime_utc, reporting_end_datetime_utc))
248
        rows_offline_meter_hourly = cursor_energy.fetchall()
249
250
        rows_offline_meter_periodically = utilities.aggregate_hourly_data_by_period(rows_offline_meter_hourly,
251
                                                                                    reporting_start_datetime_utc,
252
                                                                                    reporting_end_datetime_utc,
253
                                                                                    period_type)
254
        reporting = dict()
255
        reporting['timestamps'] = list()
256
        reporting['values_in_category'] = list()
257
        reporting['total_in_category'] = Decimal(0.0)
258
        reporting['values_in_kgce'] = list()
259
        reporting['total_in_kgce'] = Decimal(0.0)
260
        reporting['values_in_kgco2e'] = list()
261
        reporting['total_in_kgco2e'] = Decimal(0.0)
262
263
        for row_offline_meter_periodically in rows_offline_meter_periodically:
264
            current_datetime_local = row_offline_meter_periodically[0].replace(tzinfo=timezone.utc) + \
265
                timedelta(minutes=timezone_offset)
266
            if period_type == 'hourly':
267
                current_datetime = current_datetime_local.strftime('%Y-%m-%dT%H:%M:%S')
268
            elif period_type == 'daily':
269
                current_datetime = current_datetime_local.strftime('%Y-%m-%d')
270
            elif period_type == 'monthly':
271
                current_datetime = current_datetime_local.strftime('%Y-%m')
272
            elif period_type == 'yearly':
273
                current_datetime = current_datetime_local.strftime('%Y')
274
275
            actual_value = Decimal(0.0) if row_offline_meter_periodically[1] is None \
276
                else row_offline_meter_periodically[1]
277
278
            reporting['timestamps'].append(current_datetime)
279
            reporting['values_in_kgce'].append(actual_value * offline_meter['kgce'])
280
            reporting['total_in_kgce'] += actual_value * offline_meter['kgce']
281
            reporting['values_in_kgco2e'].append(actual_value * offline_meter['kgco2e'])
282
            reporting['total_in_kgco2e'] += actual_value * offline_meter['kgco2e']
283
284
        ################################################################################################################
285
        # Step 6: query reporting period energy cost
286
        ################################################################################################################
287
        query = (" SELECT start_datetime_utc, actual_value "
288
                 " FROM tbl_offline_meter_hourly "
289
                 " WHERE offline_meter_id = %s "
290
                 " AND start_datetime_utc >= %s "
291
                 " AND start_datetime_utc < %s "
292
                 " ORDER BY start_datetime_utc ")
293
        cursor_billing.execute(query, (offline_meter['id'], reporting_start_datetime_utc, reporting_end_datetime_utc))
294
        rows_offline_meter_hourly = cursor_billing.fetchall()
295
296
        rows_offline_meter_periodically = utilities.aggregate_hourly_data_by_period(rows_offline_meter_hourly,
297
                                                                                    reporting_start_datetime_utc,
298
                                                                                    reporting_end_datetime_utc,
299
                                                                                    period_type)
300
301
        for row_offline_meter_periodically in rows_offline_meter_periodically:
302
            actual_value = Decimal(0.0) if row_offline_meter_periodically[1] is None \
303
                else row_offline_meter_periodically[1]
304
305
            reporting['values_in_category'].append(actual_value)
306
            reporting['total_in_category'] += actual_value
307
308
        ################################################################################################################
309
        # Step 7: query tariff data
310
        ################################################################################################################
311
        parameters_data = dict()
312
        parameters_data['names'] = list()
313
        parameters_data['timestamps'] = list()
314
        parameters_data['values'] = list()
315
316
        tariff_dict = utilities.get_energy_category_tariffs(offline_meter['cost_center_id'],
317
                                                            offline_meter['energy_category_id'],
318
                                                            reporting_start_datetime_utc,
319
                                                            reporting_end_datetime_utc)
320
        tariff_timestamp_list = list()
321
        tariff_value_list = list()
322
        for k, v in tariff_dict.items():
323
            # convert k from utc to local
324
            k = k + timedelta(minutes=timezone_offset)
325
            tariff_timestamp_list.append(k.isoformat()[0:19])
326
            tariff_value_list.append(v)
327
328
        parameters_data['names'].append('TARIFF-' + offline_meter['energy_category_name'])
329
        parameters_data['timestamps'].append(tariff_timestamp_list)
330
        parameters_data['values'].append(tariff_value_list)
331
332
        ################################################################################################################
333
        # Step 8: construct the report
334
        ################################################################################################################
335
        if cursor_system:
336
            cursor_system.close()
337
        if cnx_system:
338
            cnx_system.disconnect()
339
340
        if cursor_energy:
341
            cursor_energy.close()
342
        if cnx_energy:
343
            cnx_energy.disconnect()
344
345
        if cursor_billing:
346
            cursor_billing.close()
347
        if cnx_billing:
348
            cnx_billing.disconnect()
349
350
        result = {
351
            "offline_meter": {
352
                "cost_center_id": offline_meter['cost_center_id'],
353
                "energy_category_id": offline_meter['energy_category_id'],
354
                "energy_category_name": offline_meter['energy_category_name'],
355
                "unit_of_measure": offline_meter['unit_of_measure'],
356
                "kgce": offline_meter['kgce'],
357
                "kgco2e": offline_meter['kgco2e'],
358
            },
359
            "reporting_period": {
360
                "increment_rate":
361
                    (reporting['total_in_category']-base['total_in_category'])/base['total_in_category']
362
                    if base['total_in_category'] > 0 else None,
363
                "total_in_category": reporting['total_in_category'],
364
                "total_in_kgce": reporting['total_in_kgce'],
365
                "total_in_kgco2e": reporting['total_in_kgco2e'],
366
                "timestamps": [reporting['timestamps'],
367
                               reporting['timestamps'],
368
                               reporting['timestamps']],
369
                "values": [reporting['values_in_category'],
370
                           reporting['values_in_kgce'],
371
                           reporting['values_in_kgco2e']],
372
            },
373
            "base_period": {
374
                "total_in_category": base['total_in_category'],
375
                "total_in_kgce": base['total_in_kgce'],
376
                "total_in_kgco2e": base['total_in_kgco2e'],
377
                "timestamps": [base['timestamps'],
378
                               base['timestamps'],
379
                               base['timestamps']],
380
                "values": [base['values_in_category'],
381
                           base['values_in_kgce'],
382
                           base['values_in_kgco2e']],
383
            },
384
            "parameters": {
385
                "names": parameters_data['names'],
386
                "timestamps": parameters_data['timestamps'],
387
                "values": parameters_data['values']
388
            },
389
        }
390
391
        resp.body = json.dumps(result)
392