Passed
Push — master ( c8b356...ca4867 )
by Guangyu
09:12 queued 13s
created

is_base_period_timestamp_exists()   A

Complexity

Conditions 4

Size

Total Lines 11
Code Lines 8

Duplication

Lines 11
Ratio 100 %

Importance

Changes 0
Metric Value
eloc 8
dl 11
loc 11
rs 10
c 0
b 0
f 0
cc 4
nop 1
1
import base64
2
import gettext
3
import os
4
import re
5
import uuid
6
7
import openpyxl.utils.cell as format_cell
8
from openpyxl import Workbook
9
from openpyxl.chart import LineChart, Reference
10
from openpyxl.chart.label import DataLabelList
11
from openpyxl.drawing.image import Image
12
from openpyxl.styles import PatternFill, Border, Side, Alignment, Font
13
14
15
####################################################################################################################
16
# PROCEDURES
17
# Step 1: Validate the report data
18
# Step 2: Generate excel file
19
# Step 3: Encode the excel file to Base64
20
####################################################################################################################
21
22
23 View Code Duplication
def export(report,
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
24
           name,
25
           base_period_start_datetime_local,
26
           base_period_end_datetime_local,
27
           reporting_start_datetime_local,
28
           reporting_end_datetime_local,
29
           period_type,
30
           language):
31
    ####################################################################################################################
32
    # Step 1: Validate the report data
33
    ####################################################################################################################
34
    if report is None:
35
        return None
36
37
    ####################################################################################################################
38
    # Step 2: Generate excel file from the report data
39
    ####################################################################################################################
40
    filename = generate_excel(report,
41
                              name,
42
                              base_period_start_datetime_local,
43
                              base_period_end_datetime_local,
44
                              reporting_start_datetime_local,
45
                              reporting_end_datetime_local,
46
                              period_type,
47
                              language)
48
    ####################################################################################################################
49
    # Step 3: Encode the excel file to Base64
50
    ####################################################################################################################
51
    binary_file_data = b''
52
    try:
53
        with open(filename, 'rb') as binary_file:
54
            binary_file_data = binary_file.read()
55
    except IOError as ex:
56
        print(str(ex))
57
58
    # Base64 encode the bytes
59
    base64_encoded_data = base64.b64encode(binary_file_data)
60
    # get the Base64 encoded data using human-readable characters.
61
    base64_message = base64_encoded_data.decode('utf-8')
62
    # delete the file from server
63
    try:
64
        os.remove(filename)
65
    except NotImplementedError as ex:
66
        print(str(ex))
67
    return base64_message
68
69
70
def generate_excel(report,
71
                   name,
72
                   base_period_start_datetime_local,
73
                   base_period_end_datetime_local,
74
                   reporting_start_datetime_local,
75
                   reporting_end_datetime_local,
76
                   period_type,
77
                   language):
78
    locale_path = './i18n/'
79
    if language == 'zh_CN':
80
        trans = gettext.translation('myems', locale_path, languages=['zh_CN'])
81
    elif language == 'de':
82
        trans = gettext.translation('myems', locale_path, languages=['de'])
83
    elif language == 'en':
84
        trans = gettext.translation('myems', locale_path, languages=['en'])
85
    else:
86
        trans = gettext.translation('myems', locale_path, languages=['en'])
87
    trans.install()
88
    _ = trans.gettext
89
    wb = Workbook()
90
    ws = wb.active
91
    ws.title = "EquipmentOutput"
92
93
    # Row height
94
    ws.row_dimensions[1].height = 102
95
    for i in range(2, 2000 + 1):
96
        ws.row_dimensions[i].height = 42
97
98
    # Col width
99
    ws.column_dimensions['A'].width = 1.5
100
101
    ws.column_dimensions['B'].width = 25.0
102
103
    for i in range(ord('C'), ord('Z')):
104
        ws.column_dimensions[chr(i)].width = 15.0
105
106
    # Font
107
    name_font = Font(name='Arial', size=15, bold=True)
108
    title_font = Font(name='Arial', size=15, bold=True)
109
110
    table_fill = PatternFill(fill_type='solid', fgColor='90ee90')
