diff --git a/felicette/utils/sys_utils.py b/felicette/utils/sys_utils.py index 1168586..f3ddeed 100644 --- a/felicette/utils/sys_utils.py +++ b/felicette/utils/sys_utils.py @@ -1,5 +1,6 @@ import sys import os +import shutil def exit_cli(print_func, message): @@ -16,3 +17,15 @@ def display_file(file_name): elif sys.platform.startswith("darwin"): os.system("open %s" % file_name) + +def remove_dir(directory): + for filename in os.listdir(directory): + file_path = os.path.join(directory, filename) + try: + if os.path.isfile(file_path) or os.path.islink(file_path): + os.unlink(file_path) + elif os.path.isdir(file_path): + shutil.rmtree(file_path) + except Exception as e: + print('Failed to delete %s. Reason: %s' % (file_path, e)) + exit_cli(print, "")