🗓 Dec 15, 2024 8:18:53 PM

✨ strip_non_printable method
🐙 improve CHEPY_PLUGINS env handling

TODO

@@ -9,11 +9,10 @@ New ideas:
   ☐ save registers https://t.co/pBNRufibY8?amp=1 
   ☐ cbor encode/decode https://github.com/agronholm/cbor2 (plugin)
   ☐ fuzzy search
-  ☐ pgp, generate, encrypt, decrypt, verify
+  ☐ 🚧 pgp, generate, encrypt, decrypt, verify
   ☐ swap little and big endian
   ☐ ignore error method
   ☐ swap bytes
-  ☐ @high disable plugins from code
   ☐ homophonic decoder
   ☐ append method for core to add data to the state
   ☐ qr create 
@@ -48,6 +47,7 @@ Misc:
   ☐ cyberchef recipe to chepy recipe converter
 
 Archive:
+  ✔ @high disable plugins from code
   ✔ 🐙 update config to use envars for plugins
   ✔ register support in callstack
   ✔ ecb no padding in aes/des

chepy/config.py

@@ -60,9 +60,11 @@ def __init__(self):
             self.__get_conf_value("false", "EnablePlugins", "Plugins")
         )
 
-        self.enable_plugins = (
-            os.environ.get("CHEPY_PLUGINS", "false") == "true"
-        ) or self.enable_plugins
+        plugin_envar = os.environ.get("CHEPY_PLUGINS", "")
+        if plugin_envar == "true":
+            self.enable_plugins = True
+        elif plugin_envar != "":
+            self.enable_plugins = False
 
         if self.enable_plugins:
             if plugin_path != "None":

chepy/modules/utils.py

@@ -495,6 +495,19 @@ def strip(self, pattern: str, ignore_case=True) -> UtilsT:
         self.state = re.sub(pattern, "", self._convert_to_str(), flags=flags)
         return self
 
+    @ChepyDecorators.call_stack
+    def strip_non_printable(self) -> UtilsT:
+        """String non printable characters
+
+        Returns:
+            Chepy: The Chepy object.
+        """
+        data = self._convert_to_bytes()
+        printable_set = set(range(32, 127)) | {9, 10, 13, 12}
+        # Filter out non-printable characters
+        self.state = bytes(b for b in data if b in printable_set)
+        return self
+
     @ChepyDecorators.call_stack
     def find_replace(self, pattern: str, repl: str, ignore_case=True) -> UtilsT:
         """Replace matched pattern with repln

chepy/modules/utils.pyi

@@ -28,6 +28,7 @@ class Utils(ChepyCore):
     def slice(self: UtilsT, start: int=..., end: int=...) -> UtilsT: ...
     def strip_ansi(self: UtilsT) -> UtilsT: ...
     def strip(self: UtilsT, pattern: str, ignore_case: bool=...) -> UtilsT: ...
+    def strip_non_printable(self: UtilsT) -> UtilsT: ...
     def find_replace(self: UtilsT, pattern: str, repl: str, ignore_case: bool=...) -> UtilsT: ...
     def escape_string(self: UtilsT) -> UtilsT: ...
     def unescape_string(self: UtilsT) -> UtilsT: ...

tests/test_utils.py

@@ -181,6 +181,10 @@ def test_strip():
     assert Chepy("some some data").strip(r"some\s").o == b"data"
 
 
+def test_string_non_printable():
+    assert Chepy("Hello\x00W\xc1orld\x1b!").strip_non_printable().o == b"HelloWorld!"
+
+
 def test_find_replace():
     assert Chepy("some some data").find_replace(r"some\s", "data").o == b"datadatadata"