mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-07 09:36:05 -08:00
99 lines
3.1 KiB
Python
Executable File
99 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# test script by running:
|
|
# /home/pi/pialert/front/plugins/csv_backup/script.py overwrite=False location=/home/pi/pialert/config
|
|
|
|
import os
|
|
import pathlib
|
|
import argparse
|
|
import sys
|
|
import hashlib
|
|
import csv
|
|
import sqlite3
|
|
from io import StringIO
|
|
from datetime import datetime
|
|
|
|
sys.path.append("/home/pi/pialert/front/plugins")
|
|
sys.path.append('/home/pi/pialert/pialert')
|
|
|
|
from plugin_helper import Plugin_Object, Plugin_Objects, decodeBase64
|
|
from logger import mylog, append_line_to_file
|
|
from helper import timeNowTZ
|
|
from const import logPath, pialertPath
|
|
|
|
|
|
CUR_PATH = str(pathlib.Path(__file__).parent.resolve())
|
|
LOG_FILE = os.path.join(CUR_PATH, 'script.log')
|
|
RESULT_FILE = os.path.join(CUR_PATH, 'last_result.log')
|
|
|
|
def main():
|
|
|
|
# the script expects a parameter in the format of devices=device1,device2,...
|
|
parser = argparse.ArgumentParser(description='Export devices data to CSV')
|
|
parser.add_argument('overwrite', action="store", help="Specify 'TRUE' to overwrite an existing file, or 'FALSE' to create a new file")
|
|
parser.add_argument('location', action="store", help="The directory where the CSV file will be saved")
|
|
values = parser.parse_args()
|
|
|
|
overwrite = values.overwrite.split('=')[1]
|
|
|
|
if (overwrite.upper() == "TRUE"):
|
|
overwrite = True
|
|
else:
|
|
overwrite = False
|
|
|
|
mylog('verbose', ['[CSVBCKP] In script'])
|
|
|
|
# Connect to the PiAlert SQLite database
|
|
conn = sqlite3.connect('/home/pi/pialert/db/pialert.db')
|
|
cursor = conn.cursor()
|
|
|
|
# Execute your SQL query
|
|
cursor.execute("SELECT * FROM Devices")
|
|
|
|
# Get column names
|
|
columns = [desc[0] for desc in cursor.description]
|
|
|
|
if overwrite:
|
|
filename = 'devices.csv'
|
|
else:
|
|
timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
|
|
filename = f'devices_{timestamp}.csv'
|
|
|
|
fullPath = os.path.join(values.location.split('=')[1], filename)
|
|
|
|
mylog('verbose', ['[CSVBCKP] Writing file ', fullPath])
|
|
|
|
# Create a CSV file in the specified location
|
|
with open(fullPath, 'w', newline='') as csvfile:
|
|
# Initialize the CSV writer
|
|
csv_writer = csv.writer(csvfile, delimiter=',', quoting=csv.QUOTE_MINIMAL)
|
|
|
|
# Wrap the header values in double quotes and write the header row
|
|
csv_writer.writerow([ '"' + col + '"' for col in columns])
|
|
|
|
# Fetch and write data rows
|
|
for row in cursor.fetchall():
|
|
# Wrap each value in double quotes and write the row
|
|
csv_writer.writerow(['"' + str(value) + '"' for value in row])
|
|
|
|
# Close the database connection
|
|
conn.close()
|
|
|
|
# Open the CSV file for reading
|
|
with open(fullPath, 'r') as file:
|
|
data = file.read()
|
|
|
|
# Replace all occurrences of """ with "
|
|
data = data.replace('"""', '"')
|
|
|
|
# Open the CSV file for writing
|
|
with open(fullPath, 'w') as file:
|
|
file.write(data)
|
|
|
|
return 0
|
|
|
|
|
|
#===============================================================================
|
|
# BEGIN
|
|
#===============================================================================
|
|
if __name__ == '__main__':
|
|
main() |