runbooks 0.7.7__py3-none-any.whl → 0.9.0__py3-none-any.whl

runbooks/inventory/list_ec2_instances.py

@@ -373,9 +373,17 @@ if __name__ == "__main__":
     logging.getLogger("s3transfer").setLevel(logging.CRITICAL)
     logging.getLogger("urllib3").setLevel(logging.CRITICAL)
 
-    print()
-    print(f"Checking for instances... ")
-    print()
+    # Import Rich display utilities for professional output
+    from runbooks.inventory.rich_inventory_display import (
+        create_inventory_progress,
+        display_ec2_inventory_results,
+        display_inventory_header,
+    )
+
+    # Display professional inventory header
+    display_inventory_header(
+        "EC2", pProfiles, AccountNum if "AccountNum" in locals() else 0, RegionNum if "RegionNum" in locals() else 0
+    )
 
     # Find credentials for all Child Accounts
     # CredentialList = get_credentials(pProfiles, pRegionList, pSkipProfiles, pSkipAccounts, pRootOnly, pAccounts, pAccessRoles, pTiming)
@@ -384,16 +392,15 @@ if __name__ == "__main__":
     )
     AccountNum = len(set([acct["AccountId"] for acct in CredentialList]))
     RegionNum = len(set([acct["Region"] for acct in CredentialList]))
-    print()
-    print(f"Searching total of {AccountNum} accounts and {RegionNum} regions")
+
+    # Update header with actual counts
+    display_inventory_header("EC2", pProfiles, AccountNum, RegionNum)
+
     if pTiming:
-        print()
         milestone_time1 = time()
-        print(
-            f"{Fore.GREEN}\t\tFiguring out what regions are available to your accounts, and capturing credentials for all accounts in those regions took: {(milestone_time1 - begin_time):.3f} seconds{Fore.RESET}"
-        )
-        print()
-    print(f"Now running through all accounts and regions identified to find resources...")
+        from runbooks.common.rich_utils import print_info
+
+        print_info(f"⏱️ Credential discovery completed in {(milestone_time1 - begin_time):.3f} seconds")
     # Collect all the instances from the credentials found
     AllInstances = find_all_instances(CredentialList, pStatus)
     # Display the information we've found thus far
@@ -414,12 +421,6 @@ if __name__ == "__main__":
     )
     display_results(sorted_all_instances, display_dict, None, pFilename)
 
-    if pTiming:
-        print(ERASE_LINE)
-        print(f"{Fore.GREEN}This script took {time() - begin_time:.2f} seconds{Fore.RESET}")
-    print(ERASE_LINE)
-
-    print(f"Found {len(AllInstances)} instances across {AccountNum} accounts across {RegionNum} regions")
-    print()
-    print("Thank you for using this script")
-    print()
+    # Display results using Rich formatting
+    timing_info = {"total_time": time() - begin_time} if pTiming else None
+    display_ec2_inventory_results(AllInstances, AccountNum, RegionNum, timing_info)

runbooks/inventory/list_ssm_parameters.py

@@ -212,8 +212,23 @@ def find_ssm_parameters(f_credentialList):
         - Logs authorization issues for security team follow-up
     """
     parameter_list = []
-    print(f"Gathering parameters from {len(f_credentialList)} accounts and regions")
-    for credential in tqdm(f_credentialList, desc="Gathering SSM Parameters", leave=True):
+    # Import Rich display utilities for professional output
+    from runbooks.common.rich_utils import console, print_info, print_success
+    from runbooks.inventory.rich_inventory_display import create_inventory_progress, display_inventory_header
+
+    # Calculate operation scope
+    account_count = len(set([cred["AccountId"] for cred in f_credentialList]))
+    region_count = len(set([cred["Region"] for cred in f_credentialList]))
+
+    # Display professional header
+    display_inventory_header("SSM Parameters", "multi-profile", account_count, region_count)
+
+    # Create Rich progress bar
+    progress = create_inventory_progress(len(f_credentialList), "🔑 Discovering SSM Parameters")
+    task = progress.add_task("Processing credentials", total=len(f_credentialList))
+    progress.start()
+
+    for credential in f_credentialList:
         try:
             # Call SSM API to discover all parameters in this account/region combination
             # Note: Parameter stores can contain 10,000+ parameters - this operation may take time
@@ -228,6 +243,15 @@ def find_ssm_parameters(f_credentialList):
                 logging.error(
                     f"Profile {credential['Profile']}: Authorization Failure for account {credential['AccountNumber']}"
                 )
+        finally:
+            # Update progress
+            progress.update(task, advance=1)
+
+    progress.stop()
+
+    # Display completion summary
+    print_success(f"✅ SSM Parameter discovery completed! Found {len(parameter_list)} parameters total")
+
     return parameter_list
 
 
@@ -271,7 +295,11 @@ if __name__ == "__main__":
     # Define AWS Landing Zone (ALZ) parameter pattern for UUID-based identification
     # Pattern matches: /UUID/numeric-suffix (e.g., /2ac07efd-153d-4069-b7ad-0d18cc398b11/105)
     ALZRegex = r"/\w{8,8}-\w{4,4}-\w{4,4}-\w{4,4}-\w{12,12}/\w{3,3}"
-    print()
+    # Import Rich utilities at module level
+    from runbooks.common.rich_utils import console, print_header
+
+    # Display module header
+    print_header("SSM Parameter Store Discovery", "0.7.8")
 
     # Execute enterprise credential discovery across organizational hierarchy
     CredentialList = get_all_credentials(