Set correct db in gravity_parse. Start using with in dnsmasq

advanced/Scripts/dnsmasq_parse.py

@@ -147,21 +147,22 @@ logfile = sys.argv[1]
 
 # Create the SQLite connection
 conn = sqlite3.connect('/etc/pihole/pihole.db')
-c = conn.cursor()
 
-create_tables()
+with conn:
+    c = conn.cursor()
 
-# Parse the log file.
+    create_tables()
-for line in open(logfile):
-    line = line.rstrip()
 
-    if ': query[' in line:
+    # Parse the log file.
-        parse_query(line)
+    for line in open(logfile):
+        line = line.rstrip()
 
-    elif ': forwarded ' in line:
+        if ': query[' in line:
-        parse_forward(line)
+            parse_query(line)
 
-    elif (': reply ' in line) or (': cached ' in line):
+        elif ': forwarded ' in line:
-        parse_reply(line)
+            parse_forward(line)
+
+        elif (': reply ' in line) or (': cached ' in line):
+            parse_reply(line)
 
-conn.commit()

advanced/Scripts/gravity_parse.py

@@ -18,7 +18,7 @@ import sqlite3
 logfile = '/etc/pihole/pihole.2.eventHorizon.txt'
 
 # Create the SQLite connection
-conn = sqlite3.connect('/etc/pihole/gravity.db')
+conn = sqlite3.connect('/etc/pihole/pihole.db')
 
 # Python auto-handle commits, no need to call for commits manually
 with conn: