warn user to delete user_config.yml and the generated CSV file

Author: ronald-bautista

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = conformity-migration-tool
-version = 0.12.0
+version = 0.13.0
 description = Migrates your visiblity information in cloudconformity.com to cloudone.trendmicro.com
 long_description = file: README.md
 long_description_content_type = text/markdown

src/conformity_migration_tool/aws_cli.py

@@ -1,3 +1,4 @@
+import atexit
 import csv
 import json
 import multiprocessing as mp
@@ -528,5 +529,20 @@ def pretty_print(obj):
     print(json.dumps(obj, indent=4, default=str))
 
 
+@atexit.register
+def warn_aws_creds():
+    print(
+        """
+
+\033[32m!!! W A R N I N G !!!\033[0m"""
+    )
+    print(
+        """
+    After you finish updating the stack, please make sure to delete
+    any generated CSV file that may contain your AWS credentials.
+    """
+    )
+
+
 if __name__ == "__main__":
     cli()

src/conformity_migration_tool/cli.py

@@ -1,3 +1,4 @@
+import atexit
 import csv
 import os
 import time
@@ -1283,5 +1284,23 @@ def empty_c1():
         # raise e
 
 
+@atexit.register
+def warn_user_config():
+    user_conf_path = user_config_path()
+    if not user_conf_path.exists():
+        return
+    print(
+        """
+
+\033[32m!!! W A R N I N G !!!\033[0m"""
+    )
+    print(
+        f"""
+    The file {user_conf_path} contains your Conformity and Cloud One API Keys.
+    Please delete the file once you are done using this tool.
+    """
+    )
+
+
 if __name__ == "__main__":
     cli()