111
    f_border = Border(left=Side(border_style='medium'),
112
                      right=Side(border_style='medium'),
113
                      bottom=Side(border_style='medium'),
114
                      top=Side(border_style='medium')
115
                      )
116
    b_border = Border(bottom=Side(border_style='medium'), )
117
118
    b_c_alignment = Alignment(vertical='bottom',
119
                              horizontal='center',
120
                              text_rotation=0,
121
                              wrap_text=True,
122
                              shrink_to_fit=False,
123
                              indent=0)
124
    c_c_alignment = Alignment(vertical='center',
125
                              horizontal='center',
126
                              text_rotation=0,
127
                              wrap_text=True,
128
                              shrink_to_fit=False,
129
                              indent=0)
130
    b_r_alignment = Alignment(vertical='bottom',
131
                              horizontal='right',
132
                              text_rotation=0,
133
                              wrap_text=True,
134
                              shrink_to_fit=False,
135
                              indent=0)
136
    # Img
137
    img = Image("excelexporters/myems.png")
138
    ws.add_image(img, 'A1')
139
140
    # Title
141
    ws['B3'].alignment = b_r_alignment
142
    ws['B3'] = _('Name') + ':'
143
    ws['C3'].border = b_border
144
    ws['C3'].alignment = b_c_alignment
145
    ws['C3'] = name
146
147
    ws['D3'].alignment = b_r_alignment
148
    ws['D3'] = _('Period Type') + ':'
149
    ws['E3'].border = b_border
150
    ws['E3'].alignment = b_c_alignment
151
    ws['E3'] = period_type
152
153
    ws['B4'].alignment = b_r_alignment
154
    ws['B4'] = _('Reporting Start Datetime') + ':'
155
    ws['C4'].border = b_border
156
    ws['C4'].alignment = b_c_alignment
157
    ws['C4'] = reporting_start_datetime_local
158
159
    ws['D4'].alignment = b_r_alignment
160
    ws['D4'] = _('Reporting End Datetime') + ':'
161
    ws['E4'].border = b_border
162
    ws['E4'].alignment = b_c_alignment
163
    ws['E4'] = reporting_end_datetime_local
164
165
    is_base_period_timestamp_exists_flag = is_base_period_timestamp_exists(report['base_period'])
166
167
    if is_base_period_timestamp_exists_flag:
168
        ws['B5'].alignment = b_r_alignment
169
        ws['B5'] = _('Base Period Start Datetime') + ':'
170
        ws['C5'].border = b_border
171
        ws['C5'].alignment = b_c_alignment
172
        ws['C5'] = base_period_start_datetime_local
173
174
        ws['D5'].alignment = b_r_alignment
175
        ws['D5'] = _('Base Period End Datetime') + ':'
176
        ws['E5'].border = b_border
177
        ws['E5'].alignment = b_c_alignment
178
        ws['E5'] = base_period_end_datetime_local
179
180
    if "reporting_period" not in report.keys() or \
181
            "names" not in report['reporting_period'].keys() or len(report['reporting_period']['names']) == 0:
182
        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...
183
        wb.save(filename)
184
185
        return filename
186
187
    ####################################################################################################################
188
189
    current_row_number = 7
190
    reporting_period_data = report['reporting_period']
191 View Code Duplication
    if "names" not in reporting_period_data.keys() or \
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
192
            reporting_period_data['names'] is None or \
193
            len(reporting_period_data['names']) == 0:
194
        pass
195
    else:
196
        ws['B' + str(current_row_number)].font = title_font
197
        ws['B' + str(current_row_number)] = name + ' ' + _('Reporting Period Output')
198
199
        current_row_number += 1
200
201
        category = reporting_period_data['names']
202
        ca_len = len(category)
203
204
        ws.row_dimensions[current_row_number].height = 60
205
        ws['B' + str(current_row_number)].fill = table_fill
206
        ws['B' + str(current_row_number)].border = f_border
207
208
        col = 'C'
209
210
        for i in range(0, ca_len):
211
            ws[col + str(current_row_number)].fill = table_fill
212
            ws[col + str(current_row_number)].font = name_font
213
            ws[col + str(current_row_number)].alignment = c_c_alignment
214
            ws[col + str(current_row_number)].border = f_border
215
            ws[col + str(current_row_number)] = \
216
                reporting_period_data['names'][i] + " (" + reporting_period_data['units'][i] + ")"
217
218
            col = chr(ord(col) + 1)
219
220
        current_row_number += 1
221
222
        ws['B' + str(current_row_number)].font = title_font
223
        ws['B' + str(current_row_number)].alignment = c_c_alignment
224
        ws['B' + str(current_row_number)].border = f_border
225
        ws['B' + str(current_row_number)] = _('Output')
226
227
        col = 'C'
228
229
        for i in range(0, ca_len):
230
            ws[col + str(current_row_number)].font = name_font
231
            ws[col + str(current_row_number)].alignment = c_c_alignment
232
            ws[col + str(current_row_number)].border = f_border
233
            ws[col + str(current_row_number)] = round(reporting_period_data['subtotals'][i], 2)
234
235
            col = chr(ord(col) + 1)
236
237
        current_row_number += 1
238
239
        ws['B' + str(current_row_number)].font = title_font
240
        ws['B' + str(current_row_number)].alignment = c_c_alignment
241
        ws['B' + str(current_row_number)].border = f_border
242
        ws['B' + str(current_row_number)] = _('Increment Rate')
243
244
        col = 'C'
245
246
        for i in range(0, ca_len):
247
            ws[col + str(current_row_number)].font = name_font
248
            ws[col + str(current_row_number)].alignment = c_c_alignment
249
            ws[col + str(current_row_number)].border = f_border
250
            ws[col + str(current_row_number)] = str(
251
                round(reporting_period_data['increment_rates'][i] * 100, 2)) + '%' \
252
                if reporting_period_data['increment_rates'][i] is not None else '-'
253
254
            col = chr(ord(col) + 1)
255
256
        current_row_number += 2
257
258
    ####################################################################################################################
259
    table_start_draw_flag = current_row_number + 1
260
261 View Code Duplication
    if 'values' not in reporting_period_data.keys() or \
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
262
            reporting_period_data['values'] is None or \
263
            len(reporting_period_data['values']) == 0 or \
264
            'timestamps' not in reporting_period_data.keys() or \
265
            reporting_period_data['timestamps'] is None or \
266
            len(reporting_period_data['timestamps']) == 0 or \
267
            len(reporting_period_data['timestamps'][0]) == 0:
268
        pass
269
    else:
270
271
        if not is_base_period_timestamp_exists_flag:
272
            reporting_period_data = report['reporting_period']
273
            times = reporting_period_data['timestamps']
274
            ca_len = len(report['reporting_period']['names'])
275
            real_timestamps_len = timestamps_data_not_equal_0(report['parameters']['timestamps'])
276
            ws['B' + str(current_row_number)].font = title_font
277
            ws['B' + str(current_row_number)] = name + ' ' + _('Detailed Data')
278
279
            current_row_number += 1
280
            # 1: Stand for blank line  2: Stand for title
281
            current_row_number += ca_len * 6 + real_timestamps_len * 6 + 1 + 2
282
            table_start_row_number = current_row_number
283
284
            time = times[0]
285
            has_data = False
286
287
            if len(time) > 0:
288
                has_data = True
289
290
            if has_data:
291
292
                ws.row_dimensions[current_row_number].height = 60
293
                current_col_number = 2
294
                col = format_cell.get_column_letter(current_col_number)
295
                ws[col + str(current_row_number)].fill = table_fill
296
                ws[col + str(current_row_number)].font = title_font
297
                ws[col + str(current_row_number)].border = f_border
298
                ws[col + str(current_row_number)].alignment = c_c_alignment
299
                ws[col + str(current_row_number)] = _('Datetime')
300
301
                for i in range(0, ca_len):
302
                    current_col_number += 1
303
                    col = format_cell.get_column_letter(current_col_number)
304
305
                    ws[col + str(current_row_number)].fill = table_fill
306
                    ws[col + str(current_row_number)].font = title_font
307
                    ws[col + str(current_row_number)].alignment = c_c_alignment
308
                    ws[col + str(current_row_number)] = reporting_period_data['names'][i] + \
309
                        " (" + reporting_period_data['units'][i] + ")"
310
                    ws[col + str(current_row_number)].border = f_border
311
312
                current_row_number += 1
313
314
                for i in range(0, len(time)):
315
                    current_col_number = 2
316
                    col = format_cell.get_column_letter(current_col_number)
317
                    ws[col + str(current_row_number)].font = title_font
318
                    ws[col + str(current_row_number)].alignment = c_c_alignment
319
                    ws[col + str(current_row_number)] = time[i]
320
                    ws[col + str(current_row_number)].border = f_border
321
322
                    for j in range(0, ca_len):
323
                        current_col_number += 1
324
                        col = format_cell.get_column_letter(current_col_number)
325
326
                        ws[col + str(current_row_number)].font = title_font
327
                        ws[col + str(current_row_number)].alignment = c_c_alignment
328
                        ws[col + str(current_row_number)] = round(reporting_period_data['values'][j][i], 2)
329
                        ws[col + str(current_row_number)].border = f_border
330
331
                    current_row_number += 1
332
333
                table_end_row_number = current_row_number - 1
334
335
                current_col_number = 2
336
                col = format_cell.get_column_letter(current_col_number)
337
                ws[col + str(current_row_number)].font = title_font
338
                ws[col + str(current_row_number)].alignment = c_c_alignment
339
                ws[col + str(current_row_number)] = _('Subtotal')
340
                ws[col + str(current_row_number)].border = f_border
341
342
                for i in range(0, ca_len):
343
                    current_col_number += 1
344
                    col = format_cell.get_column_letter(current_col_number)
345
346
                    ws[col + str(current_row_number)].font = title_font
347
                    ws[col + str(current_row_number)].alignment = c_c_alignment
348
                    ws[col + str(current_row_number)] = round(reporting_period_data['subtotals'][i], 2)
349
                    ws[col + str(current_row_number)].border = f_border
350
351
                    # line
352
                    line = LineChart()
353
                    line.title = _('Reporting Period Output') + ' - ' \
354
                        + reporting_period_data['names'][i] + " (" + reporting_period_data['units'][i] + ")"
355
                    labels = Reference(ws, min_col=2, min_row=table_start_row_number + 1, max_row=table_end_row_number)
356
                    line_data = Reference(ws, min_col=3 + i, min_row=table_start_row_number,
357
                                          max_row=table_end_row_number)
358
                    line.add_data(line_data, titles_from_data=True)
359
                    line.set_categories(labels)
360
                    line_data = line.series[0]
361
                    line_data.marker.symbol = "circle"
362
                    line_data.smooth = True
363
                    line.x_axis.crosses = 'min'
364
                    line.height = 8.25
365
                    line.width = 24
366
                    line.dLbls = DataLabelList()
367
                    line.dLbls.dLblPos = 't'
368
                    line.dLbls.showVal = True
369
                    line.dLbls.showPercent = False
370
                    chart_col = 'B'
371
                    chart_cell = chart_col + str(table_start_draw_flag + 6 * i)
372
                    ws.add_chart(line, chart_cell)
373
374
                current_row_number += 2
375
        else:
376
            base_period_data = report['base_period']
377
            reporting_period_data = report['reporting_period']
378
            base_period_timestamps = base_period_data['timestamps']
379
            reporting_period_timestamps = reporting_period_data['timestamps']
380
            # Tip:
381
            #     base_period_data['names'] == reporting_period_data['names']
382
            #     base_period_data['units'] == reporting_period_data['units']
383
            base_period_data_ca_len = len(base_period_data['names'])
384
            reporting_period_data_ca_len = len(reporting_period_data['names'])
385
            real_timestamps_len = timestamps_data_not_equal_0(report['parameters']['timestamps'])
386
            ws['B' + str(current_row_number)].font = title_font
387
            ws['B' + str(current_row_number)] = name + ' ' + _('Detailed Data')
388
389
            current_row_number += 1
390
            # 1: Stand for blank line  2: Stand for title
391
            current_row_number += reporting_period_data_ca_len * 6 + real_timestamps_len * 6 + 1 + 2
392
            table_start_row_number = current_row_number
393
394
            has_data = False
395
396
            if len(base_period_timestamps[0]) or len(reporting_period_timestamps[0]) > 0:
397
                has_data = True
398
399
            if has_data:
