nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix(data): ensure output directories are created before saving files

Browse Source 
Closes #35
This commit is contained in:
nuluh
2025-03-16 18:30:52 +07:00
parent 8ed1437d6d
commit 35e25ba4c6
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
data/QUGS/convert.py
View File

@@ -36,6 +36,7 @@ def create_damage_files(base_path, output_base, prefix):
print(f"Creating {output_file_1} from taking zzz{prefix}D{file_index}.TXT") print(f"Creating {output_file_1} from taking zzz{prefix}D{file_index}.TXT")
print("Taking datetime column on index 0...") print("Taking datetime column on index 0...")
print(f"Taking `{top_sensor}`...") print(f"Taking `{top_sensor}`...")
os.makedirs(os.path.dirname(output_file_1), exist_ok=True)
df[['Time', top_sensor]].to_csv(output_file_1, index=False) df[['Time', top_sensor]].to_csv(output_file_1, index=False)
print(Fore.GREEN + "Done") print(Fore.GREEN + "Done")
@@ -44,6 +45,7 @@ def create_damage_files(base_path, output_base, prefix):
print(f"Creating {output_file_2} from taking zzz{prefix}D{file_index}.TXT") print(f"Creating {output_file_2} from taking zzz{prefix}D{file_index}.TXT")
print("Taking datetime column on index 0...") print("Taking datetime column on index 0...")
print(f"Taking `{bottom_sensor}`...") print(f"Taking `{bottom_sensor}`...")
os.makedirs(os.path.dirname(output_file_2), exist_ok=True)
df[['Time', bottom_sensor]].to_csv(output_file_2, index=False) df[['Time', bottom_sensor]].to_csv(output_file_2, index=False)
print(Fore.GREEN + "Done") print(Fore.GREEN + "Done")
print("---") print("---")
@@ -58,8 +60,8 @@ def main():
prefix = sys.argv[3] # Define output directory prefix = sys.argv[3] # Define output directory
# Create output folders if they don't exist # Create output folders if they don't exist
for i in range(1, 5): # for i in range(1, 7):
os.makedirs(os.path.join(output_base, f'DAMAGE_{i}'), exist_ok=True) # os.makedirs(os.path.join(output_base, f'DAMAGE_{i}'), exist_ok=True)
create_damage_files(base_path, output_base, prefix) create_damage_files(base_path, output_base, prefix)
print(Fore.YELLOW + Style.BRIGHT + "All files have been created successfully.") print(Fore.YELLOW + Style.BRIGHT + "All files have been created successfully.")