Add integrity checking of db files

1 files changed, 29 insertions, 1 deletions

diff --git a/AntennaPodDbFixer.py b/AntennaPodDbFixer.py
index f772c9d..9966209 100755
--- a/AntennaPodDbFixer.py
+++ b/AntennaPodDbFixer.py
@@ -9,6 +9,24 @@ import getopt
 def usage():
     return sys.argv[0] + " [--help] [--verbose] [corrupt.db] [empty.db]"
 
+def integrity_is_ok(fileName):
+    print("Checking integrity of " + fileName + ".", end="")
+    if verbose:
+        print("sqlite3 '" + fileName + "' 'PRAGMA integrity_check;'")
+    integrityCheck = subprocess.run(
+            ["sqlite3", fileName, "PRAGMA integrity_check;"],
+            capture_output=True,
+            text=True
+        ).stdout.strip()
+    is_ok = integrityCheck == "ok"
+    if is_ok:
+        print(" Detected no errors.")
+    else:
+        print(" Errors found.")
+    if verbose:
+        print(integrityCheck)
+    return is_ok
+
 try:
     opts, args = getopt.gnu_getopt(sys.argv, "hv", ["help", "verbose"])
 except getopt.GetoptError as err:
@@ -54,6 +72,8 @@ if corruptedVersion == "0":
     exit(1)
 print("Corrupted file version: " + corruptedVersion)
 
+integrity_is_ok(inputFilePath)
+
 if len(args) >= 2:
     emptyFilePath = args[2]
 else:
@@ -76,13 +96,16 @@ repairedFilePath = inputFilePath + "-repaired.db"
 sqlFilePath = inputFilePath + ".sql.tmp"
 workingcopyFilePath = inputFilePath + ".tmp"
 
+if not integrity_is_ok(emptyFilePath):
+    print("Errors found in " + emptyFilePath + ". Giving up!", file=sys.stderr)
+    exit(1)
+
 if verbose:
     print("cp '" + emptyFilePath + "' '" + repairedFilePath + "'")
 shutil.copyfile(emptyFilePath, repairedFilePath, follow_symlinks=True)
 if os.path.exists(sqlFilePath): os.remove(sqlFilePath)
 if os.path.exists(workingcopyFilePath): os.remove(workingcopyFilePath)
 
-
 # Recover to SQL commands and insert back into a database
 print("Recovering database.")
 if verbose:
@@ -157,4 +180,9 @@ if verbose:
     print("rm '" + sqlFilePath + "' '" + workingcopyFilePath + "'")
 os.remove(sqlFilePath)
 os.remove(workingcopyFilePath)
+
+if not integrity_is_ok(repairedFilePath):
+    print("Integrity check of " + repairedFilePath + " still found errors.", file=sys.stderr)
+    exit(1)
+
 print("Done. Output file: " + repairedFilePath)