400
                ws.row_dimensions[current_row_number].height = 60
401
                current_col_number = 2
402
                col = format_cell.get_column_letter(current_col_number)
403
                ws[col + str(current_row_number)].fill = table_fill
404
                ws[col + str(current_row_number)].font = title_font
405
                ws[col + str(current_row_number)].border = f_border
406
                ws[col + str(current_row_number)].alignment = c_c_alignment
407
                ws[col + str(current_row_number)] = _('Base Period') + " - " + _('Datetime')
408
409
                for i in range(0, base_period_data_ca_len):
410
                    current_col_number += 1
411
                    col = format_cell.get_column_letter(current_col_number)
412
413
                    ws[col + str(current_row_number)].fill = table_fill
414
                    ws[col + str(current_row_number)].font = title_font
415
                    ws[col + str(current_row_number)].alignment = c_c_alignment
416
                    ws[col + str(current_row_number)] = _('Base Period') + " - " + base_period_data['names'][i] + \
417
                        " (" + base_period_data['units'][i] + ")"
418
                    ws[col + str(current_row_number)].border = f_border
419
                current_col_number += 1
420
                col = format_cell.get_column_letter(current_col_number)
421
422
                ws[col + str(current_row_number)].fill = table_fill
423
                ws[col + str(current_row_number)].font = title_font
424
                ws[col + str(current_row_number)].border = f_border
425
                ws[col + str(current_row_number)].alignment = c_c_alignment
426
                ws[col + str(current_row_number)] = _('Reporting Period') + " - " + _('Datetime')
427
428
                for i in range(0, reporting_period_data_ca_len):
429
                    current_col_number += 1
430
                    col = format_cell.get_column_letter(current_col_number)
431
                    ws[col + str(current_row_number)].fill = table_fill
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)] = _('Reporting Period') + " - " \
435
                        + reporting_period_data['names'][i] + " (" + \
436
                        reporting_period_data['units'][i] + ")"
437
                    ws[col + str(current_row_number)].border = f_border
438
439
                current_row_number += 1
440
441
                max_timestamps_len = len(base_period_timestamps[0]) \
442
                    if len(base_period_timestamps[0]) >= len(reporting_period_timestamps[0]) \
443
                    else len(reporting_period_timestamps[0])
444
445
                for i in range(0, max_timestamps_len):
446
                    current_col_number = 2
447
                    col = format_cell.get_column_letter(current_col_number)
448
                    ws[col + str(current_row_number)].font = title_font
449
                    ws[col + str(current_row_number)].alignment = c_c_alignment
450
                    ws[col + str(current_row_number)] = base_period_timestamps[0][i] \
451
                        if i < len(base_period_timestamps[0]) else None
452
                    ws[col + str(current_row_number)].border = f_border
453
454
                    for j in range(0, base_period_data_ca_len):
455
                        current_col_number += 1
456
                        col = format_cell.get_column_letter(current_col_number)
457
458
                        ws[col + str(current_row_number)].font = title_font
459
                        ws[col + str(current_row_number)].alignment = c_c_alignment
460
                        ws[col + str(current_row_number)] = round(base_period_data['values'][j][i], 2) \
461
                            if i < len(base_period_data['values'][j]) else None
462
                        ws[col + str(current_row_number)].border = f_border
463
                    current_col_number += 1
464
                    col = format_cell.get_column_letter(current_col_number)
465
466
                    ws[col + str(current_row_number)].font = title_font
467
                    ws[col + str(current_row_number)].alignment = c_c_alignment
468
                    ws[col + str(current_row_number)] = reporting_period_timestamps[0][i] \
469
                        if i < len(reporting_period_timestamps[0]) else None
470
                    ws[col + str(current_row_number)].border = f_border
471
472
                    for j in range(0, reporting_period_data_ca_len):
473
                        current_col_number += 1
474
                        col = format_cell.get_column_letter(current_col_number)
475
476
                        ws[col + str(current_row_number)].font = title_font
477
                        ws[col + str(current_row_number)].alignment = c_c_alignment
478
                        ws[col + str(current_row_number)] = round(reporting_period_data['values'][j][i], 2) \
479
                            if i < len(reporting_period_data['values'][j]) else None
