fixing bug with clean subcommand

airbnb · Mar 26, 2020 · c4944bf · c4944bf
1 parent 530a154
commit c4944bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/streamalert_cli/manage_lambda/package.py b/streamalert_cli/manage_lambda/package.py
@@ -23,9 +23,6 @@
 from streamalert_cli.helpers import run_command
 from streamalert_cli.terraform import TERRAFORM_FILES_PATH
 
-# Build .zip files in the top-level of the terraform directory
-THIS_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
-BUILD_DIRECTORY = os.path.join(THIS_DIRECTORY, '..', '..', 'terraform')
 LOGGER = get_logger(__name__)
 
 
@@ -85,6 +82,7 @@ def create(self):
             sys.exit(1)
 
         # Zip up files
+        # Build these in the top-level of the terraform directory
         result = shutil.make_archive(
             os.path.join(TERRAFORM_FILES_PATH, self.package_name), 'zip', temp_package_path)
         LOGGER.info('Successfully created %s', os.path.basename(result))

diff --git a/streamalert_cli/terraform/handlers.py b/streamalert_cli/terraform/handlers.py
@@ -298,7 +298,7 @@ def _rm_file(path):
             print('Removing terraform file: {}'.format(path))
             os.remove(path)
 
-        for root, _, files in os.walk('terraform'):
+        for root, _, files in os.walk(TERRAFORM_FILES_PATH):
             for file_name in files:
                 path = os.path.join(root, file_name)
                 if path.endswith('.tf.json'):