Support giving empty.db as argument

1 files changed, 5 insertions, 2 deletions

diff --git a/AntennaPodDbFixer.py b/AntennaPodDbFixer.py
index 15d225a..b506ba6 100755
--- a/AntennaPodDbFixer.py
+++ b/AntennaPodDbFixer.py
@@ -6,7 +6,7 @@ import os
 import sys
 
 def usage():
-    return sys.argv[0] + " [--help] [corrupt.db]"
+    return sys.argv[0] + " [--help] [corrupt.db] [empty.db]"
 
 if len(sys.argv) > 1:
     if sys.argv[1] == "--help":
@@ -38,7 +38,10 @@ if corruptedVersion == "0":
     exit(1)
 print("Corrupted file version: " + corruptedVersion)
 
-emptyFilePath = "empty/" + corruptedVersion + ".db"
+if len(sys.argv) > 2:
+    emptyFilePath = sys.argv[2]
+else:
+    emptyFilePath = "empty/" + corruptedVersion + ".db"
 if not os.path.isfile(emptyFilePath):
     print("If needed, you can download old app versions on F-Droid and export an empty database.")
     emptyFilePath = input("Enter file path to an EMPTY AntennaPod database with the same version: ")