480
                        ws[col + str(current_row_number)].border = f_border
481
482
                    current_row_number += 1
483
484
                current_col_number = 2
485
                col = format_cell.get_column_letter(current_col_number)
486
                ws[col + str(current_row_number)].font = title_font
487
                ws[col + str(current_row_number)].alignment = c_c_alignment
488
                ws[col + str(current_row_number)] = _('Subtotal')
489
                ws[col + str(current_row_number)].border = f_border
490
491
                for i in range(0, base_period_data_ca_len):
492
                    current_col_number += 1
493
                    col = format_cell.get_column_letter(current_col_number)
494
                    ws[col + str(current_row_number)].font = title_font
495
                    ws[col + str(current_row_number)].alignment = c_c_alignment
496
                    ws[col + str(current_row_number)] = round(base_period_data['subtotals'][i], 2)
497
                    ws[col + str(current_row_number)].border = f_border
498
499
                current_col_number += 1
500
                col = format_cell.get_column_letter(current_col_number)
501
502
                ws[col + str(current_row_number)].font = title_font
503
                ws[col + str(current_row_number)].alignment = c_c_alignment
504
                ws[col + str(current_row_number)] = _('Subtotal')
505
                ws[col + str(current_row_number)].border = f_border
506
507
                for i in range(0, reporting_period_data_ca_len):
508
                    current_col_number += 1
509
                    col = format_cell.get_column_letter(current_col_number)
510
                    ws[col + str(current_row_number)].font = title_font
511
                    ws[col + str(current_row_number)].alignment = c_c_alignment
512
                    ws[col + str(current_row_number)] = round(reporting_period_data['subtotals'][i], 2)
513
                    ws[col + str(current_row_number)].border = f_border
514
515
                for i in range(0, reporting_period_data_ca_len):
516
                    # line
517
                    line = LineChart()
518
                    line.title = _('Base Period Output') + ' / ' \
519
                        + _('Reporting Period Output') + ' - ' \
520
                        + reporting_period_data['names'][i] + " (" + reporting_period_data['units'][i] + ")"
521
                    labels = Reference(ws, min_col=2 + base_period_data_ca_len + 1,
522
                                       min_row=table_start_row_number + 1,
523
                                       max_row=table_start_row_number + len(reporting_period_timestamps[0]))
524
                    base_line_data = Reference(ws, min_col=3 + i, min_row=table_start_row_number,
525
                                               max_row=table_start_row_number + len(reporting_period_timestamps[0]))
526
                    reporting_line_data = Reference(ws, min_col=3 + base_period_data_ca_len + 1 + i,
527
                                                    min_row=table_start_row_number,
528
                                                    max_row=table_start_row_number
529
                                                    + len(reporting_period_timestamps[0]))
530
                    line.add_data(base_line_data, titles_from_data=True)
531
                    line.add_data(reporting_line_data, titles_from_data=True)
532
                    line.set_categories(labels)
533
                    for j in range(len(line.series)):
534
                        line.series[j].marker.symbol = "circle"
535
                        line.series[j].smooth = True
536
                    line.x_axis.crosses = 'min'
537
                    line.height = 8.25
538
                    line.width = 24
539
                    line.dLbls = DataLabelList()
540
                    line.dLbls.dLblPos = 't'
541
                    line.dLbls.showVal = True
542
                    line.dLbls.showPercent = False
543
                    chart_col = 'B'
544
                    chart_cell = chart_col + str(table_start_draw_flag + 6 * i)
545
                    ws.add_chart(line, chart_cell)
546
547
                current_row_number += 2
548
549
    ####################################################################################################################
550
    current_sheet_parameters_row_number = table_start_draw_flag + len(reporting_period_data['names']) * 6 + 1
551
    if 'parameters' not in report.keys() or \
552
            report['parameters'] is None or \
553
            'names' not in report['parameters'].keys() or \
554
            report['parameters']['names'] is None or \
555
            len(report['parameters']['names']) == 0 or \
556
            'timestamps' not in report['parameters'].keys() or \
557
            report['parameters']['timestamps'] is None or \
558
            len(report['parameters']['timestamps']) == 0 or \
559
            'values' not in report['parameters'].keys() or \
560
            report['parameters']['values'] is None or \
