excelexporters.tenantload.generate_excel()   F
last analyzed

Complexity

Conditions 44

Size

Total Lines 428
Code Lines 336

Duplication

Lines 428
Ratio 100 %

Importance

Changes 0
Metric Value
cc 44
eloc 336
nop 5
dl 428
loc 428
rs 0
c 0
b 0
f 0

How to fix   Long Method    Complexity   

Long Method

Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.

For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.

Commonly applied refactorings include:

Complexity

Complex classes like excelexporters.tenantload.generate_excel() 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 base64
2
import uuid
3
import os
4
from openpyxl.chart import (
5
    PieChart,
6
    LineChart,
7
    BarChart,
8
    Reference,
9
)
10
from openpyxl.styles import PatternFill, Border, Side, Alignment, Font
11
from openpyxl.drawing.image import Image
12
from openpyxl import Workbook
13
from openpyxl.chart.label import DataLabelList
14
15
16
####################################################################################################################
17
# PROCEDURES
18
# Step 1: Validate the report data
19
# Step 2: Generate excel file
20
# Step 3: Encode the excel file bytes to Base64
21
####################################################################################################################
22
23
24
def export(report,
25
           name,
26
           reporting_start_datetime_local,
27
           reporting_end_datetime_local,
28
           period_type):
29
    ####################################################################################################################
30
    # Step 1: Validate the report data
31
    ####################################################################################################################
32
    if report is None:
33
        return None
34
    print(report)
35
36
    ####################################################################################################################
37
    # Step 2: Generate excel file from the report data
38
    ####################################################################################################################
39
    filename = generate_excel(report,
40
                              name,
41
                              reporting_start_datetime_local,
42
                              reporting_end_datetime_local,
43
                              period_type)
44
    ####################################################################################################################
45
    # Step 3: Encode the excel file to Base64
46
    ####################################################################################################################
47
    try:
48
        with open(filename, 'rb') as binary_file:
49
            binary_file_data = binary_file.read()
50
    except IOError as ex:
51
        pass
52
53
    # Base64 encode the bytes
54
    base64_encoded_data = base64.b64encode(binary_file_data)
0 ignored issues
show
introduced by
The variable binary_file_data does not seem to be defined for all execution paths.
Loading history...
55
    # get the Base64 encoded data using human-readable characters.
56
    base64_message = base64_encoded_data.decode('utf-8')
57
    # delete the file from server
58
    try:
59
        os.remove(filename)
60
    except NotImplementedError as ex:
61
        pass
62
    return base64_message
63
64
65 View Code Duplication
def generate_excel(report, name, reporting_start_datetime_local, reporting_end_datetime_local, period_type):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
66
    wb = Workbook()
67
68
    # todo
69
    ws = wb.active
70
71
    # Row height
72
    ws.row_dimensions[1].height = 102
73
    for i in range(2, 2000 + 1):
74
        ws.row_dimensions[i].height = 42
75
76
    # Col width
77
    ws.column_dimensions['A'].width = 1.5
78
79
    ws.column_dimensions['B'].width = 25.0
80
81
    for i in range(ord('C'), ord('L')):
82
        ws.column_dimensions[chr(i)].width = 15.0
83
84
    # Font
85
    name_font = Font(name='Constantia', size=15, bold=True)
86
    title_font = Font(name='宋体', size=15, bold=True)
87
    data_font = Font(name='Franklin Gothic Book', size=11)
88
89
    table_fill = PatternFill(fill_type='solid', fgColor='1F497D')
90
    f_border = Border(left=Side(border_style='medium', color='00000000'),
91
                      right=Side(border_style='medium', color='00000000'),
92
                      bottom=Side(border_style='medium', color='00000000'),
93
                      top=Side(border_style='medium', color='00000000')
94
                      )
95
    b_border = Border(
96
        bottom=Side(border_style='medium', color='00000000'),
97
    )
98
99
    b_c_alignment = Alignment(vertical='bottom',
100
                              horizontal='center',
101
                              text_rotation=0,
102
                              wrap_text=True,
103
                              shrink_to_fit=False,
104
                              indent=0)
105
    c_c_alignment = Alignment(vertical='center',
106
                              horizontal='center',
107
                              text_rotation=0,
108
                              wrap_text=True,
109
                              shrink_to_fit=False,
110
                              indent=0)
