Spaces:
Runtime error
Runtime error
File size: 3,472 Bytes
3a43322 ae3e949 1422be7 3a43322 1422be7 3a43322 1422be7 34fda79 1422be7 3a43322 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | from flask import Flask, render_template, request, redirect, url_for, flash, send_from_directory, jsonify
import os
from werkzeug.utils import secure_filename
import json
from datetime import datetime
app = Flask(__name__)
app.secret_key = 'your_secret_key_here'
# 配置上传文件夹
UPLOAD_FOLDER = 'uploads'
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif', 'doc', 'docx', 'zip', 'rar'}
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
# 确保上传目录存在
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
def get_file_info(filepath):
"""获取文件信息"""
stat = os.stat(filepath)
return {
'name': os.path.basename(filepath),
'size': stat.st_size,
'modified': datetime.fromtimestamp(stat.st_mtime).strftime('%Y-%m-%d %H:%M:%S'),
'path': filepath
}
@app.route('/')
def index():
# 获取所有上传的文件
files = []
for filename in os.listdir(app.config['UPLOAD_FOLDER']):
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
if os.path.isfile(filepath):
files.append(get_file_info(filepath))
return render_template('index.html', files=files)
@app.route('/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
flash('没有选择文件')
return redirect(request.url)
file = request.files['file']
if file.filename == '':
flash('没有选择文件')
return redirect(url_for('index'))
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
# 如果文件已存在，则添加时间戳避免覆盖
if os.path.exists(file_path):
base, ext = os.path.splitext(filename)
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
filename = f"{base}_{timestamp}{ext}"
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
file.save(file_path)
flash(f'文件 {filename} 上传成功!')
return redirect(url_for('index'))
else:
flash('不支持的文件类型')
return redirect(url_for('index'))
@app.route('/download/<filename>')
def download_file(filename):
try:
return send_from_directory(app.config['UPLOAD_FOLDER'], filename, as_attachment=True)
except FileNotFoundError:
flash('文件不存在')
return redirect(url_for('index'))
@app.route('/delete/<filename>')
def delete_file(filename):
try:
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
os.remove(file_path)
flash(f'文件 {filename} 删除成功!')
except FileNotFoundError:
flash('文件不存在')
return redirect(url_for('index'))
@app.route('/list_files')
def list_files():
"""API接口：返回所有文件列表"""
files = []
for filename in os.listdir(app.config['UPLOAD_FOLDER']):
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
if os.path.isfile(filepath):
files.append(get_file_info(filepath))
return jsonify(files)
# 创建一个应用实例供gunicorn使用
application = app
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=5000) |