scott/statscollect
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added glob

Browse Source
This commit is contained in:
Marc Mance
2024-08-28 11:33:25 -04:00
parent c068b11b60
commit 74d3df04d3
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
collect.py
View File

@@ -8,7 +8,7 @@ import socket
import platform import platform
import argparse import argparse
import logging import logging
import pathlib import glob
from datetime import datetime from datetime import datetime
@@ -331,6 +331,7 @@ def run_debug():
print(f"Error running midclt debug: {e}") print(f"Error running midclt debug: {e}")
return None return None
def cleanCSV(): def cleanCSV():
if is_debian(): if is_debian():
dir = "/var/log/proftpd" dir = "/var/log/proftpd"
@@ -338,12 +339,12 @@ def cleanCSV():
if is_freebsd(): if is_freebsd():
dir = "/var/log" 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: for filename in csv_files:
os.remove(filename) os.remove(filename)
def collect_csv(): def collect_csv():
source_dir = os.getcwd() source_dir = os.getcwd()
@@ -520,7 +521,7 @@ def main():
# Upload data (replace with credentials and actual upload logic if needed) # Upload data (replace with credentials and actual upload logic if needed)
upload_debug() upload_debug()
cleanCSV() cleanCSV()
print("Data collection and upload completed.") print("Data collection and upload completed.")