Python文件上传功能简单实现

本文章代码上传在码云上

代码地址

[email protected]:DanYuJie/upanddown.git

这里我们使用flask框架，简单实用

目录结构：
upandown/
        static/
                css/
                js/
                  jquery.min.js
                  toastr.min.js
        templates/
                 index.html
        test.py    

首先我们需要一个页面在templates/index.html(这里使用form表单实现)

然后是后台接收

test.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from flask import Flask,render_template, request, send_from_directory,jsonify, redirect
import os
# import sys
# reload(sys)
# sys.setdefaultencoding('utf-8')
app = Flask(__name__)

# ALLOWED_EXTENSTIONS = set(['png', 'jpg', 'jpeg', 'gif'])
app.config['UPLOAD_FOLDER'] = os.getcwd()
download_floder = app.config['UPLOAD_FOLDER'] + '/upload'

def allow_file(filename):
    allow_list = ['png', 'PNG', 'jpg', 'doc', 'docx', 'txt', 'pdf', 'PDF', 'xls', 'rar', 'exe', 'md', 'zip'] 
    a = filename.split('.')[1]
    if a in allow_list:
        return True
    else:
        return False

@app.route('/main')
def home():
    return render_template('index.html')

@app.route('/getlist')
def getlist():
    file_url_list = []
    file_floder = app.config['UPLOAD_FOLDER'] + '/upload'
    file_list = os.listdir(file_floder)
    for filename in file_list:
        file_url = url_for('download',filename=filename)
        file_url_list.append(file_url)
    # print file_list
    return jsonify(file_list)

@app.route('/download/')
def download(filename):
    return send_from_directory(download_floder,filename, as_attachment=True)

@app.route('/upload', methods=['POST', 'GET'])
def upload():
    file = request.files['file']
    if not file:
        return render_template('index.html', status='null')
    # print type(file)
    if allow_file(file.filename):
        file.save(os.path.join(app.config['UPLOAD_FOLDER']+'/upload/', file.filename))
        return render_template('index.html', status='OK')
    else:
        return 'NO'

if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0')