import os import fastapi app = fastapi.FastAPI() @app.get("/") async def root(): check_env() return {"message": "Hello World"} def check_env(): var1_env = os.getenv("AUTHOR_NAME") var2_env = os.getenv("REPO_NAME") secret_env = os.getenv("SECRET") secret_env2 = os.getenv("SECRET2") with open("var1.build") as f1: var1_fs = f1.read().rstrip("\n") with open("var2.build") as f2: var2_fs = f2.read().rstrip("\n") with open("secret.build") as fs: secret_fs = fs.read().rstrip("\n") with open("secret2.build") as fs: secret_fs2 = fs.read().rstrip("\n") print(f"'{var1_env}', '{var1_fs}'") print(f"'{var2_env}', '{var2_fs}'") print(f"'{secret_env}', '{secret_fs}'") print(f"'{secret_env2}', '{secret_fs2}'") if (secret_env != secret_fs) or (secret_env2 != secret_fs2) or (var1_env != var1_fs) or (var2_env != var2_fs): raise Exception("Environment variables and build-time values mismatch!") if (secret_env != "s3cr3t") or (secret_env2 != "s3cr3t2") or (var1_env != "hf-user") or (var2_env != "facebook/mms-tts"): raise Exception("Environment have unexpected values!")