Automatically get the latest json file if only given a folder as path

edit_data_form.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 # encoding: utf-8
 
+import os
 import npyscreen
 import json
 import datetime
@@ -65,9 +66,20 @@ class FormApp(npyscreen.NPSAppManaged):
 def main(*_args):
     if len(sys.argv) > 1:
         file_path = pathlib.Path(sys.argv[1])
+
+        if(sys.argv[1].endswith(".json")):
+            # json file as argument
+
             if not file_path.exists():
                 print("Trying to open nonexistent file. Aborting.")
                 sys.exit(0)
+
+        elif file_path.is_dir():
+            # we want the latest json file in given folder
+            filesList = [f for f in os.listdir(file_path) if os.path.isfile(os.path.join(file_path, f)) and f.endswith(".json")]
+            filesList.sort()
+            file_path = os.path.join(file_path, filesList[len(filesList) - 1])
+
     else:
         file_path = pathlib.Path(pathlib.Path(__file__).absolute().parent, 'example.json')