Spaces:
Running
Running
File size: 3,327 Bytes
1e12801 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | from __future__ import annotations
import argparse
import os
import re
import subprocess
from pathlib import Path
from huggingface_hub import HfApi
DEFAULT_LOCAL_ENV = Path(__file__).resolve().parents[1] / ".env.local"
def _read_local_env(path: Path) -> dict[str, str]:
values: dict[str, str] = {}
if not path.exists():
return values
for raw_line in path.read_text(encoding="utf-8").splitlines():
line = raw_line.strip()
if not line or line.startswith("#") or "=" not in line:
continue
key, value = line.split("=", 1)
values[key.strip().lstrip("\ufeff")] = value.strip().strip('"').strip("'")
return values
def _remote_url() -> str:
try:
return subprocess.check_output(
["git", "remote", "get-url", "hf-space"],
cwd=Path(__file__).resolve().parents[1],
text=True,
stderr=subprocess.DEVNULL,
).strip()
except Exception:
return ""
def _token_from_remote(remote: str) -> str | None:
match = re.search(r"https://[^:]+:(hf_[^@]+)@huggingface\.co/spaces/", remote)
return match.group(1) if match else None
def _repo_id_from_remote(remote: str) -> str | None:
match = re.search(r"huggingface\.co/spaces/([^/\s]+)/([^/\s]+)", remote)
if not match:
return None
return f"{match.group(1)}/{match.group(2)}"
def main() -> None:
parser = argparse.ArgumentParser(description="Set Uwekezaji economy variables on Hugging Face Space.")
parser.add_argument("--repo-id", default=None, help="Space repo id, e.g. Mbonea/InvestingTest")
parser.add_argument("--env-file", default=str(DEFAULT_LOCAL_ENV), help="Local env file to read values from")
args = parser.parse_args()
env_file = Path(args.env_file)
values = _read_local_env(env_file)
remote = _remote_url()
repo_id = args.repo_id or _repo_id_from_remote(remote)
token = values.get("HF_SPACE_API_TOKEN") or os.getenv("HF_SPACE_API_TOKEN") or _token_from_remote(remote)
if not repo_id:
raise SystemExit("Could not resolve Space repo id. Pass --repo-id.")
if not token:
raise SystemExit("Could not resolve HF token from local env, environment, or hf-space remote.")
required = {
"HF_SPACE_UPLOAD_URL": values.get("HF_SPACE_UPLOAD_URL") or os.getenv("HF_SPACE_UPLOAD_URL"),
"HF_SPACE_REPO_ID": values.get("HF_SPACE_REPO_ID") or os.getenv("HF_SPACE_REPO_ID") or repo_id,
"ECONOMY_PIPELINE_ENABLED": values.get("ECONOMY_PIPELINE_ENABLED") or "true",
"ECONOMY_DAILY_RUN_TIME": values.get("ECONOMY_DAILY_RUN_TIME") or "18:30",
"ECONOMY_DOCUMENT_ROOT": values.get("ECONOMY_DOCUMENT_ROOT") or "/data/InvestingTest/PDFs",
}
missing = [key for key, value in required.items() if not value]
if missing:
raise SystemExit(f"Missing required values: {', '.join(missing)}")
api = HfApi(token=token)
for key, value in required.items():
api.add_space_variable(repo_id=repo_id, key=key, value=value, token=token)
print(f"set variable {key}")
api.add_space_secret(repo_id=repo_id, key="HF_SPACE_API_TOKEN", value=token, token=token)
print("set secret HF_SPACE_API_TOKEN")
print(f"updated Space environment for {repo_id}")
if __name__ == "__main__":
main()
|