Update script

1 files changed, 15 insertions, 0 deletions

diff --git a/scripts/add_lins_to_db b/scripts/add_lins_to_db
index f6452a5..c3c4fba 100755
--- a/scripts/add_lins_to_db
+++ b/scripts/add_lins_to_db
@@ -14,6 +14,15 @@ import sqlite3
 from urllib import request
 
 def main():
+  parser = argparse.ArgumentParser(
+    description="Tool to create the gettor SQLite database."
+  )
+
+  parser.add_argument(
+    "-f", "--filename", default="gettor.db", metavar="gettor.db",
+    help="Database filename."
+  )
+
   args = parser.parse_args()
   abs_filename = os.path.abspath(args.filename)
 
@@ -60,3 +69,9 @@ def main():
                   "INSERT INTO links(link, platform, arch, version, provider, status)"
                   "VALUES ('%s', '%s' '64', '%s', 'gitlab', 'ACTIVE')" %(releases.get(key), k, version)
               )
+
+              
+if __name__ == "__main__":
+    print_header()
+    main()
+    print_footer