osbot-utils 1.21.0__py3-none-any.whl → 1.22.0__py3-none-any.whl

osbot_utils/helpers/Local_Cache.py

@@ -1,7 +1,8 @@
 from osbot_utils.utils.Misc import list_set
 from osbot_utils.utils.Dev import pprint
 from osbot_utils.decorators.methods.cache_on_self import cache_on_self
-from osbot_utils.utils.Files    import current_temp_folder, path_combine, create_folder, safe_file_name, file_exists, file_delete
+from osbot_utils.utils.Files import current_temp_folder, path_combine, create_folder, safe_file_name, file_exists, \
+    file_delete, file_size
 from osbot_utils.utils.Json import json_save_file, json_load_file
 
 
@@ -32,6 +33,9 @@ class Local_Cache:
     def cache_exists(self):
         return file_exists(self.path_cache_file())
 
+    def cache_file_size(self):
+        return file_size(self.path_cache_file())
+
     def create(self):
         if not self.cache_exists():
             self.save()

osbot_utils/helpers/ssh/SSH.py

@@ -1,11 +1,13 @@
 from osbot_utils.base_classes.Kwargs_To_Self        import Kwargs_To_Self
+from osbot_utils.base_classes.Type_Safe import Type_Safe
 from osbot_utils.decorators.methods.cache_on_self   import cache_on_self
 from osbot_utils.helpers.ssh.SCP                    import SCP
 from osbot_utils.helpers.ssh.SSH__Execute           import SSH__Execute
 from osbot_utils.helpers.ssh.SSH__Linux             import SSH__Linux
+from osbot_utils.helpers.ssh.SSH__Linux__Amazon import SSH__Linux__Amazon
 from osbot_utils.helpers.ssh.SSH__Python            import SSH__Python
 
-class SSH(Kwargs_To_Self):
+class SSH(Type_Safe):       # todo: add ip_address to global vars here, and when that is done, add the wait_for_ssh method (that exists in EC2_Instance)
 
     def setup(self):
         self.ssh_execute().setup()
@@ -25,6 +27,11 @@ class SSH(Kwargs_To_Self):
     def ssh_linux(self):
         return SSH__Linux(ssh_execute = self.ssh_execute())
 
+    @cache_on_self
+    def ssh_linux_amazon(self):
+        return SSH__Linux__Amazon(ssh_execute=self.ssh_execute())
+
     @cache_on_self
     def ssh_python(self):
-        return SSH__Python(ssh_execute = self.ssh_execute(), ssh_linux = self.ssh_linux())
+        return SSH__Python(ssh_execute = self.ssh_execute(), ssh_linux = self.ssh_linux())
+

osbot_utils/helpers/ssh/SSH__Execute.py

@@ -21,6 +21,7 @@ class SSH__Execute(Type_Safe):
     ssh_key_file      : str
     ssh_key_user      : str
     strict_host_check : bool = False
+    print_after_exec  : bool = False
 
     # execution & other commands # todo refactor into separate class
     def exec(self, command):
@@ -29,7 +30,7 @@ class SSH__Execute(Type_Safe):
     def exec__print(self, command):
         result = self.execute_command__return_stdout(command)
         self.print_header_for_command(command)
-        print(result)
+        self.print_status__stderr__stdout(result)
         return result
 
     def execute_command(self, command):
@@ -38,13 +39,15 @@ class SSH__Execute(Type_Safe):
             with capture_duration() as duration:
                 result          = start_process("ssh", ssh_args)                                 # execute command using subprocess.run(...)
             result['duration']  = duration.data()
+            if self.print_after_exec:
+                self.print_status__stderr__stdout(result)
             return result
         return status_error(error='in execute_command not all required vars were setup')
 
     def execute_command__print(self, command):
         self.print_header_for_command(command)
         result = self.execute_command(command)
-        pprint(result)
+        self.print_status__stderr__stdout(result)
         return result
 
     def execute_ssh_args(self):
@@ -142,8 +145,14 @@ class SSH__Execute(Type_Safe):
     #     pprint(self.ls(path))
     #     return self
 
-    def print_exec(self, command=''):
-        return self.exec__print(command)
+    def print_status__stderr__stdout(self, result):
+        print()
+        print( '┌──────────────────────────────────────────')
+        print(f'├ command: {result.get("command")        }')
+        print(f'│ status : {result.get("status" ).strip()}')
+        print(f'│ stderr : {result.get("stderr" ).strip()}')
+        print(f'│ stdout : {result.get("stdout" ).strip()}')
+        return self
 
     def print_header_for_command(self, command):
         print('\n')

