-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.py
36 lines (28 loc) · 981 Bytes
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
import shutil
def to_fixed(num: float):
return float("{:.2f}".format(num))
def reset_folder(directory_path: str):
if os.path.exists(directory_path):
for item in os.listdir(directory_path):
item_path = os.path.join(directory_path, item)
if os.path.isfile(item_path) or os.path.islink(item_path):
os.unlink(item_path)
elif os.path.isdir(item_path):
shutil.rmtree(item_path)
def count_file_and_folder(folder_path: str):
folder_count = 0
total_file_count = 0
for _, dirs, files in os.walk(folder_path):
folder_count += len(dirs)
folder_file_count = len(files)
total_file_count += folder_file_count
return folder_count, total_file_count
def find_best_solution(output):
_max = output[0][0]
_index = 0
for i, item in enumerate(output[0]):
if item > _max:
_max = item
_index = i
return _index, _max