File size: 1,534 Bytes
825e978 |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import os
import csv
def clean(user_prompt):
input_directory = os.path.join("input_folder", user_prompt)
output_directory = "downloads/"+user_prompt
# Create the output directory if it doesn't exist
os.makedirs(output_directory, exist_ok=True)
# Loop through all files in the user-specified input directory
for filename in os.listdir(input_directory):
file_path = os.path.join(input_directory, filename)
# Skip directories or hidden files
if os.path.isdir(file_path) or filename.startswith("."):
continue
# Output file path (.csv extension added)
output_file = os.path.join(output_directory, filename + ".csv")
with open(file_path, "r", encoding="utf-8", errors="ignore") as file:
lines = file.readlines()
headers = []
data_rows = []
data_started = False
for line in lines:
line = line.strip()
if line.startswith("@ATTRIBUTE"):
parts = line.split()
if len(parts) >= 2:
headers.append(parts[1])
elif line.startswith("@DATA"):
data_started = True
elif data_started and line:
data_rows.append(line.split(","))
# Write to CSV
with open(output_file, "w", newline="") as csvfile:
writer = csv.writer(csvfile)
writer.writerow(headers)
writer.writerows(data_rows)
print(f"✅ CSV file created for: {filename} → {output_file}")
|