Abstracted init.json_dir()

Author: adamlui

remove-json-keys/__main__.py

@@ -15,23 +15,11 @@
 
 # Determine closest locales dir
 log.trunc(f'\nSearching for {cli.json_dir}...')
-script_dir = os.path.abspath(os.path.dirname(__file__))
-for root, dirs, files in os.walk(script_dir): # search script dir recursively
-    if cli.json_dir in dirs:
-        cli.json_dir = os.path.join(root, cli.json_dir) ; break
-else: # search script parent dirs recursively
-    parent_dir = os.path.dirname(script_dir)
-    while parent_dir and parent_dir != script_dir:
-        for root, dirs, files in os.walk(parent_dir):
-            if cli.json_dir in dirs:
-                cli.json_dir = os.path.join(root, cli.json_dir) ; break
-        if cli.json_dir : break
-        parent_dir = os.path.dirname(parent_dir)
-    else : cli.json_dir = None
-
-# Print result
-if cli.json_dir : log.trunc(f'JSON directory found!\n\n>> {cli.json_dir}\n')
-else : log.trunc(f'Unable to locate a {cli.json_dir} directory.') ; exit()
+cli.json_dir = init.json_dir(cli.json_dir)
+if cli.json_dir:
+    log.trunc(f'JSON directory found!\n\n>> {cli.json_dir}\n')
+else:
+    log.trunc(f'Unable to locate a {cli.json_dir} directory.') ; exit()
 
 # Process JSON files and remove specified keys
 keys_removed, keys_skipped, processed_cnt = [], [], 0

remove-json-keys/lib/init.py

@@ -1,11 +1,11 @@
-import argparse
+import argparse, os
 from types import SimpleNamespace as sns
 
 def cli():
 
     cli = sns(
         name='remove-json-keys',
-        version='2026.2.10.29',
+        version='2026.2.10.30',
         author=sns(name='Adam Lui', email='adam@kudoa.com', url='https://github.com/adamlui'),
         description='Remove key/value pairs from json_dir/**.json',
         urls=sns(
@@ -24,3 +24,19 @@ def cli():
     cli.json_dir = cli.args.json_dir or '_locales'
 
     return cli
+
+def json_dir(json_dir):
+    script_dir = os.path.abspath(os.path.dirname(__file__))
+    for root, dirs, files in os.walk(script_dir): # search script dir recursively
+        if json_dir in dirs:
+            json_dir = os.path.join(root, json_dir) ; break
+    else: # search script parent dirs recursively
+        parent_dir = os.path.dirname(script_dir)
+        while parent_dir and parent_dir != script_dir:
+            for root, dirs, files in os.walk(parent_dir):
+                if json_dir in dirs:
+                    json_dir = os.path.join(root, json_dir) ; break
+            if json_dir : break
+            parent_dir = os.path.dirname(parent_dir)
+        else : json_dir = None
+    return json_dir