dekshell 0.2.4__tar.gz → 0.2.5__tar.gz

{dekshell-0.2.4 → dekshell-0.2.5}/PKG-INFO

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

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/__init__.py

@@ -87,7 +87,7 @@ def shell_command_batch_core(
     def shell_exec(filepath, c=None):
         return shell_command_file(
             filepath,
-            context=(context or {}) | (c or {}),
+            context=c or {},
             marker=marker,
             ms_names=ms_names,
             marker_set_cls=marker_set_cls,

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/contexts/methods.py

@@ -3,6 +3,8 @@ import shutil
 import sys
 import tempfile
 import getpass
+from functools import reduce
+from itertools import chain
 from pathlib import Path
 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, \
@@ -138,6 +140,8 @@ path_common_methods = {
 }
 
 default_methods = {
+    'reduce': reduce,
+    'chain': chain,
     'xeval': _xeval,
     'echo': lambda *x, **y: print(*x, **dict(flush=True) | y),
     'echos': lambda *x, **y: print(*x, **dict(end='', flush=True) | y),

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/markers/__init__.py

@@ -12,10 +12,10 @@ from .function import *
 from .exec import *
 from .echo import *
 from .pip_ import *
-from .shell import *
+from .commands import *
 from .redirect import *
 from .input import *
-from .timout import *
+from .shell import *
 from .empty import *
 
 
@@ -39,12 +39,12 @@ def generate_markers(*args, **kwargs):
         EndMarker, BreakMarker, ContinueMarker,
         CommentMultiLineMarker, CommentMarker, CommentShebangMarker, CommentConfigMarker,
         PipMarker,
-        ShellMarker,
+        CommandsMarker,
         RedirectMarker, ShiftMarker,
-        TimeoutMarker,
+        TimeoutMarker, RetryMarker,
         IgnoreMarker,
         PrefixShellMarker,
-        AssignCallMarker, AssignInvokerMarker, AssignGotoMarker, AssignTimeoutMarker,
+        AssignCallMarker, AssignInvokerMarker, AssignGotoMarker, AssignTimeoutMarker, AssignRetryMarker,
         AssignExecMarker, AssignEvalMarker, AssignCmdcallMarker,
         AssignMultiLineRawStrMarker, AssignMultiLineStrMarker, AssignRawStrMarker, AssignStrMarker,
         EmptyMarker,  # must be at the tail

dekshell-0.2.4/dekshell/core/markers/shell.py → dekshell-0.2.5/dekshell/core/markers/commands.py

@@ -2,7 +2,7 @@ from .base import MarkerWithEnd
 from .empty import EmptyMarker
 
 
-class ShellMarker(MarkerWithEnd):
+class CommandsMarker(MarkerWithEnd):
     tag_head = "@@"
     target_marker_cls = EmptyMarker
 

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/markers/invoke.py

@@ -1,6 +1,6 @@
 import os
 from dektools.file import normal_path
-from ...utils.cmd import pack_context
+from ...utils.cmd import pack_context_full
 from .base import MarkerBase
 from .base.core import get_inner_vars
 
@@ -18,7 +18,7 @@ class MarkerInvokerBase(MarkerBase):
     def execute_file(cls, marker_set, filepath, args, kwargs):
         if marker_set is None:
             marker_set = get_inner_vars('__inner_marker_set__')
-        return cls._run_file(marker_set, normal_path(filepath), pack_context(args, kwargs))
+        return cls._run_file(marker_set, normal_path(filepath), pack_context_full(args, kwargs))
 
     @classmethod
     def _run_file(cls, marker_set, filepath, attrs):

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/markers/redirect.py

@@ -28,7 +28,7 @@ class RedirectMarker(MarkerRedirect):
         shell_properties = make_shell_properties(path_shell)
         if shell_properties['shell'] != current_shell:
             fp = self.eval(context, "fp")
-            fpp = os.path.dirname(fp).relace('/', os.sep)
+            fpp = os.path.dirname(fp).replace('/', os.sep)
             shell = shell_properties['sh']['rfc' if os.getcwd() == fpp else 'rf']
             args, kwargs = self.eval(context, f'({key_args}, {key_kwargs})')
             argv = ak2cmd(args, kwargs)

dekshell-0.2.5/dekshell/core/markers/shell.py

@@ -0,0 +1,56 @@
+import subprocess
+from dektools.shell import shell_timeout, shell_wrapper, shell_retry
+from .empty import MarkerShell, ShellCommand
+
+
+class TimeoutShellCommand(ShellCommand):
+    def shell(self, command, timeout=None, env=None, **kwargs):
+        try:
+            if timeout > 0:
+                return shell_timeout(command, timeout, env=env)
+            else:
+                return shell_wrapper(command, env=env)
+        except subprocess.SubprocessError as e:
+            return e
+
+
+class TimeoutMarker(MarkerShell):
+    tag_head = "timeout"
+
+    shell_cls = TimeoutShellCommand
+
+    def execute(self, context, command, marker_node, marker_set):
+        _, timeout, command = self.split_raw(command, 2)
+        if command:
+            self.execute_core_timeout(context, command, marker_node, marker_set, timeout)
+
+    @classmethod
+    def execute_core_timeout(cls, context, command, marker_node, marker_set, timeout):
+        return cls.execute_core(context, command, marker_node, marker_set, dict(timeout=int(float(timeout))))
+
+
+class RetryShellCommand(ShellCommand):
+    def shell(self, command, times=None, env=None, **kwargs):
+        try:
+            return shell_retry(command=command, times=times, env=env)
+        except subprocess.SubprocessError as e:
+            return e
+
+
+class RetryMarker(MarkerShell):
+    tag_head = "retry"
+
+    shell_cls = RetryShellCommand
+
+    def execute(self, context, command, marker_node, marker_set):
+        _, times, command = self.split_raw(command, 2)
+        if command:
+            self.execute_core_retry(context, command, marker_node, marker_set, times)
+
+    @classmethod
+    def execute_core_retry(cls, context, command, marker_node, marker_set, times):
+        try:
+            times = int(float(times))
+        except ValueError:
+            times = None
+        return cls.execute_core(context, command, marker_node, marker_set, dict(times=times))

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/core/markers/var.py

@@ -1,7 +1,7 @@
 import re
 from .base import MarkerBase, MarkerWithEnd, MarkerNoTranslator, cmd_call_prefix_simple
 from .function import CallMarker
-from .timout import TimeoutMarker
+from .shell import TimeoutMarker, RetryMarker
 from .invoke import MarkerInvokerBase, InvokeMarker, GotoMarker
 
 
@@ -114,6 +114,15 @@ class AssignTimeoutMarker(MarkerAssignValueBase, TimeoutMarker):
         return self.execute_core_timeout(context, expression, marker_node, marker_set, timeout)
 
 
+class AssignRetryMarker(MarkerAssignValueBase, RetryMarker):
+    tag_head = None
+    tag_head_re_args = r'retry[ \t\f\r\n]+'
+
+    def get_value(self, context, marker_node, marker_set, expression):
+        times, expression = self.split_raw(expression, 1)
+        return self.execute_core_retry(context, expression, marker_node, marker_set, times)
+
+
 class DelVarMarker(MarkerBase):
     tag_head = "del"
 

{dekshell-0.2.4 → dekshell-0.2.5}/dekshell/utils/cmd.py

@@ -34,7 +34,7 @@ def ak2cmd(args, kwargs=None):
 
 
 def pack_context(args, kwargs):
-    return {f'{key_arg}{i}': arg for i, arg in enumerate(args)} | {key_args: args, key_kwargs: kwargs} | kwargs
+    return {f'{key_arg}{i}': arg for i, arg in enumerate(args)} | {key_args: tuple(args), key_kwargs: kwargs} | kwargs
 
 
 def pack_context_argv():

{dekshell-0.2.4 → dekshell-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dekshell"
-version = "0.2.4"
+version = "0.2.5"
 description = ""
 authors = [
     { name = "sanzenwin", email = "sanzenwin@gmail.com" },

dekshell-0.2.4/dekshell/core/markers/timout.py

@@ -1,26 +0,0 @@
-import subprocess
-from dektools.shell import shell_timeout
-from .empty import MarkerShell, ShellCommand
-
-
-class TimeoutShellCommand(ShellCommand):
-    def shell(self, command, timeout=None, env=None, **kwargs):
-        try:
-            return shell_timeout(command, timeout, env=env, check=True)
-        except subprocess.CalledProcessError as e:
-            return e
-
-
-class TimeoutMarker(MarkerShell):
-    tag_head = "timeout"
-
-    shell_cls = TimeoutShellCommand
-
-    def execute(self, context, command, marker_node, marker_set):
-        _, timeout, command = self.split_raw(command, 2)
-        if command:
-            self.execute_core_timeout(context, command, marker_node, marker_set, timeout)
-
-    @classmethod
-    def execute_core_timeout(cls, context, command, marker_node, marker_set, timeout):
-        return cls.execute_core(context, command, marker_node, marker_set, dict(timeout=int(float(timeout))))