runbooks 1.1.5__py3-none-any.whl → 1.1.7__py3-none-any.whl

runbooks/inventory/find_ec2_security_groups.py

@@ -62,18 +62,17 @@ from time import time
 
 from ArgumentsClass import CommonArguments
 from botocore.exceptions import ClientError
-from colorama import Fore, init
+from runbooks.common.rich_utils import console
 from Inventory_Modules import (
     display_results,
     find_references_to_security_groups2,
     find_security_groups2,
     get_all_credentials,
 )
-from tqdm.auto import tqdm
+# Migrated to Rich.Progress - see rich_utils.py for enterprise UX standards
+# from tqdm.auto import tqdm
 
-init()
 __version__ = "2024.09.24"
-ERASE_LINE = "\x1b[2K"
 begin_time = time()
 
 
@@ -387,7 +386,7 @@ def check_accounts_for_security_groups(
                     logging.info(
                         f"{ERASE_LINE}Finished finding security groups in account {c_account_credentials['AccountId']} in region {c_account_credentials['Region']}"
                     )
-                    pbar.update()
+                    pbar.update(pbar_task, advance=1)  # Update Rich progress bar
                     self.queue.task_done()
 
     ###########
@@ -406,43 +405,50 @@ def check_accounts_for_security_groups(
     # Initialize thread-safe work queue
     checkqueue = Queue()
 
-    # Initialize progress bar for user feedback during long-running analysis
-    pbar = tqdm(
-        desc=f"Finding security groups from {len(fCredentialList)} locations",
-        total=len(fCredentialList),
-        unit=" locations",
-    )
+    # Import Rich display utilities for professional progress tracking
+    from runbooks.common.rich_utils import create_progress_bar
+
+    # Initialize progress tracking for user feedback during long-running analysis
+    with create_progress_bar() as progress:
+        task = progress.add_task(
+            f"[cyan]Finding security groups from {len(fCredentialList)} locations...",
+            total=len(fCredentialList)
+        )
+
+        # Make progress object available to worker threads via global (multi-threaded pattern)
+        global pbar
+        pbar = progress
+        global pbar_task
+        pbar_task = task
+
+        # Start worker thread pool for concurrent security group analysis
+        for x in range(WorkerThreads):
+            worker = FindSecurityGroups(checkqueue)
+            # Daemon threads will terminate when main thread exits
+            # This prevents hanging if an exception occurs in main thread
+            worker.daemon = True
+            worker.start()
+
+        # Queue all credential work items for concurrent processing
+        for credential in fCredentialList:
+            logging.info(f"Connecting to account {credential['AccountId']}")
+            try:
+                # Add credential set and parameters to work queue for worker thread processing
+                checkqueue.put((credential, fFragment, fExact, fDefault))
+            except ClientError as my_Error:
+                # Handle credential validation errors during queue population
+                if "AuthFailure" in str(my_Error):
+                    logging.error(
+                        f"Authorization Failure accessing account {credential['AccountId']} in '{credential['Region']}' region"
+                    )
+                    logging.warning(f"It's possible that the region '{credential['Region']}' hasn't been opted-into")
+                    # Continue queuing other credentials despite this failure
+                    pass
 
-    # Start worker thread pool for concurrent security group analysis
-    for x in range(WorkerThreads):
-        worker = FindSecurityGroups(checkqueue)
-        # Daemon threads will terminate when main thread exits
-        # This prevents hanging if an exception occurs in main thread
-        worker.daemon = True
-        worker.start()
-
-    # Queue all credential work items for concurrent processing
-    for credential in fCredentialList:
-        logging.info(f"Connecting to account {credential['AccountId']}")
-        try:
-            # Add credential set and parameters to work queue for worker thread processing
-            checkqueue.put((credential, fFragment, fExact, fDefault))
-        except ClientError as my_Error:
-            # Handle credential validation errors during queue population
-            if "AuthFailure" in str(my_Error):
-                logging.error(
-                    f"Authorization Failure accessing account {credential['AccountId']} in '{credential['Region']}' region"
-                )
-                logging.warning(f"It's possible that the region '{credential['Region']}' hasn't been opted-into")
-                # Continue queuing other credentials despite this failure
-                pass
-
-    # Wait for all queued work items to be processed by worker threads
-    # This blocks until all worker threads call task_done()
-    checkqueue.join()
-
-    # Clean up progress bar
-    pbar.close()
+        # Wait for all queued work items to be processed by worker threads
+        # This blocks until all worker threads call task_done()
+        checkqueue.join()
+        # Progress bar auto-closes when exiting context manager
 
     return AllSecurityGroups
 
@@ -659,7 +665,7 @@ if __name__ == "__main__":
         print(f"Data has been saved to {saved_filename}")
     if pTiming:
         print(ERASE_LINE)
-        print(f"{Fore.GREEN}This script took {time() - begin_time:.2f} seconds{Fore.RESET}")
+        print(f"[green]This script took {time() - begin_time:.2f} seconds")
 
     print(
         f"We found {len(AllSecurityGroups)} {'default ' if pDefault else ''}security group{'' if len(AllSecurityGroups) == 1 else 's'} across {len(AccountList)} accounts and {len(RegionList)} regions"

runbooks/inventory/find_landingzone_versions.py

@@ -76,10 +76,9 @@ import boto3
 import Inventory_Modules
 from ArgumentsClass import CommonArguments
 from botocore.exceptions import ClientError, CredentialRetrievalError, InvalidConfigError
-from colorama import init
+# colorama removed - migrated to Rich
 
 # Initialize colorama for cross-platform colored terminal output
-init()
 
 __version__ = "2023.05.31"
 
@@ -101,7 +100,6 @@ logging.basicConfig(
 
 ##########################
 # Terminal control and operational configuration constants
-ERASE_LINE = "\x1b[2K"  # ANSI escape sequence for clearing terminal line during progress display
 SkipProfiles = ["default"]  # Profile exclusion list for organizational Landing Zone discovery
 
 # Configure AWS profile discovery strategy based on user input
@@ -129,7 +127,7 @@ ALZProfiles = []  # Initialize Landing Zone profile collection for discovered Ma
 
 for profile in AllProfiles:
     # Display real-time progress during profile analysis with terminal line clearing
-    print(f"{ERASE_LINE}Checking profile: {profile}", end="\r")
+    print(f"Checking profile: {profile}", end="\r")
 
     try:
         # Analyze current profile to determine if it represents an AWS Landing Zone Management Account
@@ -164,7 +162,7 @@ for profile in AllProfiles:
             pass
 
 # Clear progress display and initialize tabular output formatting for Landing Zone inventory
-print(ERASE_LINE)
+console.print()
 fmt = "%-20s %-13s %-15s %-35s %-21s"  # Column formatting for structured Landing Zone data display
 print(fmt % ("Profile", "Account", "Region", "ALZ Stack Name", "ALZ Version"))
 print(fmt % ("-------", "-------", "------", "--------------", "-----------"))
@@ -194,7 +192,7 @@ for item in ALZProfiles:
                     )
 
 # Display comprehensive operational summary with discovery metrics
-print(ERASE_LINE)
+console.print()
 print(f"Checked {len(AllProfiles)} accounts/ Orgs. Found {len(ALZProfiles)} ALZs")
 print()
 print("Thank you for using this script.")

runbooks/inventory/find_vpc_flow_logs.py

@@ -93,13 +93,11 @@ from account_class import aws_acct_access
 from ArgumentsClass import CommonArguments
 from botocore.config import Config
 from botocore.exceptions import ClientError
-from colorama import Fore, init
+from runbooks.common.rich_utils import console
 from Inventory_Modules import RemoveCoreAccounts, display_results, get_all_credentials, get_regions3
 
 # Initialize colorama for cross-platform colored terminal output
-init()
 __version__ = "2024.03.10"
-ERASE_LINE = "\x1b[2K"  # ANSI escape sequence for clearing terminal line during progress display
 begin_time = time()  # Script execution timing for performance monitoring
 sleep_interval = 5  # Default wait interval for CloudWatch Logs query processing
 
@@ -305,13 +303,13 @@ def setup_auth_accounts_and_regions(fProfile: str) -> (aws_acct_access, list, li
 
     # Display comprehensive operational context for user confirmation and audit logging
     print(f"You asked to sum flow log data")
-    print(f"\tin these accounts: {Fore.RED}{AccountList}{Fore.RESET}")
-    print(f"\tin these regions: {Fore.RED}{RegionList}{Fore.RESET}")
+    print(f"\tin these accounts: [red]{AccountList}")
+    print(f"\tin these regions: [red]{RegionList}")
     print(f"\tFrom: {pStartDate} until {pEndDate}")
 
     # Display account exclusion information for operational transparency
     if pSkipAccounts is not None:
-        print(f"\tWhile skipping these accounts: {Fore.RED}{pSkipAccounts}{Fore.RESET}")
+        print(f"\tWhile skipping these accounts: [red]{pSkipAccounts}")
 
     return aws_acct, AccountList, RegionList
 
@@ -1040,13 +1038,13 @@ def get_cw_query_results(fquery_requests: list) -> list[dict]:
             # Implement timeout protection for long-running queries
             if waited_seconds_total > (SpannedDaysChecked * 5):
                 print(
-                    f"{ERASE_LINE}Query is still running... Waited {waited_seconds_total} seconds already, we'll have to check manually later. "
+                    f"Query is still running... Waited {waited_seconds_total} seconds already, we'll have to check manually later. "
                 )
                 break
 
             # Display real-time progress for query execution monitoring
             print(
-                f"{ERASE_LINE}Query for vpc {query['VPCId']} in account {query['AccountId']} in region {query['Region']} is still running... It's been {waited_seconds_total} seconds so far",
+                f"Query for vpc {query['VPCId']} in account {query['AccountId']} in region {query['Region']} is still running... It's been {waited_seconds_total} seconds so far",
                 end="\r",
             )
 
@@ -1181,7 +1179,7 @@ if __name__ == "__main__":
                 f"Account {credential['AccountId']} was successfully connected via role {credential.get('Role', pAccessRole)} from {aws_acct.acct_number}"
             )
             print(
-                f"{ERASE_LINE}Checking account {Fore.BLUE}{credential['AccountId']}{Fore.RESET} in region {Fore.BLUE}{credential['Region']}{Fore.RESET}...",
+                f"Checking account [blue]{credential['AccountId']} in region [blue]{credential['Region']}...",
                 end="\r",
             )
             """

runbooks/inventory/inventory_modules.py

@@ -455,12 +455,12 @@ def print_timings(fTiming: bool = False, fverbose: int = 50, fbegin_time=None, f
     """
     from time import time
 
-    from colorama import Fore, init
+    from runbooks.common.rich_utils import console
 
     init()
 
     if fTiming and fverbose < 50 and fbegin_time is not None:
-        print(f"{Fore.GREEN}{fmessage}\nThis script has taken {time() - fbegin_time:.6f} seconds so far{Fore.RESET}")
+        print(f"[green]{fmessage}\nThis script has taken {time() - fbegin_time:.6f} seconds so far")
 
 
 def make_creds(faws_acct):
@@ -4241,12 +4241,14 @@ def find_stacksets3(
         from queue import Queue
         from threading import Thread
 
-        from tqdm.auto import tqdm
+        from runbooks.common.rich_utils import create_progress_bar
 
         class GetStackSetStatus(Thread):
-            def __init__(self, queue):
+            def __init__(self, queue, progress, task_id):
                 Thread.__init__(self)
                 self.queue = queue
+                self.progress = progress
+                self.task_id = task_id
 
             def run(self):
                 while True:
@@ -4317,7 +4319,7 @@ def find_stacksets3(
                         )
                         continue
                     finally:
-                        pbar.update()
+                        self.progress.update(self.task_id, advance=1)
                         self.queue.task_done()
 
         ###########
@@ -4328,35 +4330,35 @@ def find_stacksets3(
         WorkerThreads = min(len(fStackSetsCopy), MaxWorkerThreads)
         logging.info(f"Using {WorkerThreads} threads")
 
-        pbar = tqdm(
-            desc=f"Finding all Stacksets from {len(fStackSetsCopy)} stacksets",
-            total=len(fStackSetsCopy),
-            unit=" stacksets",
-        )
+        with create_progress_bar() as progress:
+            task = progress.add_task(
+                "[cyan]Finding all Stacksets...",
+                total=len(fStackSetsCopy)
+            )
 
-        for x in range(WorkerThreads):
-            worker = GetStackSetStatus(checkqueue)
-            # Setting daemon to True will let the main thread exit even though the workers are blocking
-            worker.daemon = True
-            worker.start()
+            for x in range(WorkerThreads):
+                worker = GetStackSetStatus(checkqueue, progress, task)
+                # Setting daemon to True will let the main thread exit even though the workers are blocking
+                worker.daemon = True
+                worker.start()
+
+            for stackset in fStackSetsCopy:
+                logging.debug(f"Beginning to queue data - starting with {stackset['StackSetName']}")
+                try:
+                    # I don't know why - but double parens are necessary below. If you remove them, only the first parameter is queued.
+                    PlaceCount += 1
+                    # print(".", end='')
+                    checkqueue.put((stackset, fRegion, PlaceCount))
+                except ClientError as my_Error:
+                    if "AuthFailure" in str(my_Error):
+                        logging.error(
+                            f"Authorization Failure accessing stack set {stackset['StackSetName']} in {fRegion} region"
+                        )
+                        logging.warning(f"It's possible that the region {fRegion} hasn't been opted-into")
+                        pass
+            checkqueue.join()
+            logging.info(f"Getting the stackset operation data took {time() - begin_time:.2f} seconds")
 
-        for stackset in fStackSetsCopy:
-            logging.debug(f"Beginning to queue data - starting with {stackset['StackSetName']}")
-            try:
-                # I don't know why - but double parens are necessary below. If you remove them, only the first parameter is queued.
-                PlaceCount += 1
-                # print(".", end='')
-                checkqueue.put((stackset, fRegion, PlaceCount))
-            except ClientError as my_Error:
-                if "AuthFailure" in str(my_Error):
-                    logging.error(
-                        f"Authorization Failure accessing stack set {stackset['StackSetName']} in {fRegion} region"
-                    )
-                    logging.warning(f"It's possible that the region {fRegion} hasn't been opted-into")
-                    pass
-        checkqueue.join()
-        logging.info(f"Getting the stackset operation data took {time() - begin_time:.2f} seconds")
-        pbar.close()
         return fStackSetsCopy
 
     # Logging Settings
@@ -5341,7 +5343,7 @@ def display_results(
 ):
     from datetime import datetime
 
-    from colorama import Fore, init
+    from runbooks.common.rich_utils import console
 
     init()
     """
@@ -5757,12 +5759,12 @@ def get_all_credentials(
     from .account_class import aws_acct_access
 
     # from time import time
-    from colorama import Fore, init
+    from runbooks.common.rich_utils import console
 
     init()
     # ERASE_LINE = '\x1b[2K'
     # begin_time = time()
-    print(f"{Fore.GREEN}Timing is enabled{Fore.RESET}") if fTiming else None
+    print(f"[green]Timing is enabled") if fTiming else None
 
     AllCredentials = []
     if fSkipProfiles is None:
@@ -5848,16 +5850,18 @@ def get_credentials_for_accounts_in_org(
     from time import time
 
     from botocore.exceptions import ClientError
-    from colorama import Fore, init
-    from tqdm.auto import tqdm
+    from runbooks.common.rich_utils import console
+    from runbooks.common.rich_utils import create_progress_bar
 
     init()
     begin_time = time()
 
     class AssembleCredentials(Thread):
-        def __init__(self, queue):
+        def __init__(self, queue, progress, task_id):
             Thread.__init__(self)
             self.queue = queue
+            self.progress = progress
+            self.task_id = task_id
 
         def run(self):
             while True:
@@ -5914,7 +5918,7 @@ def get_credentials_for_accounts_in_org(
                     logging.error(f"Error: Likely that one of the supplied profiles was wrong\nError: {my_Error}")
                     continue
                 finally:
-                    pbar.update()
+                    self.progress.update(self.task_id, advance=1)
                     self.queue.task_done()
 
     if fSkipAccounts is None:
@@ -5959,47 +5963,48 @@ def get_credentials_for_accounts_in_org(
     # Defaults to 50, unless something more was passed in - which is only done for time testing.
     WorkerThreads = min(len(ChildAccounts) * len(fregions), MaxThreads)
 
-    # Create x worker threads
-    for x in range(WorkerThreads):
-        worker = AssembleCredentials(credqueue)
-        # Setting daemon to True will let the main thread exit even though the workers are blocking
-        worker.daemon = True
-        worker.start()
-
-    pbar = tqdm(
-        desc=f"Getting credentials for profile: {fprofile} with {len(ChildAccounts)} accounts in {len(fregions)} regions",
-        total=len(ChildAccounts) * len(fregions),
-        unit=" credentials",
-    )
-
     logging.info(
         f"You asked to check {len(ChildAccounts) * len(fregions)} place{'s' if len(ChildAccounts) * len(fregions) > 1 else ''}... It's going to take a moment"
     )
-    logging.info(
-        f"{Fore.GREEN}It's taken {time() - begin_time:.2f} seconds to prep WorkerThreads and such{Fore.RESET}"
-    ) if fTiming else None
-    for account in ChildAccounts:
-        if account["AccountId"] in fSkipAccounts:
-            continue
-        elif fRootOnly and not account["AccountId"] == account["MgmtAccount"]:
-            continue
-        elif accountlist and account["AccountId"] not in accountlist:
-            continue
-        AccountNum += 1
-        logging.info(f"Queuing account info for {AccountNum} / {len(ChildAccounts)} accounts in profile {fprofile}")
-        RegionNum = 0
-        for region in fregions:
-            RegionNum += 1
-            logging.info(f"\t\tRegion {RegionNum} of {len(fregions)}")
-            credqueue.put((account, fprofile, region))
-            logging.info(f"Account / Region: {account} / {region} | {datetime.now()}")
-            logging.info(f"Queue Size: {credqueue.qsize()}")
-    print(
-        f"{Fore.GREEN}Enumerating {AccountNum} account{'s' if len(ChildAccounts) * len(fregions) > 1 else ''} and {len(fregions)} regions "
-        f"took {time() - begin_time:.3f} seconds {Fore.RESET}"
-    ) if fTiming else None
-    credqueue.join()
-    pbar.close()
+
+    with create_progress_bar() as progress:
+        task = progress.add_task(
+            "[cyan]Getting credentials...",
+            total=len(ChildAccounts) * len(fregions)
+        )
+
+        # Create x worker threads
+        for x in range(WorkerThreads):
+            worker = AssembleCredentials(credqueue, progress, task)
+            # Setting daemon to True will let the main thread exit even though the workers are blocking
+            worker.daemon = True
+            worker.start()
+
+        logging.info(
+            f"[green]It's taken {time() - begin_time:.2f} seconds to prep WorkerThreads and such"
+        ) if fTiming else None
+        for account in ChildAccounts:
+            if account["AccountId"] in fSkipAccounts:
+                continue
+            elif fRootOnly and not account["AccountId"] == account["MgmtAccount"]:
+                continue
+            elif accountlist and account["AccountId"] not in accountlist:
+                continue
+            AccountNum += 1
+            logging.info(f"Queuing account info for {AccountNum} / {len(ChildAccounts)} accounts in profile {fprofile}")
+            RegionNum = 0
+            for region in fregions:
+                RegionNum += 1
+                logging.info(f"\t\tRegion {RegionNum} of {len(fregions)}")
+                credqueue.put((account, fprofile, region))
+                logging.info(f"Account / Region: {account} / {region} | {datetime.now()}")
+                logging.info(f"Queue Size: {credqueue.qsize()}")
+        print(
+            f"[green]Enumerating {AccountNum} account{'s' if len(ChildAccounts) * len(fregions) > 1 else ''} and {len(fregions)} regions "
+            f"took {time() - begin_time:.3f} seconds "
+        ) if fTiming else None
+        credqueue.join()
+
     return AllCreds
 
 
@@ -6015,19 +6020,21 @@ def get_org_accounts_from_profiles(fProfileList):
 
     from .account_class import aws_acct_access
     from botocore.exceptions import ClientError, InvalidConfigError, NoCredentialsError
-    from tqdm.auto import tqdm
+    from runbooks.common.rich_utils import create_progress_bar
 
     class AssembleCredentials(Thread):
-        def __init__(self, queue):
+        def __init__(self, queue, progress, task_id):
             Thread.__init__(self)
             self.queue = queue
+            self.progress = progress
+            self.task_id = task_id
 
         def run(self):
             # Account = dict()
             while True:
                 # Get the work from the queue and expand the tuple
                 profile = self.queue.get()
-                pbar.update()
+                self.progress.update(self.task_id, advance=1)
                 Account = {
                     "ErrorFlag": False,
                     "Success": False,
@@ -6111,17 +6118,22 @@ def get_org_accounts_from_profiles(fProfileList):
     # WorkerThreads = len(fProfileList)
     WorkerThreads = 2
 
-    # Create x worker threads
-    for x in range(WorkerThreads):
-        worker = AssembleCredentials(profilequeue)
-        # Setting daemon to True will let the main thread exit even though the workers are blocking
-        worker.daemon = True
-        worker.start()
+    with create_progress_bar() as progress:
+        task = progress.add_task(
+            f"[cyan]Getting accounts from {len(fProfileList)} profiles...",
+            total=len(fProfileList)
+        )
+
+        # Create x worker threads
+        for x in range(WorkerThreads):
+            worker = AssembleCredentials(profilequeue, progress, task)
+            # Setting daemon to True will let the main thread exit even though the workers are blocking
+            worker.daemon = True
+            worker.start()
 
-    pbar = tqdm(desc=f"Getting accounts from {len(fProfileList)} profiles", total=len(fProfileList))
+        for profile_item in fProfileList:
+            logging.info(f"Queuing profile {profile_item} / {len(fProfileList)} profiles")
+            profilequeue.put(profile_item)
+        profilequeue.join()
 
-    for profile_item in fProfileList:
-        logging.info(f"Queuing profile {profile_item} / {len(fProfileList)} profiles")
-        profilequeue.put(profile_item)
-    profilequeue.join()
     return AllAccounts

runbooks/inventory/list_cfn_stacks.py

@@ -67,10 +67,9 @@ import Inventory_Modules
 from account_class import aws_acct_access
 from ArgumentsClass import CommonArguments
 from botocore.exceptions import ClientError
-from colorama import Fore, init
+from runbooks.common.rich_utils import console
 from Inventory_Modules import display_results, get_all_credentials
 
-init()
 
 __version__ = "2024.05.31"
 
@@ -230,16 +229,16 @@ def setup_auth_accounts_and_regions(
     else:
         AccountList = [account["AccountId"] for account in ChildAccounts if account["AccountId"] in fAccountList]
 
-    print(f"You asked to find stacks with this fragment {Fore.RED}'{fStackFrag}'{Fore.RESET}")
-    print(f"in these accounts:\n{Fore.RED}{AccountList}{Fore.RESET}")
-    print(f"in these regions:\n{Fore.RED}{RegionList}{Fore.RESET}")
-    print(f"While skipping these accounts:\n{Fore.RED}{fSkipAccounts}{Fore.RESET}") if fSkipAccounts is not None else ""
+    print(f"You asked to find stacks with this fragment [red]'{fStackFrag}'")
+    print(f"in these accounts:\n[red]{AccountList}")
+    print(f"in these regions:\n[red]{RegionList}")
+    print(f"While skipping these accounts:\n[red]{fSkipAccounts}") if fSkipAccounts is not None else ""
     if fDeletionRun:
         print()
         print("And delete the stacks that are found...")
 
     if fExact:
-        print(f"\t\tFor stacks that {Fore.RED}exactly match{Fore.RESET} these fragments: {fStackFrag}")
+        print(f"\t\tFor stacks that [red]exactly match these fragments: {fStackFrag}")
     else:
         print(f"\t\tFor stacks that contains these fragments: {fStackFrag}")
 
@@ -330,7 +329,7 @@ def collect_cfnstacks(fCredentialList: list) -> list:
 
                     # Display real-time progress with colored output
                     print(
-                        f"{ERASE_LINE}{Fore.RED}Account: {credential['AccountId']} Region: {credential['Region']} Found {len(Stacks)} Stacks{Fore.RESET}",
+                        f"{ERASE_LINE}[red]Account: {credential['AccountId']} Region: {credential['Region']} Found {len(Stacks)} Stacks",
                         end="\r",
                     )
 
@@ -410,7 +409,7 @@ def display_stacks(fAllStacks: list):
     )
     print(ERASE_LINE)
     print(
-        f"{Fore.RED}Found {len(fAllStacks)} stacks across {len(AccountList)} accounts across {len(RegionList)} regions{Fore.RESET}"
+        f"[red]Found {len(fAllStacks)} stacks across {len(AccountList)} accounts across {len(RegionList)} regions"
     )
     print()
     if args.loglevel < 21:  # INFO level
@@ -551,7 +550,7 @@ if __name__ == "__main__":
 
     if pTiming:
         print(ERASE_LINE)
-        print(f"{Fore.GREEN}This script took {time() - begin_time:.2f} seconds{Fore.RESET}")
+        print(f"[green]This script took {time() - begin_time:.2f} seconds")
 
     print()
     print("Thanks for using this script...")

runbooks/inventory/list_cfn_stackset_operation_results.py

@@ -76,9 +76,8 @@ import logging
 import re
 
 from ArgumentsClass import CommonArguments
-from colorama import Fore, init
+from runbooks.common.rich_utils import console
 
-init()
 __version__ = "2024.06.20"
 
 # Configure CLI argument parsing for StackSet results analysis and correlation
@@ -123,7 +122,6 @@ logging.getLogger("urllib3").setLevel(logging.CRITICAL)  # Suppress HTTP client
 # Analysis and Data Processing
 ##########################
 
-ERASE_LINE = "\x1b[2K"  # Terminal line clearing for dynamic output updates
 
 # Initialize StackSets data structure for comprehensive deployment analysis
 StackSets = {}