python-gallery/gallery/routes/groups.py

"""
Onlylegs - Image Groups
Why groups? Because I don't like calling these albums
sounds more limiting that it actually is in this gallery
"""
from flask import Blueprint, abort, render_template, url_for

from sqlalchemy.orm import sessionmaker
from gallery import db
from gallery.utils import contrast


blueprint = Blueprint('group', __name__, url_prefix='/group')
db_session = sessionmaker(bind=db.engine)
db_session = db_session()


@blueprint.route('/', methods=['GET'])
def groups():
    """
    Group overview, shows all image groups
    """
    groups = db_session.query(db.Groups).all()

    # For each group, get the 3 most recent images
    for group in groups:
        group.author_username = db_session.query(db.Users.username)\
                                           .filter(db.Users.id == group.author_id)\
                                           .first()[0]

        # Get the 3 most recent images
        images = db_session.query(db.GroupJunction.post_id)\
                              .filter(db.GroupJunction.group_id == group.id)\
                              .order_by(db.GroupJunction.date_added.desc())\
                              .limit(3)

        # For each image, get the image data and add it to the group item
        group.images = []
        for image in images:
            group.images.append(db_session.query(db.Posts.file_name, db.Posts.post_alt,
                                                 db.Posts.image_colours, db.Posts.id)\
                                          .filter(db.Posts.id == image[0])\
                                          .first())

    return render_template('groups/list.html', groups=groups)


@blueprint.route('/<int:group_id>')
def group(group_id):
    """
    Group view, shows all images in a group
    """
    # Get the group, if it doesn't exist, 404
    group = db_session.query(db.Groups).filter(db.Groups.id == group_id).first()
    if group is None:
        abort(404, 'Group not found! D:')

    # Get the group's author username
    group.author_username = db_session.query(db.Users.username)\
                                           .filter(db.Users.id == group.author_id)\
                                           .first()[0]

    # Get all images in the group from the junction table
    junction = db_session.query(db.GroupJunction.post_id)\
                             .filter(db.GroupJunction.group_id == group_id)\
                             .order_by(db.GroupJunction.date_added.desc())\
                             .all()

    # Get the image data for each image in the group
    images = []
    for image in junction:
        image = db_session.query(db.Posts).filter(db.Posts.id == image[0]).first()
        images.append(image)

    # Check contrast for the first image in the group for the banner
    text_colour = 'rgb(var(--fg-black))'
    if images[0]:
        text_colour = contrast.contrast(images[0].image_colours[0],
                                        'rgb(var(--fg-black))',
                                        'rgb(var(--fg-white))')

    return render_template('groups/group.html',
                           group=group,
                           images=images,
                           text_colour=text_colour)


@blueprint.route('/<int:group_id>/<int:image_id>')
def group_post(group_id, image_id):
    """
    Image view, shows the image and its metadata from a specific group
    """
    # Get the image, if it doesn't exist, 404
    image = db_session.query(db.Posts).filter(db.Posts.id == image_id).first()
    if image is None:
        abort(404, 'Image not found')

    # Get the image's author username
    image.author_username = db_session.query(db.Users.username)\
                                    .filter(db.Users.id == image.author_id)\
                                    .first()[0]

    # Get all groups the image is in
    groups = db_session.query(db.GroupJunction.group_id)\
                       .filter(db.GroupJunction.post_id == image_id)\
                       .all()

    # Get the group data for each group the image is in
    image.groups = []
    for group in groups:
        group = db_session.query(db.Groups.id, db.Groups.name)\
                          .filter(db.Groups.id == group[0])\
                          .first()
        image.groups.append(group)

    # Get the next and previous images in the group
    next_url = db_session.query(db.GroupJunction.post_id)\
                         .filter(db.GroupJunction.group_id == group_id)\
                         .filter(db.GroupJunction.post_id > image_id)\
                         .order_by(db.GroupJunction.date_added.asc())\
                         .first()
    prev_url = db_session.query(db.GroupJunction.post_id)\
                         .filter(db.GroupJunction.group_id == group_id)\
                         .filter(db.GroupJunction.post_id < image_id)\
                         .order_by(db.GroupJunction.date_added.desc())\
                         .first()

    # If there is a next or previous image, get the URL for it
    if next_url is not None:
        next_url = url_for('group.group_post', group_id=group_id, image_id=next_url[0])
    if prev_url is not None:
        prev_url = url_for('group.group_post', group_id=group_id, image_id=prev_url[0])

    return render_template('image.html', image=image, next_url=next_url, prev_url=prev_url)