rpi/pihole/migratePiholeAuditLog.py

56 lines
1.6 KiB
Python

# Python-Script: migratePiholeAuditLog.py - https://github.com/Zelo72/rpi (/pihole/)
#
# Beschreibung: Migriert die AuditLog-Domains aus auditlog.list von Pihole 4 in die domain_audit
# Tabelle von Pihole 5.
# Der AuditLog wird beim Update von Version 4 auf Version 5 nicht automatisch mit
# migriert.
# Domains aus auditlog.list werden nur hinzugefügt, wenn sie in der Tabelle
# domain_audit noch nicht vorhanden sind.
#
# Aufruf: sudo python3 migratePiholeAuditLog.py
#
# Versionshistorie:
# Version 1.0.0 - [Zelo72] - initiale Version
import sys
import sqlite3
from pathlib import Path
auditlog = "/etc/pihole/auditlog.list"
gravitydb = "/etc/pihole/gravity.db"
if not Path(auditlog).exists():
print(auditlog + " nicht gefunden!")
exit(1)
if not Path(gravitydb).exists():
print(gravitydb + " nicht gefunden!")
exit(1)
try:
con = sqlite3.connect(gravitydb)
file = open(auditlog)
cur = con.cursor()
for domain in file:
cur.execute(
'select * from domain_audit where domain = "{d}"'.format(d=str.strip(domain)))
rec = cur.fetchall()
if len(rec) == 0:
s = """INSERT INTO domain_audit (domain) VALUES ('{d}')""".format(
d=str.strip(domain))
cur.execute(s)
con.commit()
print("+++ " + str.strip(domain) + " hinzugefügt.")
else:
print("=== " + str.strip(domain) + " bereits vorhanden.")
except:
print("Unerwarteter Fehler:", sys.exc_info()[0])
raise
finally:
file.close()
cur.close()
con.close()