111
    b_r_alignment = Alignment(vertical='bottom',
112
                              horizontal='right',
113
                              text_rotation=0,
114
                              wrap_text=True,
115
                              shrink_to_fit=False,
116
                              indent=0)
117
    c_r_alignment = Alignment(vertical='bottom',
118
                              horizontal='center',
119
                              text_rotation=0,
120
                              wrap_text=True,
121
                              shrink_to_fit=False,
122
                              indent=0)
123
124
    # Img
125
    img = Image("excelexporters/myems.png")
126
    img.width = img.width * 0.85
127
    img.height = img.height * 0.85
128
    # img = Image("myems.png")
129
    ws.add_image(img, 'B1')
130
131
    # Title
132
    ws.row_dimensions[3].height = 60
133
134
    ws['B3'].font = name_font
135
    ws['B3'].alignment = b_r_alignment
136
    ws['B3'] = 'Name:'
137
    ws['C3'].border = b_border
138
    ws['C3'].alignment = b_c_alignment
139
    ws['C3'].font = name_font
140
    ws['C3'] = name
141
142
    ws['D3'].font = name_font
143
    ws['D3'].alignment = b_r_alignment
144
    ws['D3'] = 'Period:'
145
    ws['E3'].border = b_border
146
    ws['E3'].alignment = b_c_alignment
147
    ws['E3'].font = name_font
148
    ws['E3'] = period_type
149
150
    ws['F3'].font = name_font
151
    ws['F3'].alignment = b_r_alignment
152
    ws['F3'] = 'Date:'
153
    ws['G3'].alignment = b_c_alignment
154
    ws['G3'].font = name_font
155
    ws['G3'] = reporting_start_datetime_local + "__" + reporting_end_datetime_local
156
    ws.merge_cells("G3:H3")
157
158
    if "reporting_period" not in report.keys() or \
159
            "timestamps" not in report['reporting_period'].keys() or len(report['reporting_period']['timestamps']) == 0:
160
        filename = str(uuid.uuid4()) + '.xlsx'
0 ignored issues
show
Comprehensibility Best Practice introduced by
The variable str does not seem to be defined.
Loading history...
161
        wb.save(filename)
162
163
        return filename
164
165
    ###############################
166
167
    has_names_data_flag = True
168
169
    if "names" not in report['reporting_period'].keys() or len(report['reporting_period']['names']) == 0:
170
        has_names_data_flag = False
171
172
    current_row_number = 6
173
    if has_names_data_flag:
174
        reporting_period_data = report['reporting_period']
175
        category = reporting_period_data['names']
176
        ca_len = len(category)
177
178
        ws['B' + str(current_row_number)].font = title_font
179
        ws['B' + str(current_row_number)] = name + '报告期平均负荷'
180
181
        current_row_number += 1
182
183
        ws.row_dimensions[current_row_number].height = 60
184
        ws['B' + str(current_row_number)].fill = table_fill
185
        ws['B' + str(current_row_number)].border = f_border
186
187
        for i in range(0, ca_len):
188
            col = chr(ord('C') + i)
189
            ws[col + str(current_row_number)].fill = table_fill
190
            ws[col + str(current_row_number)].font = name_font
191
            ws[col + str(current_row_number)].alignment = c_c_alignment
192
            ws[col + str(current_row_number)].border = f_border
193
            ws[col + str(current_row_number)] = reporting_period_data['names'][i] + \
194
                " (" + reporting_period_data['units'][i] + "/H)"
195
196
        current_row_number += 1
197
198
        ws['B' + str(current_row_number)].font = title_font
199
        ws['B' + str(current_row_number)].alignment = c_c_alignment
200
        ws['B' + str(current_row_number)].border = f_border
201
        ws['B' + str(current_row_number)] = '平均负荷'
202
203
        for i in range(0, ca_len):
204
            col = chr(ord('C') + i)
205
            ws[col + str(current_row_number)].font = name_font
206
            ws[col + str(current_row_number)].alignment = c_c_alignment
207
            ws[col + str(current_row_number)].border = f_border
208
            ws[col + str(current_row_number)] = round(reporting_period_data['averages'][i], 2)
209
210
        current_row_number += 1
211
212
        ws['B' + str(current_row_number)].font = title_font
213
        ws['B' + str(current_row_number)].alignment = c_c_alignment
214
        ws['B' + str(current_row_number)].border = f_border
215
        ws['B' + str(current_row_number)] = '单位面积值'
