[英]Testing FastAPI FormData Upload
我正在尝试使用Python和FastAPI测试文件及其元数据的上传。
这是我定义上传路由的方式:
@app.post("/upload_files")
async def creste_upload_files(uploaded_files: List[UploadFile], selectedModel: str = Form(...),
patientId: str = Form(...), patientSex: str = Form(...),
actualMedication: str = Form(...), imageDim: str = Form(...),
imageFormat: str = Form(...), dateOfScan: str = Form(...)):
for uploaded_dicom in uploaded_files:
upload_folder = "webapp/src/data/"
file_object = uploaded_dicom.file
#create empty file to copy the file_object to
upload_folder = open(os.path.join(upload_folder, uploaded_dicom.filename), 'wb+')
shutil.copyfileobj(file_object, upload_folder)
upload_folder.close()
return "hello"
(我没有使用元数据,但我稍后会使用)。
我使用 unittest 进行测试:
class TestServer(unittest.TestCase):
def setUp(self):
self.client = TestClient(app)
self.metadata = {
"patientId": "1",
"patient_age": "M",
"patientSex": "59",
"patient_description": "test",
"actualeMedication": "test",
"dateOfScan": datetime.strftime(datetime.now(), "%d/%m/%Y"),
"selectedModel": "unet",
"imageDim": "h",
"imageFormat": "h"
}
def tearDown(self):
pass
def test_dcm_upload(self):
dicom_file = pydicom.read_file("tests/data/1-001.dcm")
bytes_data = dicom_file.PixelData
files = {"uploaded_files": ("dicom_file", bytes_data, "multipart/form-data")}
response = self.client.post(
"/upload_files",
json=self.metadata,
files=files
)
print(response.json())
但似乎上传不起作用,我得到以下响应打印:
{'detail': [{'loc': ['body', 'selectedModel'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'patientId'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'patientSex'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'actualMedication'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'imageDim'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'imageFormat'], 'msg': 'field required', 'type': 'value_error.missing'}, {'loc': ['body', 'dateOfScan'], 'msg': 'field required', 'type': 'value_error.missing'}]}
我可能应该使用 Formdata 而不是正文请求 ( json=self.metadata
) json=self.metadata
但我不知道应该如何完成。
答案只是用data=self.metadata
for a formData 替换可用于 body 参数的json=self.metadata
self.metadata
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.