diff --git a/collect.py b/collect.py index 99cf733..d19564f 100644 --- a/collect.py +++ b/collect.py @@ -8,7 +8,7 @@ import socket import platform import argparse import logging -import pathlib +import glob from datetime import datetime @@ -331,6 +331,7 @@ def run_debug(): print(f"Error running midclt debug: {e}") return None + def cleanCSV(): if is_debian(): dir = "/var/log/proftpd" @@ -338,12 +339,12 @@ def cleanCSV(): if is_freebsd(): dir = "/var/log" - csv_files = glob.glob(os.path.join(dir, '*.csv')) + csv_files = glob.glob(os.path.join(dir, "*.csv")) - for filename in csv_files: os.remove(filename) - + + def collect_csv(): source_dir = os.getcwd() @@ -520,7 +521,7 @@ def main(): # Upload data (replace with credentials and actual upload logic if needed) upload_debug() cleanCSV() - + print("Data collection and upload completed.")