Đối tượng Tệp Python vào FileStorage của Flask

Question

Tôi đang thử nghiệm phương thức tải lên() của tôi trong Flask. Vấn đề duy nhất là đối tượng FileStorage trong Flask có phương thức save() mà python đối tượng File không có.

tôi có thể tạo tập tin của tôi như thế này:

file = open('documents-test/test.pdf') 

Nhưng tôi không thể kiểm tra phương pháp của tôi tải lên() vì phương pháp sử dụng tiết kiệm().

Bất kỳ ý tưởng nào về cách chuyển đổi đối tượng Tệp này thành đối tượng Flask Filestorage?

Answer 1

Các get và post phương pháp kiểm tra khách hàng Flask gọi werkzeug.test.EnvironBuilder dưới mui xe - vì vậy nếu bạn vượt qua trong một cuốn từ điển như là đối số từ khóa data với tập tin của bạn, bạn sẽ có thể làm việc với nó thì:

def test_upload(): 
    with open("document-test/test.pdf", "rb") as your_file: 
     self.app.post("/upload", data={"expected_file_key": your_file}) 
     # Your test here 

Answer 2

http://werkzeug.pocoo.org/docs/0.11/datastructures/#werkzeug.datastructures.FileStorage

Tôi cần sử dụng đối tượng FileStorage cho tiện ích bên ngoài khung kiểm tra và bản thân ứng dụng, về cơ bản sao chép cách tải lên tệp hoạt động bằng biểu mẫu. Điều này làm việc cho tôi.

from werkzeug.datastructures import FileStorage 
file = None 
with open('document-test/test.pdf', 'rb') as fp: 
    file = FileStorage(fp) 
file.save('document-test/test_new.pdf')