Spaces:
Runtime error
Runtime error
Commit ·
6e9aa02
1
Parent(s): 7cf2b34
worker: execute upload_pdf jobs
Browse files
worker.py
CHANGED
|
@@ -115,6 +115,31 @@ def poll_loop():
|
|
| 115 |
job = r.json()
|
| 116 |
job_id = job.get("job_id") or job.get("id")
|
| 117 |
print(f"claimed job {job_id}", flush=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
# processing placeholder
|
| 119 |
|
| 120 |
except Exception as e:
|
|
|
|
| 115 |
job = r.json()
|
| 116 |
job_id = job.get("job_id") or job.get("id")
|
| 117 |
print(f"claimed job {job_id}", flush=True)
|
| 118 |
+
|
| 119 |
+
payload = job.get("payload", {})
|
| 120 |
+
|
| 121 |
+
try:
|
| 122 |
+
action = payload.get("action")
|
| 123 |
+
|
| 124 |
+
if action == "upload_pdf":
|
| 125 |
+
pdf_id = payload["pdf_id"]
|
| 126 |
+
pdf_b64 = payload["pdf_b64"]
|
| 127 |
+
|
| 128 |
+
import base64
|
| 129 |
+
from sftp_store import upload_bytes
|
| 130 |
+
|
| 131 |
+
data = base64.b64decode(pdf_b64)
|
| 132 |
+
remote_path = f"pdfs/{pdf_id}.pdf"
|
| 133 |
+
|
| 134 |
+
print(f"uploading pdf {pdf_id} ({len(data)} bytes)", flush=True)
|
| 135 |
+
upload_bytes(remote_path, data)
|
| 136 |
+
print(f"uploaded to {remote_path}", flush=True)
|
| 137 |
+
|
| 138 |
+
else:
|
| 139 |
+
print(f"unknown action: {action}", flush=True)
|
| 140 |
+
|
| 141 |
+
except Exception as e:
|
| 142 |
+
print(f"job {job_id} failed: {e}", flush=True)
|
| 143 |
# processing placeholder
|
| 144 |
|
| 145 |
except Exception as e:
|