From 1e05dd2cd6a2f8ac75af8c4952cec00d8b02f3a5 Mon Sep 17 00:00:00 2001
From: askiiart <benzimm07@gmail.com>
Date: Thu, 15 Dec 2022 17:38:51 -0600
Subject: [PATCH] Major improvements in error handling, & small fix

---
 docker_wrapper.py | 54 +++++++++++++++++++++++++++++++++--------------
 1 file changed, 38 insertions(+), 16 deletions(-)

diff --git a/docker_wrapper.py b/docker_wrapper.py
index 1bacada..fbb1cc0 100644
--- a/docker_wrapper.py
+++ b/docker_wrapper.py
@@ -2,9 +2,11 @@ from subprocess import getoutput, getstatusoutput
 import pprint
 import os
 
+
 class NoContainersError(Exception):
     pass
 
+
 class Docker:
     @staticmethod
     def running_containers_info():
@@ -14,12 +16,13 @@ class Docker:
         """
         raw_info = getoutput('docker ps')
         if '\n' not in raw_info:
-            raise(NoContainersError('A running Docker container is required to run this program. Please run a docker container and try again.'))
+            raise (NoContainersError(
+                'A running Docker container is required to run this program. Please run a docker container and try again.'))
         # Header: "CONTAINER ID    IMAGE    COMMAND    CREATED    STATUS    PORTS    NAMES" (with way more spaces)
         header = raw_info[:raw_info.find('\n')+1]
         header_indices = {'CONTAINER ID': header.find('CONTAINER ID'), 'IMAGE': header.find('IMAGE'),
-                        'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
-                        'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
+                          'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
+                          'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
 
         # Remove header (example above)
         raw_info = raw_info[raw_info.find('\n')+1:]
@@ -44,8 +47,7 @@ class Docker:
                     raw_info.split('\n')[i][start_i:end_i].strip()
 
         return info
-    
-    
+
     @staticmethod
     def containers():
         """
@@ -54,7 +56,8 @@ class Docker:
         """
         raw_info = getoutput('docker container list')
         if '\n' not in raw_info:
-            raise(NoContainersError('A Docker container is required to run this program. Please create a docker container and try again.'))
+            raise (NoContainersError(
+                'A Docker container is required to run this program. Please create a docker container and try again.'))
         # Remove header
         raw_info = raw_info[raw_info.find('\n')+1:]
         info = {}
@@ -63,9 +66,9 @@ class Docker:
         containers = []
         for line in raw_info.split('\n'):
             containers.append(line.strip()[line.strip().rfind(' ')+1:])
-        
+
         return containers
-    
+
     @staticmethod
     def all_containers_info():
         """
@@ -74,12 +77,13 @@ class Docker:
         """
         raw_info = getoutput('docker ps -a')
         if '\n' not in raw_info:
-            raise(NoContainersError('A Docker container is required to run this program. Please create a docker container and try again.'))
+            raise (NoContainersError(
+                'A Docker container is required to run this program. Please create a docker container and try again.'))
         # Header: "CONTAINER ID    IMAGE    COMMAND    CREATED    STATUS    PORTS    NAMES" (with way more spaces)
         header = raw_info[:raw_info.find('\n')+1]
         header_indices = {'CONTAINER ID': header.find('CONTAINER ID'), 'IMAGE': header.find('IMAGE'),
-                        'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
-                        'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
+                          'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
+                          'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
 
         # Remove header (example above)
         raw_info = raw_info[raw_info.find('\n')+1:]
@@ -116,7 +120,7 @@ class Docker:
         dict: The info about the container 
         """
         return Docker.all_containers_info()[container]
-    
+
     @staticmethod
     def compose(dir):
         """
@@ -129,10 +133,10 @@ class Docker:
         """
         cwd = os.getcwd()
         os.chdir(dir)
-        status = getstatusoutput('docker compose up -detach --build --remove-orphans')
+        status = getstatusoutput('docker compose up --detach --build --remove-orphans')
         os.chdir(cwd)
         return status
-    
+
     @staticmethod
     def start(container):
         """
@@ -144,7 +148,7 @@ class Docker:
         int: The exit code of docker start
         """
         return getstatusoutput(f'docker start {container}')[0]
-    
+
     @staticmethod
     def stop(container):
         """
@@ -156,7 +160,7 @@ class Docker:
         int: The exit code of docker stop
         """
         return getstatusoutput(f'docker stop {container}')[0]
-    
+
     @staticmethod
     def rm(container):
         """
@@ -169,6 +173,24 @@ class Docker:
         """
         return getstatusoutput(f'docker rm {container}')[0]
 
+    @staticmethod
+    def containers_exist():
+        """
+        Checks if any containers exist
+        :returns:
+        bool: True if containers exist, False otherwise
+        """
+        return True if '\n' in getoutput('docker ps -a') else False
+
+
+def aliens_exist():
+    """
+    Checks if aliens exist
+    :returns:
+    bool: Up all night long; And there's something very wrong
+    """
+    return True if 'blink-182' in 'your playlist' else False
+
 
 if __name__ == '__main__':
     try: