quadfile/run.py

161 lines
4.6 KiB
Python
Executable File

#!/usr/bin/env python3
from flask import Flask, request, redirect, url_for, send_from_directory, abort, render_template
from werkzeug import secure_filename
from threading import Thread
import logging
import os
import json
import time
from random import randint
# Import our configuration
from conf import config
# Import Hyozan stuff
from Hyozan import db
from Hyozan.output import print_log, time_to_string
app = Flask(__name__)
# Pre-start functions
print_log('Main', 'Running in "' + os.getcwd() + '"')
print_log('Main', 'Checking for data folder')
if not os.path.exists(config['UPLOAD_FOLDER']):
print_log('Main', 'Data folder not found, creating')
os.makedirs(config['UPLOAD_FOLDER'])
log = logging.getLogger('werkzeug')
log.setLevel(logging.ERROR)
def cleaner_thread():
# This is horrid
while True:
print_log('Notice', 'Cleaner started')
delete_old()
time.sleep(config["CLEAN_INTERVAL"])
def delete_old():
targetTime = time.time() - config["TIME"]
old = db.get_old_files(targetTime)
for file in old:
print_log('Notice', 'Removing old file "' + file["file"] + '"')
try:
os.remove(os.path.join(config["UPLOAD_FOLDER"], file["file"]))
except Exception:
print_log('Warning', 'Failed to delete old file "' + file["file"] + '"')
db.delete_entry(file["file"])
def auth(key):
if config["KEY"] == "":
return True
elif config["KEY"] == key:
return True
else:
return False
def error_page(error):
return render_template('error.html', page=config["SITE_DATA"], error=error)
def allowed_file(filename):
if config["ALLOW_ALL_FILES"]:
return True
else:
return '.' in filename and filename.rsplit('.', 1)[1] in config["ALLOWED_EXTENSIONS"]
@app.route('/', methods=['GET', 'POST'])
def upload_file():
if request.method == 'POST':
print_log('Web', 'New file received')
if not auth(request.headers.get('X-Hyozan-Auth')):
abort(403)
data = dict()
file = request.files['file']
# Only continue if a file that's allowed gets submitted.
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
while os.path.exists(os.path.join(config["UPLOAD_FOLDER"], filename)):
filename = str(randint(1000,8999)) + '-' + secure_filename(filename)
thread1 = Thread(target = db.add_file, args = (filename,))
thread1.start()
print_log('Thread', 'Adding to DB')
file.save(os.path.join(config['UPLOAD_FOLDER'], filename))
thread1.join()
data["file"] = filename
data["url"] = config["DOMAIN"] + "/" + filename
print_log('Main', 'New file processed "' + filename + '"')
try:
if request.form["source"] == "web":
return render_template('link.html', data=data, page=config["SITE_DATA"])
except Exception:
return json.dumps(data)
else:
print_log('Notice', 'Forbidden file received')
return error_page("This file isn't allowed, sorry!")
# Return Web UI if we have a GET request
elif request.method == 'GET':
return render_template('upload.html', page=config["SITE_DATA"])
# Def all the static pages
@app.route('/about')
def about():
return render_template('about.html', page=config["SITE_DATA"])
@app.route('/terms')
def terms():
return render_template('terms.html', page=config["SITE_DATA"])
@app.route('/privacy')
def privacy():
return render_template('privacy.html', page=config["SITE_DATA"])
@app.route('/faq')
def faq():
return render_template('faq.html', page=config["SITE_DATA"])
# Custom 404
@app.errorhandler(404)
def page_not_found(e):
return error_page("We couldn't find that. Are you sure you know what you're looking for?"), 404
@app.route('/<filename>', methods=['GET'])
def get_file(filename):
print_log('Web', 'Hit "' + filename + '" - ' + time_to_string(time.time()))
try:
db.update_file(filename)
except Exception:
print_log('Warning', 'Unable to update access time. Is the file in the database?')
return send_from_directory(config['UPLOAD_FOLDER'], filename)
@app.route('/share/<filename>')
@app.route('/file/<filename>')
def serve_legacy(filename):
return send_from_directory('legacy', filename)
# Configure nginx to use these urls as custom error pages
@app.route('/error/<int:error>')
def nginx_error(error):
if error == 413:
return error_page("O-o-onii-chan, noo it's too big ~~"), 413
else:
error_page("We literally have no idea what just happened")
cleaner = Thread(target = cleaner_thread, )
cleaner.start()
if __name__ == '__main__':
app.run(
port=config["PORT"],
host=config["HOST"],
debug=config["DEBUG"]
)
cleaner.join(timeout=15)