Add arguments to function list parser

Author: Michael Lyons

resources/completion_builder/js.py

@@ -1,51 +1,66 @@
-# See main.py for overview.
+#! /usr/bin/env python3
 
 import json
 import re
-import sys
+from argparse import ArgumentParser
 
-app = sys.argv[1]
+def parse_function_list(sheet_app: str, write_json_file: bool = True):
 
-func_file = f'{app}_funcs_data_list.txt'
+    with open(f'{sheet_app}_funcs_data_list.txt', 'r') as file:
+        lines = [line.strip() for line in file]
 
-with open(func_file, 'r') as file:
-    func_list = [line.strip() for line in file]
+    # Debug
+    # lines = ['CALL~register_id,[argument1],...']
 
-# Debug
-# func_list = ['CALL~register_id,[argument1],...']
+    functions = []
 
-dict_list = []
+    for line in lines:
+        func_name, param_str = line.split('~')
+        param_list = param_str.split(',')
+        req_param = []
+        opt_param = []
 
-for line in func_list:
-    func_name = line.split('~')[0]
+        ellipsis = False
 
-    param_str = line.split('~')[1]
+        for param in param_list:
+            if param[0] == '[':
+                opt_param.append(re.sub(r'[\[\]]', '', param))
 
-    param_list = param_str.split(',')
+            elif param[0] == '.':
+                ellipsis = True
 
-    req_param = []
+            else:
+                req_param.append(param)
 
-    opt_param = []
+        func_dict = dict(
+            func_name = func_name,
+            req_param = req_param,
+            opt_param = opt_param,
+            ellipsis = ellipsis,
+        )
 
-    ellipsis = False
+        functions.append(func_dict)
 
-    for param in param_list:
-        if param[0] == '[':
-            opt_param.append(re.sub(r'[\[\]]', '', param))
+    if write_json_file:
+        with open(f'{sheet_app}_funcs.json', 'w') as o:
+            json.dump(functions, o, indent=4)
 
-        elif param[0] == '.':
-            ellipsis = True
+    return functions
 
-        else:
-            req_param.append(param)
 
-    func_dict = dict(func_name = func_name, req_param = req_param, opt_param = opt_param, ellipsis = ellipsis)
+if __name__ == '__main__':
+    parser = ArgumentParser(
+        description='Parse ~-delimited spreadsheet function list from a file')
+    parser.add_argument(
+        '-s', '--spreadsheet-app',
+        choices={'excel', 'gsheets', 'localc'},
+        default='excel',
+        help='spreadsheet application')
+    parser.add_argument(
+        '-j', '--dump-json',
+        type=bool,
+        default=True,
+        help='dump JSON output to a file')
+    args = parser.parse_args()
 
-    dict_list.append(func_dict)
-
-json_out = json.dumps(dict_list, indent = 4)
-
-o = open(f'{app}_funcs.json', 'w')
-
-o.write(json_out)
-o.close()
+    parse_function_list(args.spreadsheet_app)