add colors to print statements

Author: cadenmyers13

src/diffpy/cmi/packsmanager.py

@@ -28,6 +28,20 @@
 __all__ = ["PacksManager", "get_package_dir"]
 
 
+class Styles:
+    RESET = "\033[0m"
+    # styles
+    BOLD = "\033[1m"
+    UNDER = "\033[4m"
+    # colors
+    RED = "\033[31m"
+    GREEN = "\033[32m"
+    YELLOW = "\033[33m"
+    BLUE = "\033[34m"
+    MAGENTA = "\033[35m"
+    CYAN = "\033[36m"
+
+
 def get_package_dir(root_path=None):
     """Get the package directory as a context manager.
 
@@ -353,20 +367,19 @@ def install_pack(self, identifier: str | Path) -> None:
     def print_packs(self) -> None:
         """Print information about available packs."""
         uninstalled_packs, installed_packs = [], []
+        s = Styles()
         for pack in self.available_packs():
             if self.check_pack(pack):
                 installed_packs.append(pack)
             else:
                 uninstalled_packs.append(pack)
-        print("Installed Packs:")
-        print("----------------")
+        print(f"{s.BOLD}{s.UNDER}{s.BLUE}Installed Packs:{s.RESET}")
         for pack in installed_packs:
             if not installed_packs:
                 print("  (none)")
             else:
                 print(f"  {pack}")
-        print("\nAvailable Packs:")
-        print("----------------")
+        print(f"\n{s.BOLD}{s.UNDER}{s.BLUE}Available Packs:{s.RESET}")
         if not uninstalled_packs:
             print("  (all packs installed)")
         else:
@@ -375,11 +388,11 @@ def print_packs(self) -> None:
 
     def print_examples(self) -> None:
         """Print information about available examples."""
-        print("\nExamples:")
-        print("---------")
+        s = Styles()
+        print(f"\n{s.BOLD}{s.UNDER}{s.CYAN}Examples:{s.RESET}")
         examples_dict = self.available_examples()
         for pack, examples in examples_dict.items():
-            print(f"  {pack}:")
+            print(f"  {s.BOLD}{pack}:{s.RESET}")
             for ex_name, _ in examples:
                 print(f"   - {ex_name}")
 

src/diffpy/cmi/profilesmanager.py

@@ -26,7 +26,7 @@
     presence_check,
 )
 from diffpy.cmi.log import plog
-from diffpy.cmi.packsmanager import PacksManager, get_package_dir
+from diffpy.cmi.packsmanager import PacksManager, Styles, get_package_dir
 
 __all__ = ["Profile", "ProfilesManager"]
 
@@ -234,21 +234,20 @@ def install(self, identifier: Union[str, Path]) -> None:
 
     def print_profiles(self) -> None:
         """Print available and installed profiles."""
+        s = Styles()
         installed_profiles, uninstalled_profiles = [], []
         for profile_name in self.available_profiles():
             if self.check_profile(profile_name):
                 installed_profiles.append(profile_name)
             else:
                 uninstalled_profiles.append(profile_name)
-        print("\nInstalled Profiles:")
-        print("-------------------")
+        print(f"\n{s.BOLD}{s.UNDER}{s.MAGENTA}Installed Profiles:{s.RESET}")
         if not installed_profiles:
             print("  (none)")
         else:
             for profile in installed_profiles:
                 print(f"  {profile}")
-        print("\nAvailable Profiles:")
-        print("-------------------")
+        print(f"\n{s.BOLD}{s.UNDER}{s.MAGENTA}Available Profiles:{s.RESET}")
         if not uninstalled_profiles:
             print("  (all profiles installed)")
         else: