Add exit imports

terrabutler/__main__.py

@@ -1,5 +1,5 @@
-import sys
+from sys import exit
 from terrabutler.click import main
 
 if __name__ == "__main__":
-    sys.exit(main())
+    exit(main())

terrabutler/click.py

@@ -6,6 +6,7 @@
 
 import click
 from colorama import Fore
+from sys import exit
 from os import path
 from terrabutler.__init__ import (
     __name__,

terrabutler/env.py

@@ -3,6 +3,7 @@
 from terrabutler.utils import paths
 from click import confirm
 from colorama import Fore
+from sys import exit
 import boto3
 import os
 import subprocess

terrabutler/inception.py

@@ -1,6 +1,7 @@
 from terrabutler.utils import paths
 from colorama import Fore
 from os import path
+from sys import exit
 import subprocess
 
 

terrabutler/requirements.py

@@ -1,3 +1,6 @@
+from sys import exit
+
+
 def check_requirements():
     """
     Check requirements before running the application.

terrabutler/tf.py

@@ -2,6 +2,7 @@
 import signal
 import subprocess
 from colorama import Fore
+from sys import exit
 from terrabutler.settings import get_settings
 from terrabutler.utils import paths