216
217
        for i in range(0, ca_len):
218
            col = chr(ord('C') + i)
219
            ws[col + str(current_row_number)].font = name_font
220
            ws[col + str(current_row_number)].alignment = c_c_alignment
221
            ws[col + str(current_row_number)].border = f_border
222
            ws[col + str(current_row_number)] = round(reporting_period_data['averages_per_unit_area'][i], 2)
223
224
        current_row_number += 1
225
226
        ws['B' + str(current_row_number)].font = title_font
227
        ws['B' + str(current_row_number)].alignment = c_c_alignment
228
        ws['B' + str(current_row_number)].border = f_border
229
        ws['B' + str(current_row_number)] = '环比'
230
231
        for i in range(0, ca_len):
232
            col = chr(ord('C') + i)
233
            ws[col + str(current_row_number)].font = name_font
234
            ws[col + str(current_row_number)].alignment = c_c_alignment
235
            ws[col + str(current_row_number)].border = f_border
236
            ws[col + str(current_row_number)] = str(
237
                round(reporting_period_data['averages_increment_rate'][i] * 100, 2)) + "%" \
238
                if reporting_period_data['averages_increment_rate'][i] is not None else "-"
239
240
        current_row_number += 2
241
242
        ws['B' + str(current_row_number)].font = title_font
243
        ws['B' + str(current_row_number)] = name + '报告期最大负荷'
244
245
        current_row_number += 1
246
247
        ws.row_dimensions[current_row_number].height = 60
248
        ws['B' + str(current_row_number)].fill = table_fill
249
        ws['B' + str(current_row_number)].border = f_border
250
        for i in range(0, ca_len):
251
            col = chr(ord('C') + i)
252
            ws[col + str(current_row_number)].fill = table_fill
253
            ws[col + str(current_row_number)].font = name_font
254
            ws[col + str(current_row_number)].alignment = c_c_alignment
255
            ws[col + str(current_row_number)].border = f_border
256
            ws[col + str(current_row_number)] = reporting_period_data['names'][i] + \
257
                " (" + reporting_period_data['units'][i] + "/H)"
258
259
        current_row_number += 1
260
261
        ws['B' + str(current_row_number)].font = title_font
262
        ws['B' + str(current_row_number)].alignment = c_c_alignment
263
        ws['B' + str(current_row_number)].border = f_border
264
        ws['B' + str(current_row_number)] = '最大负荷'
265
266
        for i in range(0, ca_len):
267
            col = chr(ord('C') + i)
268
            ws[col + str(current_row_number)].font = name_font
269
            ws[col + str(current_row_number)].alignment = c_c_alignment
270
            ws[col + str(current_row_number)].border = f_border
271
            ws[col + str(current_row_number)] = round(reporting_period_data['maximums'][i], 2)
272
273
        current_row_number += 1
274
275
        ws['B' + str(current_row_number)].font = title_font
276
        ws['B' + str(current_row_number)].alignment = c_c_alignment
277
        ws['B' + str(current_row_number)].border = f_border
278
        ws['B' + str(current_row_number)] = '单位面积值'
279
280
        for i in range(0, ca_len):
281
            col = chr(ord('C') + i)
282
            ws[col + str(current_row_number)].font = name_font
283
            ws[col + str(current_row_number)].alignment = c_c_alignment
284
            ws[col + str(current_row_number)].border = f_border
285
            ws[col + str(current_row_number)] = round(reporting_period_data['maximums_per_unit_area'][i], 2)
286
287
        current_row_number += 1
288
289
        ws['B' + str(current_row_number)].font = title_font
290
        ws['B' + str(current_row_number)].alignment = c_c_alignment
291
        ws['B' + str(current_row_number)].border = f_border
292
        ws['B' + str(current_row_number)] = '环比'
293
294
        for i in range(0, ca_len):
295
            col = chr(ord('C') + i)
296
            ws[col + str(current_row_number)].font = name_font
297
            ws[col + str(current_row_number)].alignment = c_c_alignment
298
            ws[col + str(current_row_number)].border = f_border
299
            ws[col + str(current_row_number)] = str(
300
                round(reporting_period_data['maximums_increment_rate'][i] * 100, 2)) + "%" \
301
                if reporting_period_data['maximums_increment_rate'][i] is not None else "-"
302
303
        current_row_number += 2
304
305
        ws['B' + str(current_row_number)].font = title_font
