prep_tests.py 文件源码

def save_secrets():
    logger = logging.getLogger(__name__)
    logger.info("Reading secrets from environment variables")
    secrets = {}

    missing_key = False
    for secrets_key in secrets_keys:
        try:
            logger.info("Writing key: " + secrets_key)
            secrets[secrets_key] = os.environ[secrets_key]
        except KeyError:
            logger.warning("Missing key:", secrets_key)
            missing_key = True

    if missing_key:
        logger.critical("Aborting...")
        return

    logger.info("Writing secrets to secrets.json")
    f = Fernet(key)
    with open("config/secrets.dat", 'wb') as secrets_file:
        secrets_file.write(f.encrypt(json.dumps(secrets).encode()))