danielutils 0.9.80__tar.gz → 0.9.82__tar.gz

{danielutils-0.9.80/danielutils.egg-info → danielutils-0.9.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: danielutils
-Version: 0.9.80
+Version: 0.9.82
 Summary: A python utils library for things I find useful
 Author-email: danielnachumdev <danielnachumdev@gmail.com>
 License: MIT License
@@ -41,7 +41,7 @@ License-File: LICENSE
 [![Python 3.10.11](https://img.shields.io/badge/python-3.10.11-blue.svg)](https://www.python.org/downloads/release/python-31011/)
 [![gitleaks](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml)
 [![CodeQL](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql)
-# danielutils v=0.9.80
+# danielutils v=0.9.82
 A utils library for things that I find useful for my coding workflow.\
 Feel free to use and / or contribute / improve my code :)
 

{danielutils-0.9.80 → danielutils-0.9.82}/README.md

@@ -4,7 +4,7 @@
 [![Python 3.10.11](https://img.shields.io/badge/python-3.10.11-blue.svg)](https://www.python.org/downloads/release/python-31011/)
 [![gitleaks](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml)
 [![CodeQL](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql)
-# danielutils v=0.9.80
+# danielutils v=0.9.82
 A utils library for things that I find useful for my coding workflow.\
 Feel free to use and / or contribute / improve my code :)
 

{danielutils-0.9.80 → danielutils-0.9.82}/danielutils/__init__.py

@@ -39,4 +39,5 @@ from .print_ import *
 from .metaclasses import *
 from .generators import *
 from .university import *
-from .mock_ import *
+from .mock_ import *
+from .context_managers import *

danielutils-0.9.82/danielutils/context_managers/__init__.py

@@ -0,0 +1,2 @@
+from .temporary_file import *
+from .attr_context import *

danielutils-0.9.82/danielutils/context_managers/attr_context.py

@@ -0,0 +1,25 @@
+class AttrContext:
+    def __init__(self, obj: object, attr: str, new_value: object, *, nonexistent_is_error: bool = True) -> None:
+        self.obj = obj
+        self.attr = attr
+        self.new_value = new_value
+        self.old_value = None
+        self._has_attr: bool = hasattr(self.obj, self.attr)
+        if nonexistent_is_error and not self._has_attr:
+            raise RuntimeError(f"Nonexistent attribute '{self.attr}' in '{self.obj}'")
+
+    def __enter__(self) -> 'AttrContext':
+        self.old_value = getattr(self.obj, self.attr, None)
+        setattr(self.obj, self.attr, self.new_value)
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._has_attr:
+            setattr(self.obj, self.attr, self.old_value)
+        else:
+            delattr(self.obj, self.attr)
+
+
+__all__ = [
+    'AttrContext'
+]

danielutils-0.9.82/danielutils/context_managers/temporary_file.py

@@ -0,0 +1,46 @@
+from ..io_ import file_exists, delete_file
+import atexit
+
+
+class TemporaryFile:
+    _instances: set['TemporaryFile'] = set()
+
+    def __init__(self, path: str):
+        if file_exists(path):
+            raise RuntimeError(f"Can't create a temporary file if file '{path}' already exists.")
+        self.path = path
+        TemporaryFile._instances.add(self)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    @atexit.register
+    @staticmethod
+    def _global_close():
+        for inst in TemporaryFile._instances:
+            inst.close()
+
+    def close(self) -> None:
+        delete_file(self.path)
+
+    def read(self) -> list[str]:
+        if not file_exists(self.path):
+            return []
+        with open(self.path, 'r') as f:
+            return f.readlines()
+
+    def write(self, lines: list[str]) -> None:
+        with open(self.path, 'a') as f:
+            f.writelines(lines)
+
+    def clear(self):
+        with open(self.path, 'w') as _:
+            pass
+
+
+__all__ = [
+    'TemporaryFile'
+]

{danielutils-0.9.80 → danielutils-0.9.82}/danielutils/system/__init__.py

@@ -1,2 +1,3 @@
 from .independent import *
 from .windows import *
+from .layered_command import *

danielutils-0.9.82/danielutils/system/layered_command.py

@@ -0,0 +1,110 @@
+import sys, os
+from typing import Optional
+from ..context_managers import TemporaryFile
+import random
+
+
+class LayeredCommand:
+    class_flush_stdout: bool = True
+    class_flush_stderr: bool = True
+    class_raise_on_fail: bool = True
+    class_verbose: bool = False
+    _class_prev_instance: Optional['LayeredCommand'] = None
+    _id: int = 0
+
+    def __init__(
+            self,
+            command: Optional[str] = None,
+            *,
+            prev_instance: Optional['LayeredCommand'] = None,
+            instance_flush_stdout: Optional[bool] = None,
+            instance_flush_stderr: Optional[bool] = None,
+            instance_raise_on_fail: Optional[bool] = None,
+            instance_verbose: Optional[bool] = None
+    ):
+        self._command = command if command is not None else ""
+        self._instance_flush_stdout = instance_flush_stdout
+        self._instance_flush_stderr = instance_flush_stderr
+        self._instance_raise_on_fail = instance_raise_on_fail
+        self._instance_verbose = instance_verbose
+        self._prev_instance = prev_instance if prev_instance is not None else LayeredCommand._class_prev_instance
+        self._cur_class_prev_instance = LayeredCommand._class_prev_instance
+        LayeredCommand._class_prev_instance = self
+        self._executor = os.system
+        self._has_entered: bool = False
+
+    def __enter__(self):
+        self._open()
+        return self
+
+    def _open(self) -> None:
+        self._has_entered = True
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._prev_instance is self._cur_class_prev_instance:
+            LayeredCommand._class_prev_instance = self._prev_instance
+
+    def _build_command(self, *commands: str) -> str:
+        res = ""
+        if self._prev_instance is not None:
+            prev = self._prev_instance._build_command()
+            res += f"{prev} & " if prev != "" else ""
+        if self._command != "":
+            return res + " & ".join([self._command, *commands])
+        return res + " & ".join(commands)
+
+    def _error(self, predicate: bool, command: str, code: int, command_verbose: Optional[bool]) -> None:
+        if predicate:
+            verbose = self._merge_values(command_verbose, self._instance_verbose, LayeredCommand.class_verbose)
+            if verbose:
+                raise RuntimeError(f"command '{command}' failed with exit code {code}")
+            sys.exit(1)
+
+    @staticmethod
+    def _merge_values(command: Optional[bool], instance: Optional[bool], class_: bool) -> bool:
+        return command if command is not None else (instance if instance is not None else class_)
+
+    def execute(
+            self,
+            *commands: str,
+            command_flush_stdout: Optional[bool] = None,
+            command_flush_stderr: Optional[bool] = None,
+            command_raise_on_fail: Optional[bool] = None,
+            command_verbose: Optional[bool] = None
+    ) -> tuple[int, list[str], list[str]]:
+        if not self._has_entered:
+            raise RuntimeError(
+                "LayeredCommand must be used with a context manager. Use as: `with LayeredCommand(...) as l1:`")
+        flush_stdout = self._merge_values(command_flush_stdout, self._instance_flush_stdout, self.class_flush_stdout)
+        flush_stderr = self._merge_values(command_flush_stderr, self._instance_flush_stderr, self.class_flush_stderr)
+        raise_on_fail = self._merge_values(command_raise_on_fail, self._instance_raise_on_fail,
+                                           self.class_raise_on_fail)
+
+        command = self._build_command(*commands)
+        if flush_stdout and flush_stderr:
+            code = self._executor(command)
+            self._error(raise_on_fail and code != 0, command, code, command_verbose)
+            return code, [], []
+
+        letters = "abcdefghijklmnopqrstuvwxyz"
+        temp_name = "".join(random.choices(letters, k=50)) + f"_{LayeredCommand._id}"
+        stdout_path = f"./{temp_name}_stdout.log"
+        stderr_path = f"./{temp_name}_stderr.log"
+        LayeredCommand._id += 1
+        with TemporaryFile(stdout_path) as stdout:
+            with TemporaryFile(stderr_path) as stderr:
+                if not flush_stdout:
+                    command += f" > {stdout_path}"
+                if not flush_stderr:
+                    command += f" 2> {stderr_path}"
+                code = self._executor(command)
+                self._error(raise_on_fail and code != 0, command, code, command_verbose)
+                return code, stdout.read(), stderr.read()
+
+    def __call__(self, *args, **kwargs) -> tuple[int, list[str], list[str]]:
+        return self.execute(*args, **kwargs)
+
+
+__all__ = [
+    'LayeredCommand'
+]

{danielutils-0.9.80 → danielutils-0.9.82/danielutils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: danielutils
-Version: 0.9.80
+Version: 0.9.82
 Summary: A python utils library for things I find useful
 Author-email: danielnachumdev <danielnachumdev@gmail.com>
 License: MIT License
@@ -41,7 +41,7 @@ License-File: LICENSE
 [![Python 3.10.11](https://img.shields.io/badge/python-3.10.11-blue.svg)](https://www.python.org/downloads/release/python-31011/)
 [![gitleaks](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/gitleaks.yml)
 [![CodeQL](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql/badge.svg)](https://github.com/danielnachumdev/danielutils/actions/workflows/github-code-scanning/codeql)
-# danielutils v=0.9.80
+# danielutils v=0.9.82
 A utils library for things that I find useful for my coding workflow.\
 Feel free to use and / or contribute / improve my code :)
 

{danielutils-0.9.80 → danielutils-0.9.82}/danielutils.egg-info/SOURCES.txt

@@ -42,6 +42,9 @@ danielutils/better_builtins/typed_builtins/tdict.py
 danielutils/better_builtins/typed_builtins/tlist.py
 danielutils/better_builtins/typed_builtins/tset.py
 danielutils/better_builtins/typed_builtins/ttuple.py
+danielutils/context_managers/__init__.py
+danielutils/context_managers/attr_context.py
+danielutils/context_managers/temporary_file.py
 danielutils/conversions/__init__.py
 danielutils/conversions/main_conversions.py
 danielutils/conversions/specialized_conversions/__init__.py
@@ -142,6 +145,7 @@ danielutils/snippets/__init__.py
 danielutils/snippets/try_get.py
 danielutils/system/__init__.py
 danielutils/system/independent.py
+danielutils/system/layered_command.py
 danielutils/system/windows/__init__.py
 danielutils/system/windows/win32_ctime.py
 danielutils/system/windows/windows.py

{danielutils-0.9.80 → danielutils-0.9.82}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "danielutils"
-version = "0.9.80"
+version = "0.9.82"
 authors = [
     { name = "danielnachumdev", email = "danielnachumdev@gmail.com" },
 ]