Skip to content

Commit 50253ec

Browse files
committed
utils-logger implimented
1 parent 73692c8 commit 50253ec

File tree

1 file changed

+11
-22
lines changed

1 file changed

+11
-22
lines changed

utils.py

Lines changed: 11 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -8,36 +8,25 @@
88
class logger:
99
@staticmethod
1010
def log_title(title: str, title_len: int = TITLE_LEN):
11-
if len(title) + MIN_SIDE_PADDING * 2 + SIDE_WHITE_SPACES * 2 > title_len:
12-
need_sz = title_len - MIN_SIDE_PADDING * 2 - SIDE_WHITE_SPACES * 2
13-
14-
title = title[:need_sz]
15-
16-
padding = title_len - (len(title) + SIDE_WHITE_SPACES * 2)
17-
18-
right_padding = (padding + 1) // 2
19-
left_padding = padding // 2
20-
21-
print(
22-
f"{left_padding * '='}{SIDE_WHITE_SPACES * ' '}{title}{SIDE_WHITE_SPACES * ' '}{right_padding * '='}"
11+
final_len = max(
12+
title_len, len(title) + MIN_SIDE_PADDING * 2 + SIDE_WHITE_SPACES * 2
2313
)
2414

15+
print(f"{f" {title} ":=^{final_len}}")
16+
2517
@staticmethod
2618
def log_to_csv(csv_name: str, field_names: tuple[str], row: dict | None = None):
2719
if isinstance(row, dict):
28-
file = open(csv_name, 'a', newline='')
29-
writer = csv.DictWriter(file, fieldnames=field_names)
30-
elif row == None:
31-
file = open(csv_name, 'w', newline='')
32-
writer = csv.writer(file)
33-
row = field_names
20+
with open(csv_name, 'a', newline='') as file:
21+
writer = csv.DictWriter(file, fieldnames=field_names)
22+
writer.writerow(row)
23+
elif row is None:
24+
with open(csv_name, 'w', newline='') as file:
25+
writer = csv.writer(file)
26+
writer.writerow(field_names)
3427
else:
3528
raise TypeError(f"row has type {type(row)} but must be [ dict | None ]")
3629

37-
writer.writerow(row)
38-
39-
file.close()
40-
4130
@staticmethod
4231
def log_to_stdout(info: dict):
4332
print(info)

0 commit comments

Comments
 (0)