atomicshop 3.3.6__py3-none-any.whl → 3.3.8__py3-none-any.whl

atomicshop/__init__.py

@@ -1,4 +1,4 @@
 """Atomic Basic functions and classes to make developer life easier"""
 
 __author__ = "Den Kras"
-__version__ = '3.3.6'
+__version__ = '3.3.8'

atomicshop/consoles.py

@@ -0,0 +1,9 @@
+import msvcrt
+
+
+# === WINDOWS ONLY FUNCTIONS ===========================================================================================
+def wait_any_key(prompt="Press any key to continue..."):
+    print(prompt, end="", flush=True)
+    msvcrt.getch()  # waits for one key press (no Enter needed)
+    print()         # move to next line
+# === EOF WINDOWS ONLY FUNCTIONS =======================================================================================

atomicshop/mitm/engines/create_module_template.py

@@ -2,8 +2,12 @@ import os
 import argparse
 from typing import Literal
 
-from ... import filesystem
+from ... import filesystem, consoles
 from ...basics import strings
+from rich.console import Console
+
+
+console = Console()
 
 
 GENERAL_CLASS_NAME: str = "General"
@@ -71,6 +75,8 @@ class CreateModuleTemplate:
 
         self.create_config_file()
 
+        consoles.wait_any_key()
+
     def create_config_file(self):
         # Defining variables.
         config_lines_list: list = list()
@@ -93,7 +99,7 @@ class CreateModuleTemplate:
         with open(config_file_path, 'w') as output_file:
             output_file.write('\n'.join(config_lines_list))
 
