36 lines
976 B
Python
36 lines
976 B
Python
from simple_logger_handler import setup_logger
|
|
import os
|
|
from typing import Set
|
|
|
|
logger = setup_logger(__name__)
|
|
|
|
|
|
def return_credentials(path: str) -> str:
|
|
"""Read and return credentials from a file."""
|
|
logger.debug(f"Opening credentials for path:{path}")
|
|
try:
|
|
with open(path) as file:
|
|
return file.read().strip()
|
|
except FileNotFoundError:
|
|
logger.fatal(f"[FATAL] Secret file not found: {path}")
|
|
raise
|
|
except Exception as e:
|
|
logger.fatal(f"[FATAL] Failed to read secret file {path}: {e}")
|
|
raise
|
|
|
|
|
|
def cleanup_secret_files(secret_paths: Set[str]) -> None:
|
|
"""Delete secret files after they've been loaded into memory.
|
|
|
|
Args:
|
|
secret_paths: Set of file paths to delete
|
|
"""
|
|
for path in secret_paths:
|
|
try:
|
|
if os.path.exists(path):
|
|
os.remove(path)
|
|
logger.debug(f"Deleted secret file: {path}")
|
|
else:
|
|
logger.debug(f"Secret file already removed: {path}")
|
|
except Exception as e:
|
|
logger.warning(f"Could not delete secret file {path}: {e}") |