""" Onlylegs - API endpoints Used intermally by the frontend and possibly by other applications """ import logging import json from datetime import datetime as dt from flask import Blueprint, abort, jsonify, render_template, url_for, request, g from sqlalchemy.orm import sessionmaker from gallery.auth import login_required from . import db # Import db to create a session 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 group in groups: thumbnail = db_session.query(db.GroupJunction.post_id).filter(db.GroupJunction.group_id == group.id).order_by(db.GroupJunction.date_added.desc()).first() if thumbnail is not None: thumbnail_filename = db_session.query(db.Posts.file_name).filter(db.Posts.id == thumbnail[0]).first() group.thumbnail = thumbnail_filename[0] else: group.thumbnail = None return render_template('groups/list.html', groups=groups) @blueprint.route('/') def group(group_id): """ Group view, shows all images in a group """ group = db_session.query(db.Groups).filter(db.Groups.id == group_id).first() if group is None: abort(404, 'Group not found') group_images = db_session.query(db.GroupJunction.post_id).filter(db.GroupJunction.group_id == group_id).order_by(db.GroupJunction.date_added.desc()).all() images = [] for image in group_images: image = db_session.query(db.Posts).filter(db.Posts.id == image[0]).first() images.append(image) return render_template('groups/group.html', group=group, images=images) @blueprint.route('//') def group_post(group_id, image_id): """ Image view, shows the image and its metadata from a specific group """ img = db_session.query(db.Posts).filter(db.Posts.id == image_id).first() if img is None: abort(404, 'Image not found') img.author_username = db_session.query(db.Users.username).filter(db.Users.id == img.author_id).first()[0] groups = db_session.query(db.GroupJunction.group_id).filter(db.GroupJunction.post_id == image_id).all() img.groups = [] for group in groups: group = db_session.query(db.Groups).filter(db.Groups.id == group[0]).first() img.groups.append(group) next = 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 = 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 next is not None: next = url_for('group.group_post', group_id=group_id, image_id=next[0]) if prev is not None: prev = url_for('group.group_post', group_id=group_id, image_id=prev[0]) return render_template('image.html', image=img, next_url=next, prev_url=prev)