306
        ws['B' + str(current_row_number)] = name + '报告期负荷系数'
307
308
        current_row_number += 1
309
310
        ws.row_dimensions[current_row_number].height = 60
311
        ws['B' + str(current_row_number)].fill = table_fill
312
        ws['B' + str(current_row_number)].border = f_border
313
        for i in range(0, ca_len):
314
            col = chr(ord('C') + i)
315
            ws[col + str(current_row_number)].fill = table_fill
316
            ws[col + str(current_row_number)].font = name_font
317
            ws[col + str(current_row_number)].alignment = c_c_alignment
318
            ws[col + str(current_row_number)].border = f_border
319
            ws[col + str(current_row_number)] = reporting_period_data['names'][i]
320
321
        current_row_number += 1
322
323
        ws['B' + str(current_row_number)].font = title_font
324
        ws['B' + str(current_row_number)].alignment = c_c_alignment
325
        ws['B' + str(current_row_number)].border = f_border
326
        ws['B' + str(current_row_number)] = '负荷系数'
327
328
        for i in range(0, ca_len):
329
            col = chr(ord('C') + i)
330
            ws[col + str(current_row_number)].font = name_font
331
            ws[col + str(current_row_number)].alignment = c_c_alignment
332
            ws[col + str(current_row_number)].border = f_border
333
            ws[col + str(current_row_number)] = round(reporting_period_data['factors'][i], 2) \
334
                if reporting_period_data['factors'][i] is not None else '-'
335
336
        current_row_number += 1
337
338
        ws['B' + str(current_row_number)].font = title_font
339
        ws['B' + str(current_row_number)].alignment = c_c_alignment
340
        ws['B' + str(current_row_number)].border = f_border
341
        ws['B' + str(current_row_number)] = '环比'
342
343
        for i in range(0, ca_len):
344
            col = chr(ord('C') + i)
345
            ws[col + str(current_row_number)].font = name_font
346
            ws[col + str(current_row_number)].alignment = c_c_alignment
347
            ws[col + str(current_row_number)].border = f_border
348
            ws[col + str(current_row_number)] = str(
349
                round(reporting_period_data['factors_increment_rate'][i] * 100, 2)) + "%" \
350
                if reporting_period_data['factors_increment_rate'][i] is not None else "-"
351
352
        current_row_number += 2
353
354
    has_sub_averages_data_flag = True
355
    has_sub_maximums_data_flag = True
356
357
    if "sub_averages" not in report['reporting_period'].keys() or len(report['reporting_period']['sub_averages']) == 0:
358
        has_sub_averages_data_flag = False
359
360
    if "sub_averages" not in report['reporting_period'].keys() or len(report['reporting_period']['sub_averages']) == 0:
361
        has_sub_maximums_data_flag = False
362
363
    if has_sub_averages_data_flag or has_sub_maximums_data_flag:
364
        reporting_period_data = report['reporting_period']
365
        category = reporting_period_data['names']
366
        ca_len = len(category)
367
        times = reporting_period_data['timestamps']
368
        time = times[0]
369
370
        ws['B' + str(current_row_number)].font = title_font
371
        ws['B' + str(current_row_number)] = name + '详细数据'
372
373
        current_row_number += 1
374
        chart_start_number = current_row_number
375
376
        if has_sub_averages_data_flag:
377
            current_row_number = (current_row_number + ca_len * 6)
378
379
        if has_sub_maximums_data_flag:
380
            current_row_number = (current_row_number + ca_len * 6)
381
382
        table_start_number = current_row_number
383
384
        ws.row_dimensions[current_row_number].height = 60
385
        ws['B' + str(current_row_number)].fill = table_fill
386
        ws['B' + str(current_row_number)].font = title_font
387
        ws['B' + str(current_row_number)].alignment = c_c_alignment
388
        ws['B' + str(current_row_number)].border = f_border
389
        ws['B' + str(current_row_number)] = '日期时间'
390
391
        col = 'C'
392
393
        for i in range(0, ca_len):
394
            if has_sub_averages_data_flag:
395
                ws[col + str(current_row_number)].fill = table_fill
396
                ws[col + str(current_row_number)].font = title_font
397
                ws[col + str(current_row_number)].alignment = c_c_alignment
398
                ws[col + str(current_row_number)].border = f_border
399
                ws[col + str(current_row_number)] = reporting_period_data['names'][i] + \
