Added command line parser

arguments.py

@@ -0,0 +1,11 @@
+import argparse
+
+parser = argparse.ArgumentParser(description='App scans through a directory and stores metadata of .mkv files in MySQL database')
+parser.add_argument('database_name', type=str, help='the name of the database')
+parser.add_argument('database_user', type=str, help='the user to connect to the database')
+parser.add_argument('database_password', type=str, help='the password to connect to the database')
+parser.add_argument('folder', type=str, help='the folder containing the .mkv files')
+
+args = parser.parse_args()
+
+# access the arguments using args.database_name, args.database_user, args.database_password, and args.folder

main.py

@@ -2,12 +2,14 @@
 import os
 import json
 import sql_connection
-import ast
+import arguments
+
 
 def main():
-    rootdir = '/home/host/movies'
+    rootdir = arguments.args.folder
 
     for subdir, dirs, files in os.walk(rootdir):
+
         name = str(subdir.rsplit('/', 1)[-1]) # Title (YEAR)
         title = ''
         year = ''
@@ -32,8 +34,6 @@ def main():
 
                     dict_audio_data = json.loads(a_str)
 
-                    print(dict_audio_data["name"])
-
                     dict_data_audio_list.append(dict_audio_data)
 
                 for v in video:

sql_connection.py

@@ -1,12 +1,13 @@
 import mysql.connector
 import json
+import arguments
 
 def connect_to_database():
     return mysql.connector.connect(
         host="localhost",
-        user="movie",
-        password="1234",
-        database="moviedb"
+        user=arguments.args.database_user,
+        password=arguments.args.database_password,
+        database=arguments.args.database_name
     )
 
 def insert_movie(title, video_data, audio_data_list):