Spaces:
Sleeping
Sleeping
File size: 2,612 Bytes
2ace788 |
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 |
from flask import Flask, request, redirect, url_for, send_file, render_template, flash
from flask_cors import CORS
from werkzeug.utils import secure_filename
from pymongo.mongo_client import MongoClient
from pymongo.server_api import ServerApi
import urllib.parse
import os
import io
app = Flask(__name__)
app.secret_key = os.getenv('SECRET_KEY')
CORS(app,resources={r"/*":{"origins":"*"}})
username = urllib.parse.quote_plus(os.getenv('MONGO_USERNAME'))
password = urllib.parse.quote_plus(os.getenv('MONGO_PASSWORD'))
restUri = os.getenv('REST_URI')
uri = f'mongodb+srv://{username}:{password}{restUri}'
client = MongoClient(uri, server_api=ServerApi('1'))
db = client['file_storage']
files_collection = db['files']
def allowed_file(filename):
return True
@app.route('/')
def index():
return render_template('index.html')
@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
if request.method == 'POST':
if 'file' not in request.files:
flash('No file part')
return redirect(request.url)
files = request.files.getlist('file')
for file in files:
if file.filename == '':
flash('No selected file')
return redirect(request.url)
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
file_data = {
'filename': filename,
'data': file.read()
}
files_collection.insert_one(file_data)
flash('Files successfully uploaded')
return redirect(url_for('list_files'))
return render_template('upload.html')
@app.route('/uploads/<filename>')
def uploaded_file(filename):
file_data = files_collection.find_one({'filename': filename})
if file_data:
return send_file(
io.BytesIO(file_data['data']),
mimetype='application/octet-stream'
)
else:
return 'File not found'
@app.route('/files')
def list_files():
files = [file_data['filename'] for file_data in files_collection.find({}, {'_id': 1, 'filename': 1})]
return render_template('files.html', files=files)
@app.route('/delete/<filename>', methods=['POST'])
def delete_file(filename):
file_data = files_collection.find_one({'filename': filename})
if file_data:
files_collection.delete_one({'filename': filename})
flash('File successfully deleted')
else:
flash('File not found')
return redirect(url_for('list_files'))
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0') |