400
                    " 平均负荷(" + reporting_period_data['units'][i] + "/H)"
401
                col = chr(ord(col) + 1)
402
403
            if has_sub_maximums_data_flag:
404
                ws[col + str(current_row_number)].fill = table_fill
405
                ws[col + str(current_row_number)].font = title_font
406
                ws[col + str(current_row_number)].alignment = c_c_alignment
407
                ws[col + str(current_row_number)].border = f_border
408
                ws[col + str(current_row_number)] = reporting_period_data['names'][i] + \
409
                    " 最大负荷(" + reporting_period_data['units'][i] + "/H)"
410
                col = chr(ord(col) + 1)
411
412
        current_row_number += 1
413
414
        for i in range(0, len(time)):
415
            ws['B' + str(current_row_number)].font = title_font
416
            ws['B' + str(current_row_number)].alignment = c_c_alignment
417
            ws['B' + str(current_row_number)].border = f_border
418
            ws['B' + str(current_row_number)] = time[i]
419
420
            col = 'C'
421
            for j in range(0, ca_len):
422
423
                if has_sub_averages_data_flag:
424
                    ws[col + str(current_row_number)].font = title_font
425
                    ws[col + str(current_row_number)].alignment = c_c_alignment
426
                    ws[col + str(current_row_number)].border = f_border
427
                    ws[col + str(current_row_number)] = round(reporting_period_data['sub_averages'][j][i], 2) \
428
                        if reporting_period_data['sub_averages'][j][i] is not None else 0.00
429
                    col = chr(ord(col) + 1)
430
431
                if has_sub_maximums_data_flag:
432
                    ws[col + str(current_row_number)].font = title_font
433
                    ws[col + str(current_row_number)].alignment = c_c_alignment
434
                    ws[col + str(current_row_number)].border = f_border
435
                    ws[col + str(current_row_number)] = round(reporting_period_data['sub_maximums'][j][i], 2) \
436
                        if reporting_period_data['sub_maximums'][j][i] is not None else 0.00
437
                    col = chr(ord(col) + 1)
438
439
            current_row_number += 1
440
441
        table_end_number = current_row_number - 1
442
443
        current_chart_col_number = 3
444
        current_chart_row_number = chart_start_number
445
446
        for i in range(0, ca_len):
447
            labels = Reference(ws, min_col=2, min_row=table_start_number + 1, max_row=table_end_number)
448
449
            if has_sub_averages_data_flag:
450
                line = LineChart()
451
                line.title = '报告期 平均负荷 - ' + ws.cell(column=current_chart_col_number, row=table_start_number).value
452
                datas = Reference(ws, min_col=current_chart_col_number, min_row=table_start_number,
453
                                  max_row=table_end_number)
454
                line.add_data(datas, titles_from_data=True)
455
                line.set_categories(labels)
456
                line_data = line.series[0]
457
                line_data.marker.symbol = "circle"
458
                line_data.smooth = True
459
                line.x_axis.crosses = 'min'
460
                line.height = 8.25
461
                line.width = 24
462
                line.dLbls = DataLabelList()
463
                line.dLbls.dLblPos = 't'
464
                line.dLbls.showVal = True
465
                ws.add_chart(line, "B" + str(current_chart_row_number))
466
                current_chart_row_number += 6
467
                current_chart_col_number += 1
468
469
            if has_sub_maximums_data_flag:
470
                line = LineChart()
471
                line.title = '报告期 最大负荷 - ' + ws.cell(column=current_chart_col_number, row=table_start_number).value
472
                datas = Reference(ws, min_col=current_chart_col_number, min_row=table_start_number,
473
                                  max_row=table_end_number)
474
                line.add_data(datas, titles_from_data=True)
475
                line.set_categories(labels)
476
                line_data = line.series[0]
477
                line_data.marker.symbol = "circle"
478
                line_data.smooth = True
479
                line.x_axis.crosses = 'min'
480
                line.height = 8.25
481
                line.width = 24
482
                line.dLbls = DataLabelList()
483
                line.dLbls.dLblPos = 't'
484
                line.dLbls.showVal = True
485
                ws.add_chart(line, "B" + str(current_chart_row_number))
486
                current_chart_row_number += 6
487
                current_chart_col_number += 1
488
489
    filename = str(uuid.uuid4()) + '.xlsx'
490
    wb.save(filename)
491
492
    return filename
493