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

atomicshop/mitm/mitm_main.py

@@ -3,8 +3,6 @@ import multiprocessing
 import time
 import datetime
 
-from distro import minor_version
-
 import atomicshop   # Importing atomicshop package to get the version of the package.
 
 from .. import filesystem, queues, dns, on_exit, print_api

atomicshop/wrappers/factw/install/install_after_restart.py

@@ -103,7 +103,7 @@ def analyze_log_file(log_file: str):
 
     for plugin in PLUGIN_LIST:
         if f'{FINISHED_INSTALLING_STRINGS[0]} {plugin} {FINISHED_INSTALLING_STRINGS[1]}' not in log_content:
-            message = (f'Error: [{plugin}] installation failed.\n'
+            message = (f'Error: [{plugin}] plugin missing in the log. Installation failed.\n'
                        f'Check the log file: {log_file}\n'
                        f'Exiting...')
             print_api.print_api(message, color='red')

atomicshop/wrappers/mongodbw/install_mongodb_win.py

@@ -33,7 +33,7 @@ class MongoDBInstallationError(Exception):
 
 
 def get_latest_mongodb_download_url(
-        no_rc_version: bool = True,
+        rc_version: bool = True,
         major_specific: int = None
 ):
     response = requests.get(MONGODB_DOWNLOAD_PAGE_URL)
@@ -48,7 +48,7 @@ def get_latest_mongodb_download_url(
     windows_urls: list = []
     for url in urls_in_page:
         if 'windows' in url and 'x86_64' in url and url.endswith('.msi'):
-            if no_rc_version and '-rc' in url:
+            if not rc_version and '-rc' in url:
                 continue
             windows_urls.append(url)
 
@@ -68,109 +68,144 @@ def get_latest_mongodb_download_url(
 
 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(
-        '-nr', '--no-rc', action='store_true', help='Do not download release candidate versions.')
-    parser.add_argument(
-        '-m', '--major', type=int, help='Download the latest version of the specified major version.')
+        '-c', '--compass',
+        action='store_true',
+        help='Install MongoDB Compass.'
+    )
     parser.add_argument(
-        '-c', '--compass', action='store_true', help='Install MongoDB Compass.')
+        '-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.')
+        '-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 parser.parse_args()
+    return args
 
 
-def download_install_latest_main(
-        no_rc_version: bool = False,
+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:
+    :return: int, 0 if successful, 1 if failed.
     """
 
-    args = parse_args()
-
-    # Set the args only if they were used.
-    if args.no_rc:
-        no_rc_version = args.no_rc
-    if args.major:
-        major_specific = args.major
-    if args.compass:
-        compass = args.compass
-    if args.force:
-        force = args.force
-
     if not permissions.is_admin():
         print_api("This function requires administrator privileges.", color='red')
         return 1
 
-    if mongo_infra.is_service_running():
-        print_api("MongoDB service is running - already installed.", 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(no_rc_version=no_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')
+    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
 
-    # 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 (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 not mongo_infra.is_service_running():
-        message += "MongoDB service is not running.\n"
+    # 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 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')
+        if mongo_infra.is_service_running():
+            print_api("MongoDB service is running - already installed. Use [-f] to reinstall.", color='blue')
 
-    # 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 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
@@ -189,3 +224,21 @@ def download_install_latest_main(
         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
+    )

{atomicshop-2.18.3.dist-info → atomicshop-2.18.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atomicshop
-Version: 2.18.3
+Version: 2.18.4
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License: MIT License

{atomicshop-2.18.3.dist-info → atomicshop-2.18.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=aciLHL-9AkNeS_Ahx9NP3lvs533n84eEd96p1hCCdRU,123
+atomicshop/__init__.py,sha256=85MhliVqiJ5CTAxGKQv7PfsSM76Jo4Xtu8UHYW8DdFA,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
@@ -131,7 +131,7 @@ atomicshop/mitm/connection_thread_worker.py,sha256=Of-QmuIbUO8Qd1N_BXdGQk2TnPpZd
 atomicshop/mitm/import_config.py,sha256=0Ij14aISTllTOiWYJpIUMOWobQqGofD6uafui5uWllE,9272
 atomicshop/mitm/initialize_engines.py,sha256=NWz0yBErSrYBn0xWkJDBcHStBJ-kcsv9VtorcSP9x5M,8258
 atomicshop/mitm/message.py,sha256=mNo4Lphr_Jo6IlNX5mPJzABpogWGkjOhwI4meAivwHw,2987
-atomicshop/mitm/mitm_main.py,sha256=3F227Vh-F2ZGxOPE7eP4bgIZ72e3BzdUAHhWIMBbhFo,23490
+atomicshop/mitm/mitm_main.py,sha256=Uko4lFG96ZeZ1yVJD5CT4c48NhfX_Hu1g0-THEiZfAc,23454
 atomicshop/mitm/recs_files.py,sha256=ZAAD0twun-FtmbSniXe3XQhIlawvANNB_HxwbHj7kwI,3151
 atomicshop/mitm/shared_functions.py,sha256=0lzeyINd44sVEfFbahJxQmz6KAMWbYrW5ou3UYfItvw,1777
 atomicshop/mitm/statistic_analyzer.py,sha256=5_sAYGX2Xunzo_pS2W5WijNCwr_BlGJbbOO462y_wN4,27533
@@ -233,7 +233,7 @@ atomicshop/wrappers/factw/fact_extractor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 atomicshop/wrappers/factw/fact_extractor/docker_image.py,sha256=2FyYjnw8gxFNwISQ83OwH-iGivkFi6EAluyCZ0loHEQ,2501
 atomicshop/wrappers/factw/fact_extractor/get_extractor.py,sha256=2mfOAftHIlCcGt1s7MWdq7DsDCuI6wX3MtvcEZ4SK-0,756
 atomicshop/wrappers/factw/install/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/wrappers/factw/install/install_after_restart.py,sha256=cr8E9BAEdpJXJ1VdyXr85byU0PvE3tPV0uPALQOmiKg,4318
+atomicshop/wrappers/factw/install/install_after_restart.py,sha256=4dHn2XMbYaPJlhRCmZqqwsgHQBlG2mT7aW50pQCPtp4,4345
 atomicshop/wrappers/factw/install/pre_install_and_install_before_restart.py,sha256=GFsO9MTH0czKoxkiPJtjalilUwsmFLBCcx9Znv37S4M,5945
 atomicshop/wrappers/factw/postgresql/__init__.py,sha256=xMBn2d3Exo23IPP2F_9-SXmOlhFbwWDgS9KwozSTjA0,162
 atomicshop/wrappers/factw/postgresql/analysis.py,sha256=2Rxzy2jyq3zEKIo53z8VkjuslKE_i5mq2ZpmJAvyd6U,716
@@ -260,7 +260,7 @@ atomicshop/wrappers/loggingw/loggingw.py,sha256=uLY7DJS-3xIYQBRvI--9eFvdcnvsWSXm
 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=pmI9AwWJ2cv5h8GionSpSJkllg6kfp0M381pk6y4Y5U,4015
-atomicshop/wrappers/mongodbw/install_mongodb_win.py,sha256=3ZPqrXxj3lC-PnAKGXclylLuOqsbyXYeUpb5iGjdeUU,6626
+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/mongodbw.py,sha256=IkEw86QFyVRU-5p5s6_6yupvSxmaQxr59GKNgSEkAm4,52617
 atomicshop/wrappers/nodejsw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -319,8 +319,8 @@ atomicshop/wrappers/socketw/ssl_base.py,sha256=kmiif84kMhBr5yjQW17p935sfjR5JKG0L
 atomicshop/wrappers/socketw/statistics_csv.py,sha256=fgMzDXI0cybwUEqAxprRmY3lqbh30KAV-jOpoFKT-m8,3395
 atomicshop/wrappers/winregw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/winregw/winreg_network.py,sha256=zZQfps-CdODQaTUADbHAwKHr5RUg7BLafnKWBbKaLN4,8728
-atomicshop-2.18.3.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-2.18.3.dist-info/METADATA,sha256=9sxUGrK_-bQdIzS6k4nNclLdMORnzGbWBI4vqP7c378,10499
-atomicshop-2.18.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-2.18.3.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-2.18.3.dist-info/RECORD,,
+atomicshop-2.18.4.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-2.18.4.dist-info/METADATA,sha256=Eabxa6pWlDm-BYBMU-rJ_YCVnxKylpxEYJ2GmbskdZM,10499
+atomicshop-2.18.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-2.18.4.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-2.18.4.dist-info/RECORD,,