Parse database command line argument

1 files changed, 3 insertions, 1 deletions

diff --git a/src/main.py b/src/main.py
index 293ec8d..e791196 100644
--- a/src/main.py
+++ b/src/main.py
@@ -5,8 +5,10 @@ import os.path
 
 import db.init
 import gui.window
+import arguments
 
-database = db.init.init('database')
+args = arguments.parser().parse_args()
+database = db.init.init(args.database if args.database != None else 'database')
 app = QtWidgets.QApplication(sys.argv)
 
 # # Allows to catch Ctrl-C event