Avinashnalla7 commited on
Commit
6e9aa02
·
1 Parent(s): 7cf2b34

worker: execute upload_pdf jobs

Browse files
Files changed (1) hide show
  1. worker.py +25 -0
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: