Python上传文件|极客教程

Python上传文件

在Web开发中，经常需要实现文件上传的功能，比如用户上传头像、上传文档等。Python 提供了丰富的库和框架来处理文件上传，本文将介绍如何使用Python进行文件上传，并提供一些示例代码。

1. Flask框架实现文件上传

Flask 是一个轻量级的Web框架，提供了方便的文件上传功能。下面是一个使用 Flask 实现文件上传的示例代码：

from flask import Flask, request
from werkzeug.utils import secure_filename

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    file = request.files['file']
    filename = secure_filename(file.filename)
    file.save(filename)
    return '文件上传成功！'

if __name__ == '__main__':
    app.run()

上述代码使用了 Flask 基础库和 Werkzeug 工具库。Flask 提供了request对象来处理请求数据，request.files可以获取上传的文件对象。Werkzeug 的secure_filename方法可以确保文件名的安全性，避免了一些潜在的安全问题。

在运行以上代码后，可以通过 POST 请求访问/upload路径上传文件。上传的文件会保存在当前目录下，并返回”文件上传成功！”的提示。

2. Django框架实现文件上传

Django 是一个功能强大的Web框架，也提供了便捷的文件上传功能。下面是一个使用 Django 实现文件上传的示例代码：

from django.shortcuts import render
from django.http import HttpResponse

def upload_file(request):
    if request.method == 'POST':
        file = request.FILES['file']
        with open(file.name, 'wb') as f:
            for chunk in file.chunks():
                f.write(chunk)
        return HttpResponse('文件上传成功！')
    return render(request, 'upload.html')

在以上代码中，通过request.FILES来获取上传的文件对象。使用open函数打开文件，并逐块写入数据，以避免一次性读取大文件造成的内存占用过多。

这段示例代码需要在 Django 项目中的视图函数中调用，上传的文件会保存在当前工作目录中，并返回”文件上传成功！”的提示。

3. Bottle框架实现文件上传

Bottle 是一个轻量级的Web框架，也提供了便捷的文件上传功能。下面是一个使用 Bottle 实现文件上传的示例代码：

from bottle import route, request, run

@route('/upload', method='POST')
def upload_file():
    file = request.files.get('file')
    filename = file.filename
    file.save(filename)
    return '文件上传成功！'

if __name__ == '__main__':
    run()

在以上代码中，通过request.files.get('file')来获取上传的文件对象。使用file.save方法将文件保存在当前目录下。最后返回”文件上传成功！”的提示。

在运行以上代码后，通过 POST 请求访问/upload路径上传文件，上传的文件会保存在当前目录中，并返回”文件上传成功！”的提示。

4. Tornado框架实现文件上传

Tornado 是一个高性能的Web框架，也提供了文件上传功能。下面是一个使用 Tornado 实现文件上传的示例代码：

import tornado.ioloop
import tornado.web
import os

class UploadHandler(tornado.web.RequestHandler):
    def post(self):
        file = self.request.files['file'][0]
        filename = file['filename']
        with open(filename, 'wb') as f:
            f.write(file['body'])
        self.write('文件上传成功！')

def make_app():
    return tornado.web.Application([
        (r'/upload', UploadHandler),
    ])

if __name__ == '__main__':
    app = make_app()
    app.listen(8888)
    tornado.ioloop.IOLoop.current().start()

以上代码创建了一个 Tornado 应用，并定义了一个处理文件上传的请求处理类UploadHandler。在post方法中，通过self.request.files来获取上传的文件对象。使用open函数打开文件，并将文件内容写入到文件中。

在运行以上代码后，通过 POST 请求访问/upload路径上传文件，上传的文件会保存在当前目录中，并返回”文件上传成功！”的提示。