From 5d54475617d3654e64e455f029456937fcc951e4 Mon Sep 17 00:00:00 2001
From: Justin Campbell <campbell.justin.tyler@gmail.com>
Date: Thu, 17 Sep 2020 01:16:57 -0400
Subject: [PATCH] Working venv reset

---
 utils/venv-manager.py | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/utils/venv-manager.py b/utils/venv-manager.py
index 7e74b80..a033b81 100644
--- a/utils/venv-manager.py
+++ b/utils/venv-manager.py
@@ -124,7 +124,7 @@ def run_logged_subprocess(command: Union[str, list], timeout: int = 10, shell: b
 
 
 def create_environment() -> int:
-    """Creates a virtual environment for webqueue2]
+    """Creates a virtual environment for webqueue2
 
     Exit Codes:
         0 = Success
@@ -184,7 +184,7 @@ def create_environment() -> int:
 
 
 def delete_environment() -> int:
-    """Deletes a virtual environment for webqueue2]
+    """Deletes a virtual environment for webqueue2
 
     Exit Codes:
         0 = Success
@@ -201,6 +201,29 @@ def delete_environment() -> int:
         logger.critical(f"Failed to delete virtual environment {VENV_DIR} at {VENV_DIR}. Exiting")
         return 5
 
+def reset_environment() -> int:
+    """Resets a virtual environment for webqueue2
+
+    Exit Codes:
+        0 = Success
+        5 = Could not delete VENV_DIR
+        10 = Could not create VENV_DIR
+
+    Returns:
+        int: Exit code
+    """
+    delete_returncode = delete_environment()
+    if delete_returncode != 0:
+        logger.critical(f"Failed to reset virtual environment {VENV_DIR} at {VENV_DIR}. Exiting")
+        return 5
+    
+    create_returncode = create_environment()
+    if create_returncode != 0:
+        logger.critical(f"Failed to reset virtual environment {VENV_DIR} at {VENV_DIR}. Exiting")
+        return 10
+
+    logger.info(f"Successfully reset virtual environment {VENV_DIR} at {VENV_DIR}. Exiting")
+
 
 if __name__ == "__main__":
     args = get_args()