python-gallery/gallery/api.py

from flask import Blueprint, current_app, send_from_directory, send_file, request, g, abort, flash, jsonify
from werkzeug.utils import secure_filename

from gallery.auth import login_required

from . import db
from sqlalchemy.orm import sessionmaker
db_session = sessionmaker(bind=db.engine)
db_session = db_session()

from PIL import Image, ImageOps, ImageFilter
from . import metadata as mt

from .logger import logger

from uuid import uuid4
import io
import os

blueprint = Blueprint('api', __name__, url_prefix='/api')


@blueprint.route('/uploads/<file>', methods=['GET'])
def uploads(file):
    # Get args
    width = request.args.get('w', default=0, type=int)  # Width of image
    height = request.args.get('h', default=0, type=int)  # Height of image
    filtered = request.args.get('f', default=False, type=bool)  # Whether to apply filters to image,
    # such as blur for NSFW images

    # if no args are passed, return the raw file
    if width == 0 and height == 0 and not filtered:
        return send_from_directory(current_app.config['UPLOAD_FOLDER'],
                                   secure_filename(file),
                                   as_attachment=True)

    # Of either width or height is 0, set it to the other value to keep aspect ratio
    if width > 0 and height == 0:
        height = width
    elif width == 0 and height > 0:
        width = height

    set_ext = current_app.config['ALLOWED_EXTENSIONS']
    buff = io.BytesIO()

    # Open image and set extension
    try:
        img = Image.open(
            os.path.join(current_app.config['UPLOAD_FOLDER'],
                         secure_filename(file)))
    except Exception as e:
        logger.server(600, f"Error opening image: {e}")
        abort(500)

    img_ext = os.path.splitext(secure_filename(file))[-1].lower().replace(
        '.', '')
    img_ext = set_ext[img_ext]
    img_icc = img.info.get(
        "icc_profile")  # Get ICC profile as it alters colours

    # Resize image and orientate correctly
    img.thumbnail((width, height), Image.LANCZOS)
    img = ImageOps.exif_transpose(img)
    
    # TODO: Add filters
    # If has NSFW tag, blur image, etc.
    if filtered:
        #pass
        img = img.filter(ImageFilter.GaussianBlur(20))
    
    try:
        img.save(buff, img_ext, icc_profile=img_icc)
    except OSError:
        # This usually happens when saving a JPEG with an ICC profile
        # Convert to RGB and try again
        img = img.convert('RGB')
        img.save(buff, img_ext, icc_profile=img_icc)
    except:
        logger.server(600, f"Error resizing image: {file}")
        abort(500)

    img.close()

    # Seek to beginning of buffer and return
    buff.seek(0)
    return send_file(buff, mimetype='image/' + img_ext)


@blueprint.route('/upload', methods=['POST'])
@login_required
def upload():
    form_file = request.files['file']
    form = request.form

    if not form_file:
        return abort(404)

    img_ext = os.path.splitext(secure_filename(
        form_file.filename))[-1].replace('.', '').lower()
    img_name = f"GWAGWA_{uuid4().__str__()}.{img_ext}"

    if not img_ext in current_app.config['ALLOWED_EXTENSIONS'].keys():
        logger.add(303, f"File extension not allowed: {img_ext}")
        abort(403)

    if os.path.isdir(current_app.config['UPLOAD_FOLDER']) == False:
        os.mkdir(current_app.config['UPLOAD_FOLDER'])

    # Save to database
    try:
        tr = db.posts(img_name, form['description'], form['alt'], g.user.id)
        db_session.add(tr)
        db_session.commit()
    except Exception as e:
        logger.server(600, f"Error saving to database: {e}")
        abort(500)

    # Save file
    try:
        form_file.save(
            os.path.join(current_app.config['UPLOAD_FOLDER'], img_name))
    except Exception as e:
        logger.server(600, f"Error saving file: {e}")
        abort(500)

    return 'Gwa Gwa'


@blueprint.route('/remove/<int:id>', methods=['POST'])
@login_required
def remove(id):
    img = db_session.query(db.posts).filter_by(id=id).first()

    if img is None:
        abort(404)
    if img.author_id != g.user.id:
        abort(403)

    try:
        os.remove(
            os.path.join(current_app.config['UPLOAD_FOLDER'],
                         img.file_name))
    except Exception as e:
        logger.server(600, f"Error removing file: {e}")
        abort(500)

    try:
        db_session.query(db.posts).filter_by(id=id).delete()
        db_session.commit()
    except Exception as e:
        logger.server(600, f"Error removing from database: {e}")
        abort(500)

    logger.server(301, f"Removed image {id}")
    flash(['Image was all in Le Head!', 1])
    return 'Gwa Gwa'


@blueprint.route('/metadata/<int:id>', methods=['GET'])
def metadata(id):
    img = db_session.query(db.posts).filter_by(id=id).first()

    if img is None:
        abort(404)

    exif = mt.metadata.yoink(
        os.path.join(current_app.config['UPLOAD_FOLDER'], img.file_name))

    return jsonify(exif)


@blueprint.route('/logfile')
@login_required
def logfile():
    filename = logger.filename()
    log_dict = {}
    i = 0

    with open(filename) as f:
        for line in f:
            line = line.split(' : ')

            event = line[0].strip().split(' ')
            event_data = {
                'date': event[0],
                'time': event[1],
                'severity': event[2],
                'owner': event[3]
            }

            message = line[1].strip()
            try:
                message_data = {
                    'code': int(message[1:4]),
                    'message': message[5:].strip()
                }
            except:
                message_data = {'code': 0, 'message': message}

            log_dict[i] = {'event': event_data, 'message': message_data}

            i += 1  # Line number, starts at 0

    return jsonify(log_dict)