-        print(f"Config File Created: {config_file_path}")
+        console.print(f"Config File Created: {config_file_path}", style="bright_blue")
 
     def _create_engine_module_from_reference(
             self,
@@ -113,7 +119,7 @@ class CreateModuleTemplate:
             raise ValueError(f"Module type is not recognized: {module_type}")
 
         # Reading the module file to string.
-        with open(file_path, 'r') as input_file:
+        with open(file_path, 'r', encoding='utf-8') as input_file:
             file_content_string = input_file.read()
 
         new_module_full_path: str = str()
@@ -126,4 +132,4 @@ class CreateModuleTemplate:
                 output_file.write(new_content_string)
 
         print(f"Converted: {file_path}")
-        print(f"To: {new_module_full_path}")
+        console.print(f"To: {new_module_full_path}", style="green")

atomicshop/wrappers/socketw/socket_wrapper.py

@@ -370,7 +370,12 @@ class SocketWrapper:
 
             # If someone removed the CA certificate file manually, and now it was created, we also need to
             # clear the cached certificates.
-            shutil.rmtree(self.sni_server_certificates_cache_directory)
+            try:
+                shutil.rmtree(self.sni_server_certificates_cache_directory)
+            # If the directory doesn't exist it will throw an exception, which is OK.
+            except FileNotFoundError:
+                pass
+
             os.makedirs(self.sni_server_certificates_cache_directory, exist_ok=True)
             print_api("Removed cached server certificates.", logger=self.logger)
 

{atomicshop-3.3.6.dist-info → atomicshop-3.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicshop
-Version: 3.3.6
+Version: 3.3.8
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License-Expression: MIT

{atomicshop-3.3.6.dist-info → atomicshop-3.3.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=PKxHKogNO1jVrjMDxPohtYhd5093pVEONPVfmujNgfw,122
+atomicshop/__init__.py,sha256=ngY9n0Huc9qRb7eeksH6N2bPuoeEhjgl3Fjm17woyLg,122
 atomicshop/_basics_temp.py,sha256=6cu2dd6r2dLrd1BRNcVDKTHlsHs_26Gpw8QS6v32lQ0,3699
 atomicshop/_create_pdf_demo.py,sha256=Yi-PGZuMg0RKvQmLqVeLIZYadqEZwUm-4A9JxBl_vYA,3713
 atomicshop/_patch_import.py,sha256=ENp55sKVJ0e6-4lBvZnpz9PQCt3Otbur7F6aXDlyje4,6334
@@ -8,6 +8,7 @@ atomicshop/command_line_processing.py,sha256=u5yT9Ger_cu7ni5ID0VFlRbVD46ARHeNC9t
 atomicshop/config_init.py,sha256=G6_f25zPxyPlht5jJC4ZDrrQiuJfjHiuPd22VBjd9Cg,2683
 atomicshop/console_output.py,sha256=AOSJjrRryE97PAGtgDL03IBtWSi02aNol8noDnW3k6M,4667
 atomicshop/console_user_response.py,sha256=OHcjuzWAys6WmfRnMIU_nkJA634kKmJh6T8w1VtUTJM,2714
+atomicshop/consoles.py,sha256=jlEKqMV7OUCq_vCHMwePcSxRa5yZhrdEyfaVcoWj2_k,464
 atomicshop/datetimes.py,sha256=IQZ66lmta-ZqxYbyHzm_9eugbJFSilXK1e0kfMgoXGg,18371
 atomicshop/diff_check.py,sha256=vxTDccVbGZHEge6Ja9_ArLWwslOUgIoJAdYPylh4cZg,27176
 atomicshop/dns.py,sha256=XB0tijVi1bxWLWKV9hPzpt75jK4SrGbZCV5VJbiiQ74,7185
@@ -49,15 +50,12 @@ atomicshop/versioning.py,sha256=e5W6m9AF3__M5nntqI9CqNAeHqkwY9JhlnpYeZ1CEus,970
 atomicshop/virtualization.py,sha256=LPP4vjE0Vr10R6DA4lqhfX_WaNdDGRAZUW0Am6VeGco,494
 atomicshop/web.py,sha256=hkAS0MeMW_mGgxhFCX1xckT3yslCJ2cMP2nx9ADpe3o,13024
 atomicshop/websocket_parse.py,sha256=aLHWyKqaYqEn_MRBWm2L6rIl6QPmqbVrjEXE_rBzwCw,16711
-atomicshop/a_installs/pywintrace.py,sha256=oZ9BATsJ2YsdIkZZWLar_dAUV32-oDiT7MJxthC0NnU,738
 atomicshop/a_installs/ubuntu/docker_rootless.py,sha256=9IPNtGZYjfy1_n6ZRt7gWz9KZgR6XCgevjqq02xk-o0,281
 atomicshop/a_installs/ubuntu/docker_sudo.py,sha256=JzayxeyKDtiuT4Icp2L2LyFRbx4wvpyN_bHLfZ-yX5E,281
 atomicshop/a_installs/ubuntu/elastic_search_and_kibana.py,sha256=yRB-l1zBxdiN6av-FwNkhcBlaeu4zrDPjQ0uPGgpK2I,244
 atomicshop/a_installs/ubuntu/mongodb.py,sha256=xuRJS1qqOZ0EZp7of5R3tTjSu6CwBIxYg8-NaM7othE,230
-atomicshop/a_installs/ubuntu/pycharm.py,sha256=Ld7YQBwPxrjuZcTG1K4kZqjbBdt8aooCVRa15u5FOmE,157
 atomicshop/a_installs/win/fibratus.py,sha256=TU4e9gdZ_zI73C40uueJ59pD3qmN-UFGdX5GFoVf6cM,179
 atomicshop/a_installs/win/mongodb.py,sha256=AqyItXu19aaoe49pppDxtEkXey6PMy0PoT2Y_RmPpPE,179
-atomicshop/a_installs/win/pycharm.py,sha256=j_RSd7aDOyC3yDd-_GUTMLlQTmDrqtVFG--oUfGLiZk,140
 atomicshop/a_installs/win/wsl_ubuntu_lts.py,sha256=dZbPRLNKFeMd6MotjkE6UDY9cOiIaaclIdR1kGYWI50,139
 atomicshop/a_mains/dns_gateway_setting.py,sha256=ncc2rFQCChxlNP59UshwmTonLqC6MWblrVAzbbz-13M,149
 atomicshop/a_mains/github_wrapper.py,sha256=F-PoZknVCxWPN0PTO6l7ZNiaYvo7OVFKFI_zlPt56ps,169
@@ -142,7 +140,7 @@ atomicshop/mitm/recs_files.py,sha256=tv8XFhYZMkBv4DauvpiAdPgvSo0Bcm1CghnmwO7dx8M
 atomicshop/mitm/shared_functions.py,sha256=0lzeyINd44sVEfFbahJxQmz6KAMWbYrW5ou3UYfItvw,1777
 atomicshop/mitm/statistic_analyzer.py,sha256=5_sAYGX2Xunzo_pS2W5WijNCwr_BlGJbbOO462y_wN4,27533
 atomicshop/mitm/engines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/mitm/engines/create_module_template.py,sha256=2ceRWPzofpRf-R8VAMgXrx2D7vumkhfsrr120YFGgH0,5764
+atomicshop/mitm/engines/create_module_template.py,sha256=PHE2pVC9JNgaIh2o7M5dFMrkdOkmIyHLoO2mdzE5BdM,5938
 atomicshop/mitm/engines/create_module_template_main_example.py,sha256=LeQ44Rp2Gi_KbIDY_4OMS0odkSK3zFZWra_oAka5eJY,243
 atomicshop/mitm/engines/__parent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/engines/__parent/parser___parent.py,sha256=HHaCXhScl3OlPjz6eUxsDpJaZyk6BNuDMc9xCkeo2Ws,661
@@ -292,9 +290,6 @@ atomicshop/wrappers/psutilw/processes.py,sha256=ihYnxfMTVEXHWy92iewktoZGxazx3v5Q
 atomicshop/wrappers/psutilw/psutil_networks.py,sha256=79FplDAj45ofBCudlft77O3lfi0RDwJ3ijye0ZGRR44,2870
 atomicshop/wrappers/psutilw/psutilw.py,sha256=q3EwgprqyrR4zLCjl4l5DHFOQoukEvQMIPjNB504oQ0,21262
 atomicshop/wrappers/psycopgw/psycopgw.py,sha256=XJvVf0oAUjCHkrYfKeFuGCpfn0Oxj3u4SbKMKA1508E,7118
-atomicshop/wrappers/pycharmw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/wrappers/pycharmw/ubuntu.py,sha256=vOvGWvTbzvTja9tRrJW2yJb0_r1EV11RrENGHckzvVE,1051
-atomicshop/wrappers/pycharmw/win.py,sha256=hNP-d95z1zhcCpYqhHE5HZVYxaAlt8JJCNXh65jZsHc,2757
 atomicshop/wrappers/pywin32w/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/pywin32w/cert_store.py,sha256=dV1XyoKTFKZ-HCIVqU2Nd6CTZ8HANqjAXv26rsNzO6s,4365
 atomicshop/wrappers/pywin32w/console.py,sha256=LstHajPLgXp9qQxFNR44QfH10nOnNp3bCJquxaTquns,1175
@@ -324,15 +319,15 @@ atomicshop/wrappers/socketw/sender.py,sha256=aX_K8l_rHjd5AWb8bi5mt8-YTkMYVRDB6Dn
 atomicshop/wrappers/socketw/sni.py,sha256=YlKavbExcPFfHFLYAJ3i3W6QorY7o4mbQp39g-DnDKA,17911
 atomicshop/wrappers/socketw/socket_client.py,sha256=McBd3DeCy787oDGCEMUEP2awWy3vdkPqr9w-aFh2fBM,22502
 atomicshop/wrappers/socketw/socket_server_tester.py,sha256=Qobmh4XV8ZxLUaw-eW4ESKAbeSLecCKn2OWFzMhadk0,6420
-atomicshop/wrappers/socketw/socket_wrapper.py,sha256=u_v0pjMMrgsdtI0iPPddiLe2wXnBoqTgNM9Y3zjGD4U,41013
+atomicshop/wrappers/socketw/socket_wrapper.py,sha256=VZe27EQhExaiLQ0FEW4ePJhNSwPMyPzgcl6oljMSbGg,41185
 atomicshop/wrappers/socketw/ssl_base.py,sha256=62-hPm7zla1rh3m_WvDnXqKH-sDUTdiRptD8STCkgdk,2313
 atomicshop/wrappers/socketw/statistics_csv.py,sha256=_gA8bMX6Sw_UCXKi2y9wNAwlqifgExgDGfQIa9pFxQA,5543
 atomicshop/wrappers/winregw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/winregw/winreg_installed_software.py,sha256=Qzmyktvob1qp6Tjk2DjLfAqr_yXV0sgWzdMW_9kwNjY,2345
 atomicshop/wrappers/winregw/winreg_network.py,sha256=ih0BVNwByLvf9F_Lac4EdmDYYJA3PzMvmG0PieDZrsE,9905
-atomicshop-3.3.6.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.3.6.dist-info/METADATA,sha256=IL5-xRKSd3KMQq1-z8F1hdeVTz9jNmhxtEdbXTAeKtM,9311
-atomicshop-3.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-atomicshop-3.3.6.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
-atomicshop-3.3.6.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.3.6.dist-info/RECORD,,
+atomicshop-3.3.8.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.3.8.dist-info/METADATA,sha256=rYMsgx4ny4J9YRNDVBfecWRdf94iHj8B0AX4EBezbUA,9311
+atomicshop-3.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+atomicshop-3.3.8.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
+atomicshop-3.3.8.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.3.8.dist-info/RECORD,,

atomicshop/a_installs/pywintrace.py

@@ -1,26 +0,0 @@
-import argparse
-import subprocess
-import sys
-
-
-WHEEL = (
-    "https://github.com/fireeye/pywintrace/releases/download/"
-    "v0.3.0/pywintrace-0.3.0-py3-none-any.whl"
-)
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser(
-        prog="pywintrace",
-        description="Utility wrapper for installing FireEye’s pywintrace wheel"
-    )
-    sub = parser.add_subparsers(dest="command", required=True)
-    sub.add_parser("install", help="Download and install pywintrace v0.3.0")
-    args = parser.parse_args()
-
-    if args.command == "install":
-        subprocess.check_call([sys.executable, "-m", "pip", "install", WHEEL])
-        print("pywintrace 0.3.0 installed")
-
-if __name__ == "__main__":
-    main()

atomicshop/a_installs/ubuntu/pycharm.py

@@ -1,8 +0,0 @@
-#! /usr/bin/env python3
-import sys
-
-from atomicshop.wrappers.pycharmw import ubuntu
-
-
-if __name__ == '__main__':
-    sys.exit(ubuntu.install_main())

atomicshop/a_installs/win/pycharm.py

@@ -1,9 +0,0 @@
-from atomicshop.wrappers.pycharmw import win
-
-
-def main():
-    win.download_install_main()
-
-
-if __name__ == "__main__":
-    main()

atomicshop/wrappers/pycharmw/ubuntu.py

@@ -1,36 +0,0 @@
-import argparse
-
-from ... import process
-from ...print_api import print_api
-
-
-def parse_args():
-    """
-    Parse command line arguments.
-
-    :return: Parsed arguments.
-    """
-    parser = argparse.ArgumentParser(description='Install PyCharm Community Edition.')
-    parser.add_argument(
-        '--enable_sudo_execution', action='store_true',
-        help='There is a problem when trying to run snapd installed Pycharm as sudo, need to enable this.')
-
-    return parser.parse_args()
-
-
-def install_main():
-    """
-    Main function to install the latest PyCharm Community Edition.
-
-    Usage:
-    python -m atomicshop.a_installs.ubuntu.pycharm
-    """
-
-    args = parse_args()
-
-    process.execute_script('sudo snap install pycharm-professional --classic', shell=True)
-
-    if args.enable_sudo_execution:
-        process.execute_script('xhost +SI:localuser:root', shell=True)
-        print_api('Run the following command to start PyCharm as root: [sudo snap run pycharm-professional]', color='blue')
-    return 0

atomicshop/wrappers/pycharmw/win.py

@@ -1,81 +0,0 @@
-import requests
-from bs4 import BeautifulSoup
-import subprocess
-
-from ... import web, filesystem
-from ...print_api import print_api
-
-
-# URL to the PyCharm Community Edition download page
-PYCHARM_DOWNLOAD_URL = 'https://www.jetbrains.com/pycharm/download/#section=windowsC'
-
-
-def download_install_main():
-    """
-    Main function to download and install the latest PyCharm Community Edition.
-
-    Usage:
-    python -m atomicshop.mains.installs.pycharm
-
-    Or run the main function directly.
-        from atomicshop.wrappers import pycharmw
-
-
-        def main():
-            pycharmw.download_install_main()
-
-
-        if __name__ == "__main__":
-            main()
-    """
-
-    def get_latest_pycharm_download_link():
-        url = "https://www.jetbrains.com/pycharm/download/"
-        headers = {
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
-        }
-
-        response = requests.get(url, headers=headers)
-        if response.status_code != 200:
-            raise Exception("Failed to load the download page")
-
-        soup = BeautifulSoup(response.text, 'html.parser')
-        download_link = None
-
-        # Find the Professional version download link
-        for a in soup.find_all('a', href=True):
-            if '/download?code=PCC&platform=windows' in a['href']:
-                download_link = a['href']
-                break
-
-        if not download_link:
-            raise Exception("Could not find the download link for the latest version of PyCharm Professional")
-
-        return f"https:{download_link}"
-
-    installer_path: str = None
-    try:
-        print_api("Fetching the latest PyCharm download link...")
-        download_url = get_latest_pycharm_download_link()
-        print_api(f"Download URL: {download_url}")
-
-        print_api("Starting the download...")
-        file_name = "pycharm-latest.exe"
-        # download_file(download_url, file_name)
-        installer_path = web.download(file_url=download_url, file_name=file_name, use_certifi_ca_repository=True)
-        print_api(f"Downloaded the latest version of PyCharm to {file_name}", color='green')
-    except Exception as e:
-        print_api(f"An error occurred: {e}")
-
-    if not installer_path:
-        print_api("Failed to download the latest version of PyCharm", color='red')
-        return 1
-
-    # Install PyCharm
-    # Run the installer
-    print_api("Running the installer...")
-    subprocess.run([installer_path, '/S'], check=True)  # /S for silent installation
-    print_api("Installation complete.", color='green')
-
-    # Remove the installer
-    filesystem.remove_file(installer_path)