osbot_utils/helpers/ssh/SSH__Linux__Amazon.py

@@ -0,0 +1,14 @@
+from osbot_utils.helpers.ssh.SSH__Linux import SSH__Linux
+
+
+class SSH__Linux__Amazon(SSH__Linux):
+
+    def install_python3(self):
+        execute_commands = ('sudo yum install -y python3.11                 && '  
+                            'curl -O https://bootstrap.pypa.io/get-pip.py   && '
+                            'sudo python3.11 get-pip.py'                       )
+
+        return self.ssh_execute.execute_command__return_stdout(execute_commands)
+
+    def pip_install(self, package_name):
+        return self.ssh_execute.execute_command__return_stdout(f'pip3.11 install {package_name}')

osbot_utils/utils/Misc.py

@@ -384,6 +384,9 @@ def random_text(prefix:str=None,length:int=12, lowercase=False):
 def random_uuid():
     return str(uuid.uuid4())
 
+def random_uuid_short():
+    return str(uuid.uuid4())[0:6]
+
 def remove(target_string, string_to_remove):                        # todo: refactor to str_*
     return replace(target_string, string_to_remove, '')
 

osbot_utils/version

@@ -1 +1 @@
-v1.21.0
+v1.22.0

{osbot_utils-1.21.0.dist-info → osbot_utils-1.22.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: osbot_utils
-Version: 1.21.0
+Version: 1.22.0
 Summary: OWASP Security Bot - Utils
 Home-page: https://github.com/owasp-sbot/OSBot-Utils
 License: MIT
@@ -22,7 +22,7 @@ Description-Content-Type: text/markdown
 
 Powerful Python util methods and classes that simplify common apis and tasks.
 
-![Current Release](https://img.shields.io/badge/release-v1.21.0-blue)
+![Current Release](https://img.shields.io/badge/release-v1.22.0-blue)
 [![codecov](https://codecov.io/gh/owasp-sbot/OSBot-Utils/graph/badge.svg?token=GNVW0COX1N)](https://codecov.io/gh/owasp-sbot/OSBot-Utils)
 
 

{osbot_utils-1.21.0.dist-info → osbot_utils-1.22.0.dist-info}/RECORD

@@ -57,7 +57,7 @@ osbot_utils/graphs/mgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 osbot_utils/helpers/CFormat.py,sha256=1_XvqGwgU6qC97MbzcKF0o7s9mCXpU5Kq9Yf-1ixUwY,6808
 osbot_utils/helpers/CPrint.py,sha256=ztKPNmT8BGxeyPXSQKRs63PqqbgxKDz_BiZmzFMup9g,1413
 osbot_utils/helpers/Dict_To_Attr.py,sha256=NdhXl5mJH7-NaBk213amzc5Nfy3tJgW-N_uYIRE4hoc,208
-osbot_utils/helpers/Local_Cache.py,sha256=nxyorvXbXyWdlKkLeSeWmUhw1gE4SpOdso919IKfIE4,3046
+osbot_utils/helpers/Local_Cache.py,sha256=0JZZX3fFImcwtbBvxAQl-EbBegSNJRhRMYF6ovTH6zY,3141
 osbot_utils/helpers/Local_Caches.py,sha256=HvuP5CURyVm_fVvJX-S4dml2bhRauzgA3be237yTaeY,1814
 osbot_utils/helpers/Print_Table.py,sha256=LEXbyqGg_6WSraI4cob4bNNSu18ddqvALp1zGK7bPhs,19126
 osbot_utils/helpers/Python_Audit.py,sha256=shpZlluJwqJBAlad6xN01FkgC1TsQ48RLvR5ZjmrKa4,1539
@@ -217,11 +217,12 @@ osbot_utils/helpers/sqlite/tables/Sqlite__Table__Files.py,sha256=ZlhTqroHd9T8vqN
 osbot_utils/helpers/sqlite/tables/Sqlite__Table__Nodes.py,sha256=GT8h3wD4hGvEtqQuBs0sBbcu2ydktRHTi95PEL2ffHQ,1721
 osbot_utils/helpers/sqlite/tables/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 osbot_utils/helpers/ssh/SCP.py,sha256=9PgJbyWKfxJj00Ijaj7o6ffxPXuNoureb6JlHMPbHww,3330
-osbot_utils/helpers/ssh/SSH.py,sha256=El2Op9b_pcr7msolwbbtkW74R2f_BJN_jdf3SQk_gk8,1081
+osbot_utils/helpers/ssh/SSH.py,sha256=YC3mwLjHxiCrt4t2s15xvAdFcz4yfQVMwqE8mByKmVU,1458
 osbot_utils/helpers/ssh/SSH__Cache__Requests.py,sha256=Dqh4biVcuaXbQVvn3Tx-kSGBGHiF-2wVsgu96EhD6gU,3359
-osbot_utils/helpers/ssh/SSH__Execute.py,sha256=2RjLEf8UuxsYjRoJDmbNdmx5LGgEN4pDS-DhXe3_mDs,6252
+osbot_utils/helpers/ssh/SSH__Execute.py,sha256=D5tQGbSPaNsrMvhTVT0sZIHpqF1rIYXYiaWzGExcwm4,6849
 osbot_utils/helpers/ssh/SSH__Health_Check.py,sha256=WDmBD6ejNcBeicXfjpsiNzH-WR3Jejx0re3WfwjSWyQ,2083
 osbot_utils/helpers/ssh/SSH__Linux.py,sha256=O1uyKcklaj2tHqQZln7dVinzjl9-EI52KzP8ojQr244,4330
+osbot_utils/helpers/ssh/SSH__Linux__Amazon.py,sha256=ZJFb7LFTvclAuhH5OoOtJ361NoX9ecHTaFX-iSmnzmk,596
 osbot_utils/helpers/ssh/SSH__Python.py,sha256=O2DAwkbXzwkis8lffoqIL2NPSfYcN44Mr8i9Ey2iMKk,2066
 osbot_utils/helpers/ssh/TestCase__SSH.py,sha256=rlhkiVr1OR_3uiwqK2dVZ-yBwZZpUhMq6BPT2p21H1s,1598
 osbot_utils/helpers/ssh/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -270,7 +271,7 @@ osbot_utils/utils/Int.py,sha256=PmlUdU4lSwf4gJdmTVdqclulkEp7KPCVUDO6AcISMF4,116
 osbot_utils/utils/Json.py,sha256=UNaBazuH1R40fsHjpjuK8kmAANmUHoK9Q0PUeYmgPeY,6254
 osbot_utils/utils/Json_Cache.py,sha256=mLPkkDZN-3ZVJiDvV1KBJXILtKkTZ4OepzOsDoBPhWg,2006
 osbot_utils/utils/Lists.py,sha256=CLEjgZwAixJAFlubWEKjnUUhUN85oqvR7UqExVW7rdY,5502
-osbot_utils/utils/Misc.py,sha256=5pVxQLCOcGNL3PtasOtQCN8ostg2X-H7mucdZjgg_ns,16580
+osbot_utils/utils/Misc.py,sha256=ljscBemI5wOhfkl1BVpsqshacTOCKkOisV4er9xPCWM,16640
 osbot_utils/utils/Objects.py,sha256=SNtQ1nJnqihwTcmIf_Hg9P3V7fHjxcVHSzC7vKKI34Q,14419
 osbot_utils/utils/Png.py,sha256=V1juGp6wkpPigMJ8HcxrPDIP4bSwu51oNkLI8YqP76Y,1172
 osbot_utils/utils/Process.py,sha256=lr3CTiEkN3EiBx3ZmzYmTKlQoPdkgZBRjPulMxG-zdo,2357
@@ -281,8 +282,8 @@ osbot_utils/utils/Toml.py,sha256=-bg9srF7ef31UCNFa0aL3CDeDKp14XeUVwRMQzMb_dI,107
 osbot_utils/utils/Version.py,sha256=Ww6ChwTxqp1QAcxOnztkTicShlcx6fbNsWX5xausHrg,422
 osbot_utils/utils/Zip.py,sha256=YFahdBguVK71mLdYy4m7mqVAQ5al-60QnTmYK-txCfY,6784
 osbot_utils/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-osbot_utils/version,sha256=o-_WEzInxIO2B4oYaxJD3HXoT3MWSTq6e3KoW9rZp-I,8
-osbot_utils-1.21.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-osbot_utils-1.21.0.dist-info/METADATA,sha256=0aIOGo4sPuKtaD1s0iAq3KhhLvqy_rF2vGKgrCq5H9o,1266
-osbot_utils-1.21.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-osbot_utils-1.21.0.dist-info/RECORD,,
+osbot_utils/version,sha256=z7NTPYmQ0L2Y8UsRZopbRpNKn5uryAcux3sKpD_TKh0,8
+osbot_utils-1.22.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+osbot_utils-1.22.0.dist-info/METADATA,sha256=iNHC9Qu30A7DwLcvMTVwBkWT_EHdzVUYP4XXooShCKI,1266
+osbot_utils-1.22.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+osbot_utils-1.22.0.dist-info/RECORD,,