|
|
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!") |
|
|
|
