Issues (1577)

myems-api/excelexporters/equipmentoutput.py (4 issues)

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