bin/nixpkgs_drv_pname: add --db flag

1 files changed, 2 insertions, 1 deletions

diff --git a/bin/nixpkgs_drv_pname.py b/bin/nixpkgs_drv_pname.py
index 7209537..24a8d16 100755
--- a/bin/nixpkgs_drv_pname.py
+++ b/bin/nixpkgs_drv_pname.py
@@ -87,6 +87,7 @@ def args_get():
     parser = argparse.ArgumentParser()
     parser.add_argument('-r', '--ref', default="master")
     parser.add_argument('-a', '--arch', default="x86_64-linux")
+    parser.add_argument('-d', '--db', default="drv_pname_dag.db")
     return parser.parse_args()
 
 if __name__ == '__main__':
@@ -101,7 +102,7 @@ if __name__ == '__main__':
     if proc.stdout is None:
         raise EOFError('Failed to evaluate nixpkgs')
 
-    con = sqlite3.connect('drv_pname_dag.db')
+    con = sqlite3.connect(args.db)
     cur = con.cursor()
     cur.execute("""
         CREATE TABLE IF NOT EXISTS pnames (