feat: delete temp file (#677) (#685)

terraform-compliance · Mar 20, 2023 · 328e405 · 328e405
1 parent 621205e
commit 328e405
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/terraform_compliance/main.py b/terraform_compliance/main.py
@@ -1,4 +1,5 @@
 import os
+import atexit
 from argparse import ArgumentParser
 from tempfile import mkdtemp
 from git import Repo
@@ -33,8 +34,14 @@ class ArgHandling(object):
     pass
 
 
+def cleanup():
+    os.remove(Defaults().cache_file)
+
+
 def cli(arghandling=ArgHandling(), argparser=ArgumentParser(prog=__app_name__,
                                                             description='BDD Test Framework for Hashicorp terraform')):
+    atexit.register(cleanup)
+
     args = arghandling
     parser = argparser
     parser.add_argument('--terraform', '-t', dest='terraform_file', metavar='terraform_file', type=str, nargs='?',
@@ -151,4 +158,4 @@ def cli(arghandling=ArgHandling(), argparser=ArgumentParser(prog=__app_name__,
 
 
 if __name__ == '__main__':
-    cli()
+    cli()