Fix error handling & add custom info printer

basic_management.py

@@ -1,13 +1,41 @@
-from docker_wrapper import Docker
+from docker_wrapper import Docker, NoContainersError
-from pprint import pprint
+from subprocess import getoutput
+
+
+def container_to_str(container):
+    """
+    Returns info about a Docker container as a string
+    :parameters:f
+    container: The name of the container (str)
+
+    :returns:
+    str: The container info as a string, formatted for printing
+    """
+    # Header: "CONTAINER ID    IMAGE    COMMAND    CREATED    STATUS    PORTS    NAMES" (with way more spaces)
+    info = Docker.container_info(container)
+    info_str = f'{container}\n'
+    info_str += f'  CONTAINER ID: {info["CONTAINER ID"]}\n'
+    info_str += f'  IMAGE: {info["IMAGE"]}\n'
+    info_str += f'  COMMAND: {info["COMMAND"]}\n'
+    info_str += f'  CREATED: {info["CREATED"]}\n'
+    info_str += f'  STATUS: {info["STATUS"]}\n'
+    info_str += f'  PORTS: {info["PORTS"]}\n'
+    info_str += f'  NAMES: {info["NAMES"]}'
+
+    return info_str
+
+try:
+    while True:
+        if not Docker.containers_exist():
+            raise(NoContainersError('No containers exist! Please create a container and try again.'))
 
-while True:
         # Main Menu
         print('Select the container to manage:')
+
         containers = Docker.containers()
         for i in range(len(containers)):
             print(f'  {i} - {containers[i]}')
-    print('  q - quit')
+        print('  q - Quit')
 
         container_i = input()
         print()
@@ -16,6 +44,8 @@ while True:
             exit(0)
 
         while True:
+            if not Docker.containers_exist():
+                raise(NoContainersError('No containers exist! Please create a container and try again.'))
             # Container Menu
             container = containers[int(container_i)]
             print()
@@ -33,31 +63,35 @@ while True:
                 break
 
             elif selection == 'view':
-            pprint(Docker.container_info(container))  # TODO: Make better, custom printer
+                print(container_to_str(container))
 
             elif selection == 'start':
                 print('Starting...')
-            status = Docker.start(container)
+                Docker.start(container)
 
             elif selection == 'stop':
                 print('Stopping...')
-            status = Docker.stop(container)
+                Docker.stop(container)
-            print('Done.')
+                print('Done')
 
             elif selection == 'rm':
                 break_later = False
                 while selection != 'y' and selection != 'n' and selection != 'Y' and selection != 'N':
                     print(f'WARNING! This will DELETE {container}!')
-                print(f'Are you absolutely sure you want to delete {container}? (y/N)')
+                    print(
+                        f'Are you absolutely sure you want to delete {container}? (y/N)')
                     selection = input()
                     if selection == 'y' or selection == 'Y':
+                        print('Stopping...')
+                        Docker.stop(container)
                         print('Deleting...')
-                    status = Docker.rm(container)
+                        Docker.rm(container)
                         print('Done')
                         break_later = True
                     elif selection == 'n' or selection == 'N':
                         print(f'Operation cancelled, returning to {container} menu...')
                 if break_later:
+                    print()
                     break
 
             elif selection == 'menu':
@@ -67,3 +101,7 @@ while True:
             else:
                 print('Selection invalid. Please try again.')
             print()
+except NoContainersError as e:
+    print('Error:', e)
+    print('No containers found. Exiting...')
+    exit()