Improve argument handling of c2json (#22170)

* Improve argument handling of c2json

* Add automagic

1 files changed, 2 insertions, 4 deletions

diff --git a/lib/python/qmk/path.py b/lib/python/qmk/path.py
index 9d248451b8..94582a05e0 100644
--- a/lib/python/qmk/path.py
+++ b/lib/python/qmk/path.py
@@ -19,13 +19,11 @@ def is_keyboard(keyboard_name):
         return rules_mk.exists()
 
 
-def under_qmk_firmware():
+def under_qmk_firmware(path=Path(os.environ['ORIG_CWD'])):
     """Returns a Path object representing the relative path under qmk_firmware, or None.
     """
-    cwd = Path(os.environ['ORIG_CWD'])
-
     try:
-        return cwd.relative_to(QMK_FIRMWARE)
+        return path.relative_to(QMK_FIRMWARE)
     except ValueError:
         return None