561
            len(report['parameters']['values']) == 0 or \
562
            timestamps_data_all_equal_0(report['parameters']['timestamps']):
563
        pass
564
    else:
565
566
        ################################################################################################################
567
        # new worksheet
568
        ################################################################################################################
569
570
        parameters_data = report['parameters']
571
        parameters_names_len = len(parameters_data['names'])
572
573
        file_name = (re.sub(r'[^A-Z]', '', ws.title))+'_'
574
        parameters_ws = wb.create_sheet(file_name + _('Parameters'))
575
576
        parameters_timestamps_data_max_len = \
577
            get_parameters_timestamps_lists_max_len(list(parameters_data['timestamps']))
578
579
        # Row height
580
        parameters_ws.row_dimensions[1].height = 102
581
        for i in range(2, 7 + 1):
582
            parameters_ws.row_dimensions[i].height = 42
583
584
        for i in range(8, parameters_timestamps_data_max_len + 10):
585
            parameters_ws.row_dimensions[i].height = 60
586
587
        # Col width
588
        parameters_ws.column_dimensions['A'].width = 1.5
589
590
        parameters_ws.column_dimensions['B'].width = 25.0
591
592
        for i in range(3, 12 + parameters_names_len * 3):
593
            parameters_ws.column_dimensions[format_cell.get_column_letter(i)].width = 15.0
594
595
        # Img
596
        img = Image("excelexporters/myems.png")
597
        parameters_ws.add_image(img, 'A1')
598
599
        # Title
600
601
        parameters_ws['B3'].alignment = b_r_alignment
602
        parameters_ws['B3'] = _('Name') + ':'
603
        parameters_ws['C3'].border = b_border
604
        parameters_ws['C3'].alignment = b_c_alignment
605
        parameters_ws['C3'] = name
606
607
        parameters_ws['D3'].alignment = b_r_alignment
608
        parameters_ws['D3'] = _('Period Type') + ':'
609
        parameters_ws['E3'].border = b_border
610
        parameters_ws['E3'].alignment = b_c_alignment
611
        parameters_ws['E3'] = period_type
612
613
        parameters_ws['B4'].alignment = b_r_alignment
614
        parameters_ws['B4'] = _('Reporting Start Datetime') + ':'
615
        parameters_ws['C4'].border = b_border
616
        parameters_ws['C4'].alignment = b_c_alignment
617
        parameters_ws['C4'] = reporting_start_datetime_local
618
619
        parameters_ws['D4'].alignment = b_r_alignment
620
        parameters_ws['D4'] = _('Reporting End Datetime') + ':'
621
        parameters_ws['E4'].border = b_border
622
        parameters_ws['E4'].alignment = b_c_alignment
623
        parameters_ws['E4'] = reporting_end_datetime_local
624
625
        parameters_ws_current_row_number = 6
626
627
        parameters_ws['B' + str(parameters_ws_current_row_number)].font = title_font
628
        parameters_ws['B' + str(parameters_ws_current_row_number)] = name + ' ' + _('Parameters')
629
630
        parameters_ws_current_row_number += 1
631
632
        parameters_table_start_row_number = parameters_ws_current_row_number
633
634
        parameters_ws.row_dimensions[parameters_ws_current_row_number].height = 80
635
636
        parameters_ws_current_row_number += 1
637
638
        table_current_col_number = 2
639
640
        for i in range(0, parameters_names_len):
641
642
            if len(parameters_data['timestamps'][i]) == 0:
643
                continue
644
645
            col = format_cell.get_column_letter(table_current_col_number)
646
647
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].fill = table_fill
648
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].border = f_border
649
650
            col = format_cell.get_column_letter(table_current_col_number + 1)
651
652
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].fill = table_fill
653
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].border = f_border
654
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].font = name_font
655
            parameters_ws[col + str(parameters_ws_current_row_number - 1)].alignment = c_c_alignment
656
            parameters_ws[col + str(parameters_ws_current_row_number - 1)] = parameters_data['names'][i]
657
658
            table_current_row_number = parameters_ws_current_row_number
659
660
            for j, value in enumerate(list(parameters_data['timestamps'][i])):
661
                col = format_cell.get_column_letter(table_current_col_number)
662
663
                parameters_ws[col + str(table_current_row_number)].border = f_border
664
                parameters_ws[col + str(table_current_row_number)].font = title_font
665
                parameters_ws[col + str(table_current_row_number)].alignment = c_c_alignment
666
                parameters_ws[col + str(table_current_row_number)] = value
667
668
                col = format_cell.get_column_letter(table_current_col_number + 1)
669
670
                parameters_ws[col + str(table_current_row_number)].border = f_border
671
                parameters_ws[col + str(table_current_row_number)].font = title_font
672
                parameters_ws[col + str(table_current_row_number)].alignment = c_c_alignment
673
                parameters_ws[col + str(table_current_row_number)] = round(parameters_data['values'][i][j], 2)
674
675
                table_current_row_number += 1
676
677
            table_current_col_number = table_current_col_number + 3
678
679
        ################################################################################################################
680
        # parameters chart and parameters table
681
        ################################################################################################################
682
683
        ws['B' + str(current_sheet_parameters_row_number)].font = title_font
684
        ws['B' + str(current_sheet_parameters_row_number)] = name + ' ' + _('Parameters')
685
686
        current_sheet_parameters_row_number += 1
687
688
        chart_start_row_number = current_sheet_parameters_row_number
689
690
        col_index = 0
691
692
        for i in range(0, parameters_names_len):
693
694
            if len(parameters_data['timestamps'][i]) == 0:
695
                continue
696
697
            line = LineChart()
698
            data_col = 3 + col_index * 3
699
            labels_col = 2 + col_index * 3
700
            col_index += 1
701
            line.title = _('Parameters') + ' - ' + \
702
                parameters_ws.cell(row=parameters_table_start_row_number, column=data_col).value
703
            labels = Reference(parameters_ws, min_col=labels_col, min_row=parameters_table_start_row_number + 1,
704
                               max_row=(len(parameters_data['timestamps'][i]) + parameters_table_start_row_number))
705
            line_data = Reference(parameters_ws, min_col=data_col, min_row=parameters_table_start_row_number,
706
                                  max_row=(len(parameters_data['timestamps'][i]) + parameters_table_start_row_number))
707
            line.add_data(line_data, titles_from_data=True)
708
            line.set_categories(labels)
709
            line_data = line.series[0]
710
            line_data.marker.symbol = "circle"
711
            line_data.smooth = True
712
            line.x_axis.crosses = 'min'
713
            line.height = 8.25
714
            line.width = 24
715
            line.dLbls = DataLabelList()
716
            line.dLbls.dLblPos = 't'
717
            line.dLbls.showVal = False
718
            line.dLbls.showPercent = False
719
            chart_col = 'B'
720
            chart_cell = chart_col + str(chart_start_row_number)
721
            chart_start_row_number += 6
722
            ws.add_chart(line, chart_cell)
723
724
        current_sheet_parameters_row_number = chart_start_row_number
725
726
        current_sheet_parameters_row_number += 1
727
    ####################################################################################################################
728
    filename = str(uuid.uuid4()) + '.xlsx'
729
    wb.save(filename)
730
731
    return filename
732
733
734
def timestamps_data_all_equal_0(lists):
735
    for i, value in enumerate(list(lists)):
736
        if len(value) > 0:
737
            return False
738
739
    return True
740
741
742
def get_parameters_timestamps_lists_max_len(parameters_timestamps_lists):
743
    max_len = 0
744
    for i, value in enumerate(list(parameters_timestamps_lists)):
745
        if len(value) > max_len:
746
            max_len = len(value)
747
748
    return max_len
749
750
751
def timestamps_data_not_equal_0(lists):
752
    number = 0
753
    for i, value in enumerate(list(lists)):
754
        if len(value) > 0:
755
            number += 1
756
    return number
757
758
759 View Code Duplication
def is_base_period_timestamp_exists(base_period_data):
0 ignored issues
show
Duplication introduced by
This code seems to be duplicated in your project.
Loading history...
760
    timestamps = base_period_data['timestamps']
761
762
    if len(timestamps) == 0:
763
        return False
764
765
    for timestamp in timestamps:
766
        if len(timestamp) > 0:
767
            return True
768
769
    return False
770