HariLogicgo commited on
Commit
feb34ce
·
1 Parent(s): e6468b0

target added

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -213,6 +213,16 @@ async def upload_source(image: UploadFile = File(...)):
213
  file_id = await fs_bucket.upload_from_stream(image.filename, contents, metadata={"type": "source"})
214
  return {"source_id": str(file_id)}
215
 
 
 
 
 
 
 
 
 
 
 
216
  @fastapi_app.post("/target")
217
  async def upload_target(image: UploadFile = File(...)):
218
  contents = await image.read()
 
213
  file_id = await fs_bucket.upload_from_stream(image.filename, contents, metadata={"type": "source"})
214
  return {"source_id": str(file_id)}
215
 
216
+ @fastapi_app.get("/targets")
217
+ async def list_targets():
218
+ files = []
219
+ async for file in database.fs.files.find({"metadata.type": "target", "metadata.predefined": True}):
220
+ files.append({
221
+ "id": str(file["_id"]),
222
+ "filename": file["filename"]
223
+ })
224
+ return {"targets": files}
225
+
226
  @fastapi_app.post("/target")
227
  async def upload_target(image: UploadFile = File(...)):
228
  contents = await image.read()