79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
# server/app.py
|
|
from flask import Flask, request, jsonify, render_template, redirect, url_for, send_from_directory
|
|
from flask_talisman import Talisman
|
|
import os
|
|
|
|
app = Flask(__name__, template_folder='../templates')
|
|
Talisman(app)
|
|
|
|
RECEIVED_FILES_DIR = "../assets"
|
|
if not os.path.exists(RECEIVED_FILES_DIR):
|
|
os.makedirs(RECEIVED_FILES_DIR)
|
|
|
|
# Data storage for links and images
|
|
uploaded_links = []
|
|
uploaded_images = []
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return render_template("index.html")
|
|
|
|
@app.route('/upload/link', methods=['POST'])
|
|
def upload_link():
|
|
data = request.form
|
|
if 'link' not in data or 'uploader' not in data:
|
|
return jsonify({"error": "Link and uploader's name are required"}), 400
|
|
|
|
link_info = {'link': data['link'], 'uploader': data['uploader']}
|
|
uploaded_links.append(link_info)
|
|
|
|
with open(os.path.join(RECEIVED_FILES_DIR, "links.txt"), "a") as f:
|
|
f.write(f"{data['uploader']}: {data['link']}\n")
|
|
|
|
return redirect(url_for('index'))
|
|
|
|
@app.route('/upload/image', methods=['POST'])
|
|
def upload_image():
|
|
if 'file' not in request.files or 'uploader' not in request.form:
|
|
return jsonify({"error": "File and uploader's name are required"}), 400
|
|
|
|
file = request.files['file']
|
|
uploader = request.form['uploader']
|
|
|
|
if file.filename == '':
|
|
return jsonify({"error": "No selected file"}), 400
|
|
|
|
save_path = os.path.join(RECEIVED_FILES_DIR, file.filename)
|
|
file.save(save_path)
|
|
|
|
uploaded_images.append({'filename': file.filename, 'uploader': uploader})
|
|
|
|
return redirect(url_for('index'))
|
|
|
|
@app.route('/uploads')
|
|
def view_uploads():
|
|
return render_template("uploads.html", links=uploaded_links, images=uploaded_images)
|
|
|
|
@app.route('/assets/<filename>')
|
|
def get_image(filename):
|
|
return send_from_directory(RECEIVED_FILES_DIR, filename)
|
|
|
|
@app.route('/rename/<filename>', methods=['POST'])
|
|
def rename_file(filename):
|
|
new_name = request.form.get('new_name')
|
|
if new_name and os.path.exists(os.path.join(RECEIVED_FILES_DIR, filename)):
|
|
os.rename(os.path.join(RECEIVED_FILES_DIR, filename), os.path.join(RECEIVED_FILES_DIR, new_name))
|
|
|
|
# Update internal records
|
|
for image in uploaded_images:
|
|
if image['filename'] == filename:
|
|
image['filename'] = new_name
|
|
break
|
|
|
|
return redirect(url_for('view_uploads'))
|
|
return jsonify({"error": "Invalid file name"}), 400
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=5000, ssl_context='adhoc')
|
|
|