mpremote: Allow user configuration on Windows

Josverl · Josverl · commit 8492145519d7 · 2023-03-12T22:53:46.000+01:00
micropython/micropython#9573 Signed-off-by: Jos Verlinde <Jos_Verlinde@hotmail.com>
diff --git a/tools/mpremote/mpremote/main.py b/tools/mpremote/mpremote/main.py
@@ -331,17 +331,22 @@ def load_user_config():
     # Create empty config object.
     config = __build_class__(lambda: None, "Config")()
     config.commands = {}
-
-    # Get config file name.
-    path = os.getenv("XDG_CONFIG_HOME")
-    if path is None:
-        path = os.getenv("HOME")
+    # use $XDG_CONFIG_HOME, if on Windows use $APPDATA
+    if os.name != "nt":
+        # Get config file name.
+        path = os.getenv("XDG_CONFIG_HOME")
+        if path is None:
+            path = os.getenv("HOME")
+            if path is None:
+                return config
+            path = os.path.join(path, ".config")
+    else:
+        path = os.getenv("HOME") or os.getenv("APPDATA")
         if path is None:
             return config
-        path = os.path.join(path, ".config")
     path = os.path.join(path, _PROG)
     config_file = os.path.join(path, "config.py")
-
+    print("Loading config from: ", config_file)
     # Check if config file exists.
     if not os.path.exists(config_file):
         return config
@@ -351,6 +356,9 @@ def load_user_config():
         config_data = f.read()
     prev_cwd = os.getcwd()
     os.chdir(path)
+    # pass in the config path so that the config file can use it
+    config.__dict__["config_path"] = path
+    config.__dict__["__file__"] = config_file
     exec(config_data, config.__dict__)
     os.chdir(prev_cwd)
 
