mirror of
https://github.com/Derpy-Leggies/OnlyLegs.git
synced 2025-01-01 12:26:13 +00:00
141 lines
3.5 KiB
Python
141 lines
3.5 KiB
Python
from flask import Flask, render_template, send_from_directory, abort, url_for, jsonify, redirect, request, session
|
|
from werkzeug.utils import secure_filename
|
|
import mysql.connector
|
|
from mysql.connector import Error
|
|
import os
|
|
|
|
# Get database stuff
|
|
DB_USER = os.environ.get('USERNAME')
|
|
DB_PASS = os.environ.get('PASSWORD')
|
|
DB_HOST = os.environ.get('HOST')
|
|
DB_PORT = os.environ.get('PORT')
|
|
|
|
DB_NAME = os.environ.get('DATABASE')
|
|
|
|
try:
|
|
DB = mysql.connector.connect(host=DB_HOST,
|
|
port=DB_PORT,
|
|
database=DB_NAME,
|
|
user=DB_USER,
|
|
password=DB_PASS)
|
|
if DB.is_connected():
|
|
db_Info = DB.get_server_info()
|
|
print("Connected to MySQL Server version ", db_Info)
|
|
|
|
cursor = DB.cursor()
|
|
cursor.execute("select database();")
|
|
|
|
record = cursor.fetchone()
|
|
print("You're connected to database: ", record)
|
|
|
|
except Error as e:
|
|
print("Error while connecting to MySQL", e)
|
|
|
|
|
|
# Set flask config
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
UPLOAD_FOLDER = os.path.join(BASE_DIR, 'uploads')
|
|
|
|
app = Flask(__name__)
|
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
|
|
|
|
|
#
|
|
# ERROR HANDLERS
|
|
#
|
|
@app.errorhandler(405)
|
|
def method_not_allowed(e):
|
|
error = '405'
|
|
msg = 'Method sussy wussy'
|
|
return render_template('error.html', error=error, msg=msg), 404
|
|
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(e):
|
|
error = '404'
|
|
msg = 'Could not find what you need!'
|
|
return render_template('error.html', error=error, msg=msg), 404
|
|
|
|
|
|
@app.errorhandler(403)
|
|
def forbidden(e):
|
|
error = '403'
|
|
msg = 'Go away! This is no place for you!'
|
|
return render_template('error.html', error=error, msg=msg), 403
|
|
|
|
|
|
@app.errorhandler(410)
|
|
def gone(e):
|
|
error = '410'
|
|
msg = 'The page is no longer available! *sad face*'
|
|
return render_template('error.html', error=error, msg=msg), 410
|
|
|
|
|
|
@app.errorhandler(500)
|
|
def internal_server_error(e):
|
|
error = '500'
|
|
msg = 'Server died inside :c'
|
|
return render_template('error.html', error=error, msg=msg), 500
|
|
|
|
|
|
#
|
|
# ROUTES
|
|
#
|
|
@app.route('/')
|
|
def home():
|
|
return render_template('home.html')
|
|
|
|
|
|
@app.route('/image/<request_id>')
|
|
def image(request_id):
|
|
# Check if request_id is valid
|
|
try:
|
|
request_id = int(request_id)
|
|
except ValueError:
|
|
abort(404)
|
|
|
|
# SQL injection prevention
|
|
sql = "SELECT imagename FROM images WHERE id = %s"
|
|
img_id = (request_id,)
|
|
|
|
# Get image details
|
|
cursor = DB.cursor()
|
|
cursor.execute(sql, img_id)
|
|
result = cursor.fetchone()
|
|
|
|
return render_template('image.html', fileName=result[0], id=request_id)
|
|
|
|
|
|
#
|
|
# METHODS
|
|
#
|
|
@app.route('/fileList/<item_type>', methods=['GET'])
|
|
def image_list(item_type):
|
|
if request.method != 'GET':
|
|
abort(405)
|
|
|
|
cursor = DB.cursor()
|
|
cursor.execute("SELECT id,imagename FROM images ORDER BY id DESC")
|
|
|
|
item_list = cursor.fetchall()
|
|
|
|
return jsonify(item_list)
|
|
|
|
|
|
@app.route('/uploads/<quality>/<request_file>', methods=['GET'])
|
|
def uploads(quality, request_file):
|
|
if request.method != 'GET':
|
|
abort(405)
|
|
|
|
quality = secure_filename(quality)
|
|
quality_dir = os.path.join(app.config['UPLOAD_FOLDER'], quality)
|
|
if not os.path.isdir(quality_dir):
|
|
abort(404)
|
|
|
|
request_file = secure_filename(request_file)
|
|
|
|
if not os.path.isfile(os.path.join(quality_dir, request_file)):
|
|
abort(404)
|
|
|
|
return send_from_directory(quality_dir, request_file)
|