1 files changed, 8 insertions, 3 deletions
diff --git a/update_sharing.py b/update_sharing.py
index 5ec6c7b..1ff0fd8 100755
--- a/update_sharing.py
+++ b/update_sharing.py
@@ -1,5 +1,6 @@
 #!/usr/bin/python
 
+import optparse
 import sqlite3
 
 from dedup.utils import fetchiter
@@ -37,8 +38,7 @@ def process_pkgdict(cursor, pkgdict):
                     insert_key = (pid1, pid2, fid1, fid2)
                     add_values(cursor, insert_key, pkgnumfiles, pkgsize)
 
-def main():
-    db = sqlite3.connect("test.sqlite3")
+def main(db):
     cur = db.cursor()
     cur.execute("PRAGMA foreign_keys = ON;")
     cur.execute("DELETE FROM sharing;")
@@ -61,4 +61,9 @@ def main():
     db.commit()
 
 if __name__ == "__main__":
-    main()
+    parser = optparse.OptionParser()
+    parser.add_option("-d", "--database", action="store",
+                      default="test.sqlite3",
+                      help="path to the sqlite3 database file")
+    options, args = parser.parse_args()
+    main(sqlite3.connect(options.database))