Issues (1656)

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