initial commit: added user_config and load_user_info, tests not written yet

Author: yucongalicechen

src/diffpy/labpdfproc/labpdfprocapp.py

@@ -4,6 +4,7 @@
 from diffpy.labpdfproc.functions import apply_corr, compute_cve
 from diffpy.labpdfproc.tools import (
     known_sources,
+    load_user_info,
     load_user_metadata,
     set_input_lists,
     set_output_directory,
@@ -96,6 +97,7 @@ def get_args(override_cli_inputs=None):
 
 def main():
     args = get_args()
+    args = load_user_info(args)
     args = set_input_lists(args)
     args.output_directory = set_output_directory(args)
     args.wavelength = set_wavelength(args)

src/diffpy/labpdfproc/tools.py

@@ -1,5 +1,7 @@
 from pathlib import Path
 
+from diffpy.labpdfproc.user_config import find_conf_file, read_conf_file, user_info_conf
+
 WAVELENGTHS = {"Mo": 0.71, "Ag": 0.59, "Cu": 1.54}
 known_sources = [key for key in WAVELENGTHS.keys()]
 
@@ -171,3 +173,17 @@ def load_user_metadata(args):
             setattr(args, key, value)
     delattr(args, "user_metadata")
     return args
+
+
+def load_user_info(args):
+    conf_file_path = find_conf_file()
+    if conf_file_path is not None:
+        conf_file = read_conf_file(conf_file_path)
+        if "username" in conf_file and "useremail" in conf_file:
+            setattr(args, "username", conf_file["username"])
+            setattr(args, "useremail", conf_file["useremail"])
+            return args
+        else:
+            return user_info_conf(args)
+    else:
+        return user_info_conf(args)

src/diffpy/labpdfproc/user_config.py

@@ -0,0 +1,37 @@
+import json
+import os
+from pathlib import Path
+
+CONFIG_FILE = "labpdfprocconfig.json"
+
+
+def find_conf_file():
+    # Return config file path if such a file exists
+    if os.path.exists(CONFIG_FILE):
+        return Path(CONFIG_FILE).resolve()
+    return None
+
+
+def read_conf_file(file_path):
+    with open(file_path, "r") as f:
+        return json.load(f)
+
+
+def write_conf_file(file_path, username, useremail):
+    config = {"username": username, "useremail": useremail}
+    with open(file_path, "w") as f:
+        json.dump(config, f)
+
+
+def prompt_user_info():
+    username = input("Please enter your username (or press Enter to skip): ")
+    useremail = input("Please enter your email (or press Enter to skip): ")
+    return username, useremail
+
+
+def user_info_conf(args):
+    username, useremail = prompt_user_info()
+    write_conf_file(CONFIG_FILE, username, useremail)
+    setattr(args, "username", username)
+    setattr(args, "useremail", useremail)
+    return args