dekshell 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

dekshell/core/contexts/methods.py

@@ -6,8 +6,10 @@ import getpass
 from functools import reduce
 from itertools import chain
 from pathlib import Path
+from io import BytesIO, StringIO
 from dektools.file import sure_dir, write_file, read_text, remove_path, sure_parent_dir, normal_path, \
-    format_path_desc, read_file, split_ext, path_ext, clear_dir, copy_recurse_ignore, path_is_empty,\
+    format_path_desc, read_file, split_ext, path_ext, clear_dir, copy_recurse_ignore, path_is_empty, \
+    read_lines, \
     split_file, combine_split_files, remove_split_files, meta_split_file, tree, iglob, \
     where, where_list, which, which_list
 from dektools.hash import hash_file
@@ -129,8 +131,21 @@ def _install(name):
     shell_wrapper(command)
 
 
+def _io(x=None):
+    if x is None:
+        x = b''
+    return BytesIO(x) if isinstance(x, bytes) else StringIO(x)
+
+
+def _cd(path=None):
+    if path is None:
+        path = os.path.expanduser('~')
+    os.chdir(path)
+    return path
+
+
 path_common_methods = {
-    'cd': os.chdir,
+    'cd': _cd,
     'cwd': lambda: os.getcwd(),
     'which': lambda *x: which(*x) or '',
     'where': lambda *x: where(*x) or '',
@@ -140,6 +155,7 @@ path_common_methods = {
 }
 
 default_methods = {
+    'io': _io,
     'reduce': reduce,
     'chain': chain,
     'xeval': _xeval,
@@ -179,6 +195,7 @@ default_methods = {
         'rm': _remove_path,
         'wf': write_file,
         'rt': read_text,
+        'rl': read_lines,
         'rf': read_file,
         'ci': copy_recurse_ignore,
 

dekshell/core/markers/__init__.py

@@ -26,7 +26,8 @@ def generate_markers(*args, **kwargs):
         ErrorEchoMarker, EchoNoWrapMarker, EchoMarker,
         InputMarker,
         DelVarMarker,
-        ExecLinesUpdateMarker, ExecLinesMarker, ExecMarker, ExecCmdcallLinesMarker, ExecCmdcallMarker,
+        ExecLinesUpdateMarker, ExecLinesMarker, ExecMarker, ExecCmdcallSimpleLinesMarker, ExecCmdcallLinesMarker,
+        ExecCmdcallMarker, ExecCmdcallChainMarker,
         EnvShellMarker, EnvMarker,
         IfMarker, IfElifMarker, IfElseMarker,
         WhileMarker, WhileElseMarker,
@@ -45,7 +46,7 @@ def generate_markers(*args, **kwargs):
         IgnoreMarker,
         PrefixShellMarker,
         AssignCallMarker, AssignInvokerMarker, AssignGotoMarker, AssignTimeoutMarker, AssignRetryMarker,
-        AssignExecMarker, AssignEvalMarker, AssignCmdcallMarker,
+        AssignExecMarker, AssignEvalMarker, AssignCmdcallMarker, AssignCmdcallChainMarker,
         AssignMultiLineRawStrMarker, AssignMultiLineStrMarker, AssignRawStrMarker, AssignStrMarker,
         EmptyMarker,  # must be at the tail
     ]

dekshell/core/markers/base/__init__.py

@@ -9,7 +9,7 @@ from ....utils.cmd import cmd2ak
 
 cmd_call_prefix_simple = '>'
 cmd_call_prefix = cmd_call_prefix_simple * 2
-cmd_call_prefix_chain = cmd_call_prefix_simple * 3
+cmd_call_prefix_chain = '*'
 cmd_call_eval_prefix = '='
 cmd_call_eval_escape = '\\'
 
@@ -104,10 +104,25 @@ class MarkerBase:
     def remove_var(context, name):
         context.remove_variable(name)
 
+    @classmethod
+    def cmd_call_split(cls, s, max_item=None):
+        if max_item == 0:
+            return [], s.lstrip()
+        m = re.match(rf'[ \t\f\r]*({cmd_call_prefix_simple}[{cmd_call_prefix_simple}]?)[ \t\f\r]*([^\W\d][\w.]*)', s)
+        if not m:
+            return [], s.lstrip()
+        rest = s[m.span()[1]:]
+        if max_item is not None:
+            max_item -= 1
+        items, body = cls.cmd_call_split(rest, max_item)
+        items = [(m.group(2), m.group(1) == cmd_call_prefix_simple), *items]
+        return items, body
+
     @classmethod
     def cmd_call_parse(cls, context, s, simple):
+        s = s.lstrip()
         if simple:
-            argv = cls.split_raw(s, 1)
+            argv = re.split(r'[ \t\f\r]+', s, 1)
             args = argv[1:]
             kwargs = {}
         else:
@@ -117,12 +132,6 @@ class MarkerBase:
         func = argv[0]
         return func, args, kwargs
 
-    @classmethod
-    def cmd_call(cls, context, s, simple):
-        func, args, kwargs = cls.cmd_call_parse(context, s, simple)
-        func = cls.eval(context, func)
-        return func(*args, **kwargs)
-
     @classmethod
     def translate_case(cls, context, s, case=None):
         if case is None:
@@ -231,22 +240,32 @@ class MarkerBase:
         __inner_context__ = context  # noqa
         expression = expression.lstrip()
         if expression.startswith(cmd_call_prefix_chain):
-            expression = cls.translate_case(context, expression, translate)
+            max_item = None
             expression = expression[len(cmd_call_prefix_chain):]
-            index = re.search(
-                r"%s|%s" % (re.escape(cmd_call_prefix_simple), re.escape(cmd_call_eval_prefix)), expression).span()[0]
-            funcs = cls.split_raw(expression[:index])
-            value = cls.eval_mixin(context, expression[index:], False)
-            for func in reversed(funcs):
-                if func:
-                    value = cls.eval(context, func)(value)
-            return value
-        elif expression.startswith(cmd_call_prefix):
-            expression = cls.translate_case(context, expression, translate)
-            return cls.cmd_call(context, expression[len(cmd_call_prefix):], False)
-        elif expression.startswith(cmd_call_prefix_simple):
+        else:
+            max_item = 1
+        if expression.startswith(cmd_call_prefix_simple):
             expression = cls.translate_case(context, expression, translate)
-            return cls.cmd_call(context, expression[len(cmd_call_prefix_simple):], True)
+        funcs, body = cls.cmd_call_split(expression, max_item)
+        if funcs:
+            func, simple = funcs[-1]
+            f = cls.eval(context, func)
+            if simple:
+                value = f(body)
+            else:
+                if body:
+                    args, kwargs = cls.cmd2ak(cls.split(body))
+                    args, kwargs = cls.var_map_batch(context, *args, **kwargs)
+                else:
+                    args, kwargs = [], {}
+                value = f(*args, **kwargs)
+            for func, simple in reversed(funcs[:-1]):
+                f = cls.eval(context, func)
+                if simple:
+                    value = f(value)
+                else:
+                    value = f(*value)
+            return value
         else:
             if expression.startswith(cmd_call_eval_prefix):
                 expression = expression[len(cmd_call_eval_prefix):]

dekshell/core/markers/exec.py

@@ -1,4 +1,4 @@
-from .base import MarkerBase, MarkerWithEnd, MarkerNoTranslator, cmd_call_prefix, cmd_call_prefix_simple, \
+from .base import MarkerBase, MarkerWithEnd, MarkerNoTranslator, cmd_call_prefix_simple, \
     cmd_call_prefix_chain
 from .empty import EmptyMarker
 
@@ -30,15 +30,22 @@ class ExecLinesUpdateMarker(MarkerWithEnd, MarkerNoTranslator):
         return []
 
 
-class ExecCmdcallMarker(MarkerBase):
-    tag_head = cmd_call_prefix_simple
-
+class ExecCmdcallMarkerBase(MarkerBase):
     def execute(self, context, command, marker_node, marker_set):
         self.eval_mixin(context, command, False)
 
 
-class ExecCmdcallLinesMarker(MarkerWithEnd):
-    tag_head = f'{cmd_call_prefix_chain}{cmd_call_prefix_simple}'
+class ExecCmdcallMarker(ExecCmdcallMarkerBase):
+    tag_head = cmd_call_prefix_simple
+
+
+class ExecCmdcallChainMarker(ExecCmdcallMarkerBase):
+    tag_head = cmd_call_prefix_chain + cmd_call_prefix_simple
+
+
+class ExecCmdcallSimpleLinesMarker(MarkerWithEnd):
+    command_append_prefix = cmd_call_prefix_simple
+    tag_head = cmd_call_prefix_simple * 3
     cmd_call_marker_cls = ExecCmdcallMarker
     targets_marker_cls = (EmptyMarker,)
 
@@ -49,7 +56,7 @@ class ExecCmdcallLinesMarker(MarkerWithEnd):
             if child.is_type(*self.targets_marker_cls):
                 node = marker_set.node_cls(
                     marker,
-                    self.cmd_call_marker_cls.tag_head + ' ' + child.command,
+                    self.command_append_prefix + ' ' + child.command,
                     child.index,
                     marker_node,
                     child.command
@@ -58,3 +65,8 @@ class ExecCmdcallLinesMarker(MarkerWithEnd):
             else:
                 result.append(child)
         return result
+
+
+class ExecCmdcallLinesMarker(ExecCmdcallSimpleLinesMarker):
+    command_append_prefix = cmd_call_prefix_simple * 2
+    tag_head = cmd_call_prefix_simple * 4

dekshell/core/markers/var.py

@@ -1,12 +1,12 @@
 import re
-from .base import MarkerBase, MarkerWithEnd, MarkerNoTranslator, cmd_call_prefix_simple
+from .base import MarkerBase, MarkerWithEnd, MarkerNoTranslator, cmd_call_prefix_simple, cmd_call_prefix_chain
 from .function import CallMarker
 from .shell import TimeoutMarker, RetryMarker
 from .invoke import MarkerInvokerBase, InvokeMarker, GotoMarker
 
 
 class MarkerAssignBase(MarkerBase):
-    tag_head_re = r"[^\W\d]\w*[ \t\f\r\n]*%s"
+    tag_head_re = r"[^\W\d]\w*[ \t\f\r]*%s"
 
 
 class MarkerAssignValueBase(MarkerAssignBase):
@@ -73,10 +73,16 @@ class AssignExecMarker(MarkerAssignBase, MarkerWithEnd, MarkerNoTranslator):
 
 
 class AssignCmdcallMarker(MarkerAssignValueBase):
-    tag_head_re_args = cmd_call_prefix_simple
+    tag_head_re_args_raw = cmd_call_prefix_simple
+    tag_head_re_args = re.escape(tag_head_re_args_raw)
 
     def get_value(self, context, marker_node, marker_set, expression):
-        return self.eval_mixin(context, cmd_call_prefix_simple + expression, False)
+        return self.eval_mixin(context, self.tag_head_re_args_raw + expression, False)
+
+
+class AssignCmdcallChainMarker(AssignCmdcallMarker):
+    tag_head_re_args_raw = cmd_call_prefix_chain + cmd_call_prefix_simple
+    tag_head_re_args = re.escape(tag_head_re_args_raw)
 
 
 class AssignCallMarker(MarkerAssignValueBase, CallMarker):

{dekshell-0.2.7.dist-info → dekshell-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dekshell
-Version: 0.2.7
+Version: 0.2.9
 Author-Email: sanzenwin <sanzenwin@gmail.com>
 License: MIT
 Requires-Python: >=3.8

{dekshell-0.2.7.dist-info → dekshell-0.2.9.dist-info}/RECORD

@@ -1,15 +1,15 @@
-dekshell-0.2.7.dist-info/METADATA,sha256=6KZynQkCledNS27CLnsBIOlxnCBf2QDi9JX8HvTSz7M,572
-dekshell-0.2.7.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-dekshell-0.2.7.dist-info/entry_points.txt,sha256=d-kbfULiUTZWIBBsrQF3J_-wESncF-4K2rwHT08grlI,75
+dekshell-0.2.9.dist-info/METADATA,sha256=jQr2EOL1fMZ89s4wFbcoj3FgePYiLdUMwAk6uJYwEt8,572
+dekshell-0.2.9.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+dekshell-0.2.9.dist-info/entry_points.txt,sha256=d-kbfULiUTZWIBBsrQF3J_-wESncF-4K2rwHT08grlI,75
 dekshell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dekshell/click/__entry__.py,sha256=CMuxUzXoEe4TcHFZwv-MNFwHnu1HSZCDpXFpqQ814uM,42
 dekshell/click/__init__.py,sha256=rFzB_exzPQaIcbbms5PdHAU3NGdl2MBaYbq9v7g7BOI,1870
 dekshell/core/__init__.py,sha256=ZAJtMPFCVClHCPkoP5rmTq1vVLLMQRa8MP-QJXLBgNM,5337
 dekshell/core/contexts/__init__.py,sha256=ynsfv37azOKfI2UKd0iPl2M6iBW-k5cb1BqSLOWuJpI,482
-dekshell/core/contexts/methods.py,sha256=EXIGyrywib0ZBfxXkxFVUGVpPHAusVb-x3grDq06AFM,7164
+dekshell/core/contexts/methods.py,sha256=tOfgG-t4oh0-BjKgK9uk34bgZ1Zmp1AM-kpmB78jM_w,7485
 dekshell/core/contexts/properties.py,sha256=6rEQjExvAmZ69OfhPLLIQwnynRzAjX0nIbcBuZ5yuzM,2314
-dekshell/core/markers/__init__.py,sha256=O5N5rURhTW0sZvnVVvtTX_w53zHYjb6KfcUShY46YmM,1762
-dekshell/core/markers/base/__init__.py,sha256=AQYr7H0F06izu5jhu8EgLYJ2ERzJh0eS38h9gtl6OZs,13067
+dekshell/core/markers/__init__.py,sha256=BRp4z_IkbwV7xQr-5wugd7GCCyGp8k4z_p8qRgVjKxw,1850
+dekshell/core/markers/base/__init__.py,sha256=To25RVX3h1MluV3OmktVMw2bigex79GjWHUPWtIlczI,13433
 dekshell/core/markers/base/core.py,sha256=99aRcSXohpfTW5OSO1CUyvnptr21ueKjizCk-FK6rcM,10830
 dekshell/core/markers/commands.py,sha256=4frmkrTa7-7ZYWztXEmYprbWRkoCIuJuiy28pBGcz8A,987
 dekshell/core/markers/comment.py,sha256=28iccgLs_0bRdXLhHyQR2I_kzlWdeMSqqNUFW-2vkes,818
@@ -17,7 +17,7 @@ dekshell/core/markers/define.py,sha256=LpMSfz9ziXq2aFJ6oMpUFFo93TpBx7GxKYNzCeht4
 dekshell/core/markers/echo.py,sha256=1H61qQbY9tZnrKsdTI_shTmDY5ZaSGipDuBSRptAuZw,660
 dekshell/core/markers/empty.py,sha256=xMYoZIxn7Tt1RjFtLhJERisra8jqsGjJ0G7OID0oD3o,776
 dekshell/core/markers/env.py,sha256=6ZtiMNdKFbGR_DBjG6C7A8L_lJsiymN5Y5AbnTzyMrE,1158
-dekshell/core/markers/exec.py,sha256=zT3mI7bBSgOpBICCjeb-vhAWbra9i8rABuUAVNnuMns,2032
+dekshell/core/markers/exec.py,sha256=3vQc-jH7toU_89rNFTipowr0ZrNn79pqmNbYuHxCPDI,2370
 dekshell/core/markers/for_.py,sha256=uNOEwyDsjffttEBtUYasdlj7FP_sGnQzWuTV5d5esHY,2142
 dekshell/core/markers/function.py,sha256=RPh56_rYbWXCTkNPiJu8zNEj-OE6txKgEw7Im9fBM3c,3592
 dekshell/core/markers/if_.py,sha256=bBG1fDCZd5alE5FSm0QhgHl6ZJn5awiqZAXnPnO-thc,1112
@@ -26,7 +26,7 @@ dekshell/core/markers/invoke.py,sha256=To_U_FscoCbf7osvjC5MQTk2gjH5Mu_-06Kb22YQP
 dekshell/core/markers/pip_.py,sha256=mxXa_oqcrk4Qz-HvhlkMoirsl-SbM9Fz610leNqP1o0,832
 dekshell/core/markers/redirect.py,sha256=ZMqHnntzJabgqlNXU7Xj1GcYNlfwOyUsZ3rrJTd1yXk,1589
 dekshell/core/markers/shell.py,sha256=6Gy59gjI-BFIQadwGxApD6w3LndIRMao_4D9pMpB5Hk,1906
-dekshell/core/markers/var.py,sha256=rj-gIdu-V8RQigmZ87skCLS-H103ndGgh3zrf3gQEZw,4571
+dekshell/core/markers/var.py,sha256=O-lpWhnLOkKMwF1jR0rVwyHLa8YUWwacCs80wFzwfdQ,4838
 dekshell/core/markers/while_.py,sha256=e7lI5jsIM-qxNHMY_wUyCMvvesdXQibR5Ez2jNO9csc,1195
 dekshell/core/plugin/__init__.py,sha256=jAB_KnnHJsyJR_zIfBU_HNLngyhcyyqVv05PdlNZtF8,428
 dekshell/core/redirect.py,sha256=6YCJpG0TkQ4WMt7LBtDD_W1T-C-QkLtGRQw0S60qe54,1058
@@ -36,4 +36,4 @@ dekshell/utils/cmd.py,sha256=akykpgPCovXpvavKF28G-eNM3mOZ4TY0LlSWGozDp34,1204
 dekshell/utils/pkg.py,sha256=TgYqRqawoJfjkxt6UAHnp9ttmpjuHiWRFbqxADOS1VE,1337
 dekshell/utils/serializer.py,sha256=aIdF2Wzo-qHmIshv46jn1XD0X66vQ1JFdU-g3ZFbH2w,386
 dekshell/utils/shell.py,sha256=0NoA2-SOOMinbmZZipwzL-npBbzPOdWEfdPVYqq5G5g,92
-dekshell-0.2.7.dist-info/RECORD,,
+dekshell-0.2.9.dist-info/RECORD,,