pyrecli 0.1.3__tar.gz → 0.2.0__tar.gz

{pyrecli-0.1.3 → pyrecli-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyrecli
-Version: 0.1.3
+Version: 0.2.0
 Summary: Command line utilities for DiamondFire templates
 Home-page: https://github.com/Amp63/pyrecli
 License: MIT

{pyrecli-0.1.3 → pyrecli-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyrecli"
-version = "0.1.3"
+version = "0.2.0"
 description = "Command line utilities for DiamondFire templates"
 authors = ["Amp"]
 readme = "README.md"

pyrecli-0.2.0/pyrecli/command/slice.py

@@ -0,0 +1,21 @@
+from result import Result, Err
+from pyrecli.util import read_input_file, write_output_file, parse_templates_from_string
+
+def slice_command(input_path: str, output_path: str, target_length: int) -> Result[None, str]:
+    input_result = read_input_file(input_path)
+    if input_result.is_err():
+        return Err(input_result.err_value)
+
+    templates_result = parse_templates_from_string(input_result.ok_value)
+    if templates_result.is_err():
+        return Err(templates_result.err_value)
+    templates = templates_result.ok_value
+
+    if not templates:
+        return Err(f'Could not find any templates in {input_path}')
+    
+    first_template = templates[0]
+    sliced_templates = first_template.slice(target_length)
+    built_templates = [t.build() for t in sliced_templates]
+
+    return write_output_file(output_path, '\n'.join(built_templates))

{pyrecli-0.1.3 → pyrecli-0.2.0}/pyrecli/pyrecli.py

@@ -8,6 +8,15 @@ from pyrecli.command.script import script_command
 from pyrecli.command.rename import rename_command
 from pyrecli.command.grabinv import grabinv_command
 from pyrecli.command.docs import docs_command
+from pyrecli.command.slice import slice_command
+
+
+def slice_target_length(value):
+    MINIMUM_LENGTH = 5
+    ivalue = int(value)
+    if ivalue < MINIMUM_LENGTH:
+        raise argparse.ArgumentTypeError(f'Target length must be at least {MINIMUM_LENGTH} codeblocks')
+    return ivalue
 
 
 def main():
@@ -36,18 +45,23 @@ def main():
     parser_rename.add_argument('var_to_rename', help='The variable to rename', type=str)
     parser_rename.add_argument('new_var_name', help='The new name for the variable', type=str)
     parser_rename.add_argument('--var_to_rename_scope', '-s', help='The scope to match', type=str, default=None)
-    parser_rename.add_argument('--output_path', '-o', help='The file or directory to output to', type=str, default=None)
+    parser_rename.add_argument('--output_path', '-o', help='The file to output to', type=str, default=None)
 
     parser_grabinv = subparsers.add_parser('grabinv', help='Save all templates in the inventory to a file with CodeClient')
     parser_grabinv.add_argument('output_path', help='The file to output template data to', type=str)
 
     parser_docs = subparsers.add_parser('docs', help='Generate markdown documentation from template data')
     parser_docs.add_argument('input_path', help='The file containing template data', type=str)
-    parser_docs.add_argument('output_path', help='The file or directory to output to', type=str)
+    parser_docs.add_argument('output_path', help='The file to output to', type=str)
     parser_docs.add_argument('title', help='The title for the docs', type=str)
     parser_docs.add_argument('--include_hidden', '-ih', help='Include hidden functions and processes', action='store_true')
     parser_docs.add_argument('--notoc', help='Omit the table of contents', action='store_true')
 
+    parser_slice = subparsers.add_parser('slice', help='Slice a template into multiple smaller templates')
+    parser_slice.add_argument('input_path', help='The file containing template data', type=str)
+    parser_slice.add_argument('output_path', help='The file to output template data to', type=str)
+    parser_slice.add_argument('target_length', help='The maximum length of each sliced template', type=slice_target_length)
+    
     parsed_args = parser.parse_args()
 
     match parsed_args.command:
@@ -81,6 +95,12 @@ def main():
                 parsed_args.input_path, parsed_args.output_path,
                 parsed_args.title, parsed_args.include_hidden, parsed_args.notoc
             )
+        
+        case 'slice':
+            command_result = slice_command(
+                parsed_args.input_path, parsed_args.output_path,
+                parsed_args.target_length
+            )
     
     if command_result.is_err():
         print(command_result.err_value)

{pyrecli-0.1.3 → pyrecli-0.2.0}/pyrecli/util.py

@@ -1,5 +1,6 @@
 import os
 import re
+import sys
 from result import Result, Ok, Err
 import websocket
 from dfpyre import DFTemplate
@@ -47,9 +48,10 @@ def parse_templates_from_string(templates: str) -> Result[list[DFTemplate], str]
 def read_input_file(path: str) -> Result[str, str]:
     if path == '-':
         try:
-            input_string = input()
+            input_string = sys.stdin.read()
         except EOFError:
-            return Ok(input_string)
+            pass
+        return Ok(input_string.strip())
     
     if not os.path.isfile(path):
         return Err(f'"{path}" is not a file.')