atomicshop 3.3.12__py3-none-any.whl → 3.3.13__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.12'
+__version__ = '3.3.13'

atomicshop/mitm/config_static.py

@@ -140,9 +140,10 @@ class Certificates:
     sni_server_certificate_from_server_socket_download_directory: str
 
     domains_all_times: list[str]
-    enable_sslkeylogfile_env_to_client_ssl_context: bool = True
-    sslkeylog_file_path: str = None
+    sslkeylog_file_path: str
+
     sslkeylog_file_name: str = "sslkeylog.txt"
+    enable_sslkeylogfile_env_to_client_ssl_context: bool = True
 
 
 @dataclass

atomicshop/wrappers/mongodbw/mongo_infra.py

@@ -1,17 +1,7 @@
-import os
-from typing import Union
-
 from pymongo import MongoClient
 import pymongo.errors
 
-from ... import filesystem
-
-if os.name == 'nt':
-    from ... import get_process_list
 
-
-WHERE_TO_SEARCH_FOR_MONGODB_EXE: str = 'C:\\Program Files\\MongoDB\\Server\\'
-MONGODB_EXE_NAME: str = 'mongod.exe'
 MONGODB_DEFAULT_URI: str = 'mongodb://localhost:27017/'
 
 
@@ -39,31 +29,3 @@ def test_connection(
         if raise_exception:
             raise MongoDBNoConnectionError(f"Could not connect to the MongoDB server on: ")
         return False
-
-
-def is_service_running() -> bool:
-    """
-    Check if the MongoDB service is running.
-    :return: bool, True if the MongoDB service is running, False otherwise.
-    """
-
-    if os.name == 'nt':
-        current_processes: dict = (
-            get_process_list.GetProcessList(get_method='pywin32', connect_on_init=True).get_processes())
-
-        for pid, process_info in current_processes.items():
-            if MONGODB_EXE_NAME in process_info['name']:
-                return True
-    else:
-        raise NotImplementedError("This function is not implemented for this OS.")
-
-    return False
-
-
-def is_installed() -> Union[str, None]:
-    """
-    Check if MongoDB is installed.
-    :return: string if MongoDB executable is found, None otherwise.
-    """
-
-    return filesystem.find_file(MONGODB_EXE_NAME, WHERE_TO_SEARCH_FOR_MONGODB_EXE)

atomicshop/wrappers/ubuntu_terminal.py

@@ -4,10 +4,15 @@ import subprocess
 import shutil
 import time
 
+from rich.console import Console
+
 from ..print_api import print_api
 from ..permissions import ubuntu_permissions
 
 
+console = Console()
+
+
 def install_packages(
         package_list: list[str],
         timeout_seconds: int = 0,
@@ -216,8 +221,7 @@ def start_enable_service_check_availability(
         enable_service_bool: bool = True,
         check_service_running: bool = True,
         user_mode: bool = False,
-        sudo: bool = True,
-        print_kwargs: dict = None
+        sudo: bool = True
 ):
     """
     Function starts and enables a service and checks its availability.
@@ -231,7 +235,6 @@ def start_enable_service_check_availability(
     :param check_service_running: bool, if True, the function will check if the service is running.
     :param user_mode: bool, if True, the service will be started and enabled in user mode.
     :param sudo: bool, if True, the command will be executed with sudo.
-    :param print_kwargs: dict, the print arguments.
     :return:
     """
 
@@ -245,18 +248,18 @@ def start_enable_service_check_availability(
         enable_service(service_name, user_mode=user_mode,sudo=sudo)
 
     if check_service_running:
-        print_api(
-            f"Waiting {str(wait_time_seconds)} seconds for the program to start before availability check...",
-            **(print_kwargs or {}))
-        time.sleep(wait_time_seconds)
+        print(f"Waiting up to {str(wait_time_seconds)} seconds for the program to start.")
+        count: int = 0
+        while not is_service_running(service_name, user_mode=user_mode) and count < wait_time_seconds:
+            count += 1
+            time.sleep(1)
 
         if not is_service_running(service_name, user_mode=user_mode):
-            print_api(
-                f"[{service_name}] service failed to start.", color='red', error_type=True, **(print_kwargs or {}))
+            console.print(f"[{service_name}] service failed to start.", style='red', markup=False)
             if exit_on_error:
                 sys.exit(1)
         else:
-            print_api(f"[{service_name}] service is running.", color='green', **(print_kwargs or {}))
+            console.print(f"[{service_name}] service is running.", style='green', markup=False)
 
 
 def add_path_to_bashrc(as_regular_user: bool = False):

{atomicshop-3.3.12.dist-info → atomicshop-3.3.13.dist-info}/METADATA

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

{atomicshop-3.3.12.dist-info → atomicshop-3.3.13.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=TBMr-8fSW7-pm2urzQlk8alPCT-9WMhHJRs5SESDf_M,123
+atomicshop/__init__.py,sha256=MGjnRWFGJfLj9k_F5wy-o3STJj-zwtz-OfK8-ZufwXQ,123
 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
@@ -53,9 +53,6 @@ atomicshop/websocket_parse.py,sha256=aLHWyKqaYqEn_MRBWm2L6rIl6QPmqbVrjEXE_rBzwCw
 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/win/fibratus.py,sha256=TU4e9gdZ_zI73C40uueJ59pD3qmN-UFGdX5GFoVf6cM,179
-atomicshop/a_installs/win/mongodb.py,sha256=AqyItXu19aaoe49pppDxtEkXey6PMy0PoT2Y_RmPpPE,179
 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
@@ -129,7 +126,7 @@ atomicshop/file_io/tomls.py,sha256=vZ_Wng5alLf8z6HSEZj7PS0XKDA-Iies9ihVWOkTcKo,1
 atomicshop/file_io/xlsxs.py,sha256=v_dyg9GD4LqgWi6wA1QuWRZ8zG4ZwB6Dz52ytdcmmmI,2184
 atomicshop/file_io/xmls.py,sha256=zh3SuK-dNaFq2NDNhx6ivcf4GYCfGM8M10PcEwDSpxk,2104
 atomicshop/mitm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/mitm/config_static.py,sha256=4_lsZVVHCKMK8KZJeY3sXOiIJGRTUwoyckVo4EVa1FY,8931
+atomicshop/mitm/config_static.py,sha256=KzO8DjZjRHfkQMYSIGTkW4jLNPzMR8visTqs1H6ZQ-U,8926
 atomicshop/mitm/config_toml_editor.py,sha256=2p1CMcktWRR_NW-SmyDwylu63ad5e0-w1QPMa8ZLDBw,1635
 atomicshop/mitm/connection_thread_worker.py,sha256=p-93_zdq3HzWpR7NF-gIq0JvvX0L8jz3_TSD3tBhV4o,33255
 atomicshop/mitm/import_config.py,sha256=6fPiV_3v2ym1LgiLlbGoiIPaB6pfRNM9UBXmRnP-pMQ,18560
@@ -200,7 +197,7 @@ atomicshop/wrappers/pipw.py,sha256=mu4jnHkSaYNfpBiLZKMZxEX_E2LqW5BVthMZkblPB_c,1
 atomicshop/wrappers/process_wrapper_pbtk.py,sha256=ycPmBRnv627RWks6N8OhxJQe8Gu3h3Vwj-4HswPOw0k,599
 atomicshop/wrappers/pyopensslw.py,sha256=pklvXEbi0fHu5n1eRKKHEDLN_nsIqCTXv5Lv0bzReTo,7071
 atomicshop/wrappers/sysmonw.py,sha256=CdawuWuy_uUi3ALCm6lKP7pSyKeTk1MXyzOaTMbBSO8,5346
-atomicshop/wrappers/ubuntu_terminal.py,sha256=vGEBznz_EKzjhsFAXzefep3SwdP2toYPxWkagOjME3I,12328
+atomicshop/wrappers/ubuntu_terminal.py,sha256=hHXsr_jEHU4frCwgo1XWS5Gb5nmg-Q06tiem4N8RYeE,12316
 atomicshop/wrappers/wslw.py,sha256=2Z7X0j5M2hoRZjbHfm_vqwNXZeptsdkNCdhdcM_S9vo,6998
 atomicshop/wrappers/certauthw/certauth.py,sha256=qLo593_MLU8VfbhYoNQ3J5BvtZuE71aFQROysEt6_dk,12225
 atomicshop/wrappers/certauthw/certauthw.py,sha256=4WvhjANI7Kzqrr_nKmtA8Kf7B6rute_5wfP65gwQrjw,8082
@@ -257,8 +254,6 @@ atomicshop/wrappers/factw/rest/firmware.py,sha256=FezneouU1lUO9uZ6_8ZQNxr4MDlFIo
 atomicshop/wrappers/factw/rest/router.py,sha256=fdGok5ESBxcZHIBgM93l4yTPRGoeooQNsrPWIETieGk,710
 atomicshop/wrappers/factw/rest/statistics.py,sha256=vznwzKP1gEF7uXz3HsuV66BU9wrp73N_eFqpFpye9Qw,653
 atomicshop/wrappers/factw/rest/status.py,sha256=4O3xS1poafwyUiLDkhyx4oMMe4PBwABuRPpOMnMKgIU,641
-atomicshop/wrappers/fibratusw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/wrappers/fibratusw/install.py,sha256=GnaAAqcXRhovxZ3x5uB9RAXTMCh5xd5k1niCKTzh4Z0,3242
 atomicshop/wrappers/loggingw/consts.py,sha256=JWiUJEydjhwatBxtIJsGTmDUSTLbmIRidtR6qRLMaIY,1608
 atomicshop/wrappers/loggingw/filters.py,sha256=48UVhJHemCS0agXmQP8dHvAHM8r9DFphJ1TNEBP3Dlg,3545
 atomicshop/wrappers/loggingw/formatters.py,sha256=ZY12IokVY1G_Wzn2Zlv9qjK-e8CtIK6yUgUfPHvH2BU,5802
@@ -267,9 +262,7 @@ atomicshop/wrappers/loggingw/loggers.py,sha256=mmM__XR3W4QC82wbsDRG_M4_0JYGGEP0Q
 atomicshop/wrappers/loggingw/loggingw.py,sha256=Cp_9IsiloPYpp9uzEQtUX-jCfcgGe4B2nFBle8x4wqQ,43582
 atomicshop/wrappers/loggingw/reading.py,sha256=sCNlgqLNH5XdKqOOjjEox7CvViMHzs6h7-hwCnx4NKk,17566
 atomicshop/wrappers/mongodbw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/wrappers/mongodbw/install_mongodb_ubuntu.py,sha256=2eEOb35T259lhn5koynfTIm1hanxD02zN97ExGSBM2o,4021
-atomicshop/wrappers/mongodbw/install_mongodb_win.py,sha256=64EUQYx7VuMC3ndO2x3nSErh5NZ_BsqMwGvPcybfC-Q,8499
-atomicshop/wrappers/mongodbw/mongo_infra.py,sha256=IjEF0jPzQz866MpTm7rnksnyyWQeUT_B2h2DA9ryAio,2034
+atomicshop/wrappers/mongodbw/mongo_infra.py,sha256=vEXKy1Mi6LZ7QNEtRb8d-5xT5PYFhXbeEfFsXfsgpYo,918
 atomicshop/wrappers/mongodbw/mongodbw.py,sha256=CHbDfW9CXzXUs3V97DYQpt-dYlt_gB60JhwqG2tVFQY,56049
 atomicshop/wrappers/playwrightw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/playwrightw/_tryouts.py,sha256=l1BLkFsiIMNlgv7nfZd1XGEvXQkIQkIcg48__9OaC00,4920
@@ -325,9 +318,9 @@ atomicshop/wrappers/socketw/statistics_csv.py,sha256=_gA8bMX6Sw_UCXKi2y9wNAwlqif
 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.12.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.3.12.dist-info/METADATA,sha256=UT8jiyOESmyvNu499k17z5YofzY-uq_a_XLS2E6kknw,9312
-atomicshop-3.3.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-atomicshop-3.3.12.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
-atomicshop-3.3.12.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.3.12.dist-info/RECORD,,
+atomicshop-3.3.13.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.3.13.dist-info/METADATA,sha256=OXcv3FuM8llsBWRC4ms6FaYLM_g3gUXQPvpCuMPoS3U,9312
+atomicshop-3.3.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+atomicshop-3.3.13.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
+atomicshop-3.3.13.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.3.13.dist-info/RECORD,,

atomicshop/a_installs/ubuntu/mongodb.py

@@ -1,12 +0,0 @@
-#! /usr/bin/env python3
-import sys
-
-from atomicshop.wrappers.mongodbw import install_mongodb_ubuntu
-
-
-def main():
-    install_mongodb_ubuntu.install_main(compass=True)
-
-
-if __name__ == "__main__":
-    sys.exit(main())

atomicshop/a_installs/win/fibratus.py

@@ -1,9 +0,0 @@
-from atomicshop.wrappers.fibratusw import install
-
-
-def main():
-    install.install_fibratus(remove_file_after_installation=True)
-
-
-if __name__ == '__main__':
-    main()

atomicshop/a_installs/win/mongodb.py

@@ -1,9 +0,0 @@
-from atomicshop.wrappers.mongodbw import install_mongodb_win
-
-
-def main():
-    install_mongodb_win.download_install_latest_main()
-
-
-if __name__ == "__main__":
-    main()

atomicshop/wrappers/fibratusw/install.py

@@ -1,80 +0,0 @@
-import os.path
-import subprocess
-import time
-from typing import Literal
-
-from .. import githubw, msiw
-from ... import filesystem
-from ...print_api import print_api
-
-
-DEFAULT_INSTALLATION_EXE_PATH = r"C:\Program Files\Fibratus\Bin\fibratus.exe"
-WAIT_SECONDS_FOR_EXECUTABLE_TO_APPEAR_AFTER_INSTALLATION: float = 10
-
-
-def install_fibratus(
-        installation_file_download_directory: str = None,
-        place_to_download_file: Literal['working', 'temp', 'script'] = 'temp',
-        remove_file_after_installation: bool = False
-):
-    """
-    Download latest release from GitHub and install Fibratus.
-    :param installation_file_download_directory: Directory to download the installation file. If None, the download
-        directory will be automatically determined, by the 'place_to_download_file' parameter.
-    :param place_to_download_file: Where to download the installation file.
-        'working' is the working directory of the script.
-        'temp' is the temporary directory.
-        'script' is the directory of the script.
-    :param remove_file_after_installation: Whether to remove the installation file after installation.
-    :return:
-    """
-
-    if not installation_file_download_directory:
-        installation_file_download_directory = filesystem.get_download_directory(
-            place=place_to_download_file, script_path=__file__)
-
-    github_wrapper = githubw.GitHubWrapper(user_name='rabbitstack', repo_name='fibratus')
-    fibratus_setup_file_path: str = github_wrapper.download_latest_release(
-        target_directory=installation_file_download_directory,
-        string_pattern='*fibratus-*-amd64.msi',
-        exclude_string='slim')
-
-    # Install the MSI file
-    msiw.install_msi(msi_path=fibratus_setup_file_path)
-
-    count = 0
-    while count != WAIT_SECONDS_FOR_EXECUTABLE_TO_APPEAR_AFTER_INSTALLATION:
-        if os.path.isfile(DEFAULT_INSTALLATION_EXE_PATH):
-            break
-        count += 1
-        time.sleep(1)
-
-    if count == WAIT_SECONDS_FOR_EXECUTABLE_TO_APPEAR_AFTER_INSTALLATION:
-        message = \
-            (f"Fibratus installation failed. The executable was not found after "
-             f"{str(WAIT_SECONDS_FOR_EXECUTABLE_TO_APPEAR_AFTER_INSTALLATION)} seconds.\n"
-             f"{DEFAULT_INSTALLATION_EXE_PATH}")
-        print_api(message, color="red")
-
-    result = None
-    # Check if the installation was successful
-    try:
-        result = subprocess.run([DEFAULT_INSTALLATION_EXE_PATH], capture_output=True, text=True)
-    except FileNotFoundError:
-        print_api("Fibratus executable not found.", color="red")
-
-    if result:
-        if result.returncode == 0:
-            print_api("Fibratus installed successfully. Please restart.", color="green")
-        else:
-            print_api("Fibratus installation failed.", color="red")
-            print_api(result.stderr)
-            raise Exception("Fibratus installation failed.")
-    else:
-        print_api("Fibratus executable not found.", color="red")
-
-    # Wait for the installation to finish before removing the file.
-    time.sleep(5)
-
-    if remove_file_after_installation:
-        filesystem.remove_file(fibratus_setup_file_path)

atomicshop/wrappers/mongodbw/install_mongodb_ubuntu.py

@@ -1,100 +0,0 @@
-import subprocess
-import sys
-import os
-
-
-from ... import print_api, web
-from .. import ubuntu_terminal
-from ...permissions import ubuntu_permissions
-
-
-COMPASS_INSTALLATION_SCRIPT_URL: str = \
-    'https://raw.githubusercontent.com/mongodb/mongo/master/src/mongo/installer/compass/install_compass'
-
-
-def run_command(command):
-    """Run a system command and exit if the command fails."""
-    try:
-        subprocess.run(command, check=True, shell=True)
-    except subprocess.CalledProcessError as e:
-        print_api.print_api(f"Error: {e}", color='red')
-        sys.exit(1)
-
-
-def install_mongodb(version):
-    """Install the specified major version of MongoDB on Ubuntu."""
-
-    if not version.endswith(".0"):
-        version = f"{version}.0"
-
-    print_api.print_api(f"Installing MongoDB {version} on Ubuntu...")
-    print_api.print_api(f"Installing Prerequisites...")
-    ubuntu_terminal.update_system_packages()
-    ubuntu_terminal.install_packages(["wget", "curl", "gnupg"])
-
-    # Step 1: Import the MongoDB public GPG key
-    print_api.print_api("Step 1: Importing the MongoDB public GPG key...")
-    run_command(f"curl -fsSL https://pgp.mongodb.com/server-{version}.asc | "
-                f"sudo gpg --dearmor --yes -o /usr/share/keyrings/mongodb-server-{version}.gpg")
-
-    # Step 2: Create the MongoDB list file for APT
-    print_api.print_api("Step 2: Creating MongoDB APT list file...")
-    distro_version = subprocess.check_output("lsb_release -sc", shell=True).decode('utf-8').strip()
-    run_command(
-        f"echo 'deb [ arch=amd64,arm64 signed-by=/usr/share/keyrings/mongodb-server-{version}.gpg ] "
-        f"https://repo.mongodb.org/apt/ubuntu {distro_version}/mongodb-org/{version} multiverse' | "
-        f"sudo tee /etc/apt/sources.list.d/mongodb-org-{version}.list")
-
-    # Step 3: Update the APT package index
-    print_api.print_api("Step 3: Updating the APT package index...")
-    ubuntu_terminal.update_system_packages()
-
-    # Step 4: Install the latest version of MongoDB for the specified major version
-    print_api.print_api(f"Step 4: Installing MongoDB version {version}...")
-    ubuntu_terminal.install_packages(["mongodb-org"])
-
-    # Step 5: Start MongoDB service and enable it on startup
-    print_api.print_api("Step 5: Starting MongoDB service and enabling it on startup...")
-    ubuntu_terminal.start_enable_service_check_availability("mongod")
-
-    print_api.print_api(f"MongoDB {version} installation complete!", color='green')
-
-
-def install_main(
-        compass: bool = False,
-):
-    """
-    Install the latest minor version of MongoDB Community Server on Ubuntu by providing the major version.
-    :param compass: bool, if True, MongoDB Compass will be installed.
-    :return:
-    """
-    # Ensure the user provides a MongoDB major version as an argument.
-    if len(sys.argv) != 2:
-        message: str = ("Usage: python install_mongodb.py <mongo_major_version>\n"
-                        "Example: python install_mongodb.py 8")
-        print_api.print_api(message, color='red')
-        return 1
-
-    mongo_version = sys.argv[1]
-
-    # Call the 'install' function with the major version.
-    install_mongodb(mongo_version)
-
-    if not compass:
-        return 0
-
-    # It doesn't matter what you do with the MSI it will not install Compass, only if you run it manually.
-    # So we will use installation script from their GitHub.
-    print_api.print_api("Downloading MongoDB Compass installation script...")
-    compass_script_path: str = web.download(COMPASS_INSTALLATION_SCRIPT_URL)
-
-    print_api.print_api("Installing MongoDB Compass from script...")
-    ubuntu_permissions.set_executable(compass_script_path)
-    run_command(f'sudo -E python3 {compass_script_path}')
-
-    # Clean up the installer file
-    if os.path.exists(compass_script_path):
-        os.remove(compass_script_path)
-        print_api.print_api("Cleaned up the Compass installer file.")
-
-    return 0

atomicshop/wrappers/mongodbw/install_mongodb_win.py

@@ -1,244 +0,0 @@
-import os
-import requests
-from typing import Union
-import argparse
-import subprocess
-
-from ... import urls, web
-from ...permissions import permissions
-from ...print_api import print_api
-from .. import msiw
-from . import mongo_infra
-
-
-MONGODB_DOWNLOAD_PAGE_URL: str = 'https://www.mongodb.com/try/download/community'
-COMPASS_INSTALLATION_SCRIPT_URL: str = \
-    'https://raw.githubusercontent.com/mongodb/mongo/master/src/mongo/installer/compass/Install-Compass.ps1'
-
-
-class MongoDBWebPageNoSuccessCodeError(Exception):
-    pass
-
-
-class MongoDBNoDownloadLinksError(Exception):
-    pass
-
-
-class MongoDBNoDownloadLinkForWindowsError(Exception):
-    pass
-
-
-class MongoDBInstallationError(Exception):
-    pass
-
-
-def get_latest_mongodb_download_url(
-        rc_version: bool = True,
-        major_specific: int = None
-):
-    response = requests.get(MONGODB_DOWNLOAD_PAGE_URL)
-
-    if response.status_code != 200:
-        raise MongoDBWebPageNoSuccessCodeError("Failed to load the download page.")
-
-    urls_in_page: list = urls.find_urls_in_text(response.text)
-    if not urls_in_page:
-        raise MongoDBNoDownloadLinksError("Could not find the download link for MongoDB Community Server.")
-
-    windows_urls: list = []
-    for url in urls_in_page:
-        if 'windows' in url and 'x86_64' in url and url.endswith('.msi'):
-            if not rc_version and '-rc' in url:
-                continue
-            windows_urls.append(url)
-
-    if major_specific:
-        for url in windows_urls:
-            if f'-{major_specific}.' in url:
-                windows_urls = [url]
-                break
-
-    if not windows_urls:
-        raise MongoDBNoDownloadLinkForWindowsError(
-            "Could not find the download link for MongoDB Community Server for Windows x86_64.")
-
-    # Return the latest URL only.
-    return windows_urls[0]
-
-
-def parse_args():
-    parser = argparse.ArgumentParser(description='Install MongoDB Community Server.')
-    # Mutually exclusive group for 'ir' and 'er'
-    group = parser.add_mutually_exclusive_group()
-    group.add_argument(
-        '-er', '--exclude-rc',
-        action='store_true',
-        help='Install latest version, but exclude release candidate versions.'
-    )
-    group.add_argument(
-        '-ir', '--include-rc',
-        action='store_true',
-        help='Install the latest version of MongoDB including release candidates.'
-    )
-
-    # Optional arguments
-    parser.add_argument(
-        '-c', '--compass',
-        action='store_true',
-        help='Install MongoDB Compass.'
-    )
-    parser.add_argument(
-        '-m', '--major',
-        type=int,
-        help='Download the latest version of the specified major version.'
-    )
-    parser.add_argument(
-        '-f', '--force',
-        action='store_true',
-        help='Force the installation even if MongoDB is already installed.'
-    )
-
-    args = parser.parse_args()
-
-    # Validation logic: At least one of '-nr', '-ir', or '-c' must be provided
-    if not (args.no_rc or args.include_rc or args.compass):
-        parser.error("At least one of '-nr', '-ir', or '-c' must be specified.")
-
-    return args
-
-
-def download_install_process(
-        rc_version: bool = False,
-        no_rc_version: bool = True,
-        major_specific: int = None,
-        compass: bool = False,
-        force: bool = False
-) -> int:
-    """
-    Download and install the latest version of MongoDB Community Server.
-
-    :param rc_version: bool, if True, the latest RC version will be downloaded.
-    :param no_rc_version: bool, if True, the latest non-RC version will be downloaded.
-    :param major_specific: int, if set, the latest version of the specified major version will be downloaded.
-    :param compass: bool, if True, MongoDB Compass will be installed.
-    :param force: bool, if True, MongoDB will be installed even if it is already installed.
-    :return: int, 0 if successful, 1 if failed.
-    """
-
-    if not permissions.is_admin():
-        print_api("This function requires administrator privileges.", color='red')
-        return 1
-
-    if rc_version and no_rc_version:
-        print_api("Both 'rc_version' and 'no_rc_version' cannot be True at the same time.", color='red')
-        return 1
-
-    if not (rc_version or no_rc_version) and not compass:
-        print_api("At least one of 'rc_version', 'no_rc_version', or 'compass' must be True.", color='red')
-        return 1
-
-    # If we need to install mongo db.
-    if rc_version or no_rc_version:
-        if rc_version:
-            download_rc_version: bool = True
-        elif no_rc_version:
-            download_rc_version: bool = False
-        else:
-            raise ValueError("Invalid value for 'rc_version' and 'no_rc_version'.")
-
-        if mongo_infra.is_service_running():
-            print_api("MongoDB service is running - already installed. Use [-f] to reinstall.", color='blue')
-
-            if not force:
-                return 0
-        else:
-            print_api("MongoDB is service is not running.")
-
-            mongo_is_installed: Union[str, None] = mongo_infra.is_installed()
-            if mongo_infra.is_installed():
-                message = f"MongoDB is installed in: {mongo_is_installed}\n" \
-                          f"The service is not running. Fix the service or use the 'force' parameter to reinstall."
-                print_api(message, color='yellow')
-
-                if not force:
-                    return 0
-
-        print_api("Fetching the latest MongoDB download URL...")
-        mongo_installer_url = get_latest_mongodb_download_url(rc_version=download_rc_version, major_specific=major_specific)
-
-        print_api(f"Downloading MongoDB installer from: {mongo_installer_url}")
-        installer_file_path: str = web.download(mongo_installer_url)
-
-        print_api("Installing MongoDB...")
-        try:
-            msiw.install_msi(
-                installer_file_path,
-                silent_no_gui=True,
-                no_restart=True,
-                terminate_required_processes=True,
-                create_log_near_msi=True,
-                scan_log_for_errors=True,
-                additional_args='ADDLOCAL="ServerService"'
-            )
-        except msiw.MsiInstallationError as e:
-            print_api(f'{e} Exiting...', color='red')
-            return 1
-
-        # Check if MongoDB is installed.
-        message: str = ''
-        mongo_is_installed = mongo_infra.is_installed()
-        if not mongo_is_installed:
-            message += "MongoDB Executable not found.\n"
-
-        if not mongo_infra.is_service_running():
-            message += "MongoDB service is not running.\n"
-
-        if message:
-            message += f"MSI Path: {installer_file_path}"
-            print_api(message, color='red')
-            return 1
-        else:
-            success_message: str = f"MongoDB installed successfully to: {mongo_is_installed}\n" \
-                                   f"Service is running."
-            print_api(success_message, color='green')
-
-        # Clean up the installer file
-        if os.path.exists(installer_file_path):
-            os.remove(installer_file_path)
-            print_api("Cleaned up the installer file.")
-
-    if not compass:
-        return 0
-
-    # It doesn't matter what you do with the MSI it will not install Compass, only if you run it manually.
-    # So we will use installation script from their GitHub.
-    print_api("Downloading MongoDB Compass installation script...")
-    compass_script_path: str = web.download(COMPASS_INSTALLATION_SCRIPT_URL)
-
-    print_api("Installing MongoDB Compass from script...")
-    subprocess.run(["powershell", "-ExecutionPolicy", "Bypass", "-File", compass_script_path])
-
-    # Clean up the installer file
-    if os.path.exists(compass_script_path):
-        os.remove(compass_script_path)
-        print_api("Cleaned up the Compass installer file.")
-
-    return 0
-
-
-def download_install_latest_main() -> int:
-    """
-    Download and install the latest version of MongoDB Community Server.
-
-    :return: int, 0 if successful, 1 if failed.
-    """
-
-    args = parse_args()
-
-    return download_install_process(
-        rc_version=args.include_rc,
-        no_rc_version=args.exclude_rc,
-        major_specific=args.major,
-        compass=args.compass,
-        force=args.force
-    )