Spaces:
Sleeping
Sleeping
| """ | |
| Where the raw copy of an uploaded audience file is kept. An upload is never just parsed and | |
| discarded, so the original file is stored as well. | |
| With SUPABASE_URL and SUPABASE_SERVICE_KEY set the file goes to a private Supabase Storage | |
| bucket (SUPABASE_BUCKET, default "uploads"). Without them it goes to a local folder, which is | |
| right for development and wrong for a host whose disk is wiped on restart. | |
| """ | |
| import os | |
| from pathlib import Path | |
| import httpx | |
| TIMEOUT_SECONDS = 30 | |
| CONTENT_TYPES = { | |
| "csv": "text/csv", | |
| "xlsx": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", | |
| } | |
| class StorageError(Exception): | |
| """The file could not be stored. The upload is refused rather than kept only in memory.""" | |
| def _remote() -> tuple[str, str, str] | None: | |
| url = os.getenv("SUPABASE_URL", "").strip().rstrip("/") | |
| key = os.getenv("SUPABASE_SERVICE_KEY", "").strip() | |
| if not url or not key: | |
| return None | |
| return url, key, os.getenv("SUPABASE_BUCKET", "uploads").strip() or "uploads" | |
| def save_upload(org_id: str, database_id: str, ext: str, data: bytes) -> str: | |
| """Stores the original file as {org_id}/{database_id}.{ext}; returns where it went.""" | |
| name = f"{org_id}/{database_id}.{ext}" | |
| remote = _remote() | |
| if remote is None: | |
| path = Path(os.getenv("UPLOADS_DIR", "uploads")) / name | |
| path.parent.mkdir(parents=True, exist_ok=True) | |
| path.write_bytes(data) | |
| return str(path) | |
| url, key, bucket = remote | |
| try: | |
| res = httpx.post( | |
| f"{url}/storage/v1/object/{bucket}/{name}", | |
| content=data, | |
| headers={ | |
| "Authorization": f"Bearer {key}", | |
| "Content-Type": CONTENT_TYPES.get(ext.lower(), "application/octet-stream"), | |
| "x-upsert": "true", | |
| }, | |
| timeout=TIMEOUT_SECONDS, | |
| ) | |
| except httpx.HTTPError as err: | |
| raise StorageError(f"Could not reach file storage: {err.__class__.__name__}") from err | |
| if res.status_code >= 300: | |
| raise StorageError(f"File storage refused the file ({res.status_code})") | |
| return f"{bucket}/{name}" | |