Bryan Van Deusen
141 lines
4.5 KiB
Python
141 lines
4.5 KiB
Python
# app/main.py
|
|
|
|
from flask import Blueprint, render_template, flash, redirect, url_for, abort, send_from_directory, request
|
|
from flask_login import login_required, current_user
|
|
|
|
from sqlalchemy import desc, func
|
|
|
|
from app.utils.image_importer import import_images_task
|
|
from app.models import ImageRecord
|
|
from app import db
|
|
import os
|
|
|
|
main = Blueprint('main', __name__)
|
|
|
|
@main.route('/')
|
|
def index():
|
|
from random import choice
|
|
from app.models import ImageRecord
|
|
|
|
image = ImageRecord.query.order_by(func.random()).first()
|
|
background_url = None
|
|
|
|
if image:
|
|
path = image.thumb_path or image.filepath
|
|
filename = path.replace('/images/', '')
|
|
background_url = url_for('main.serve_image', filename=filename)
|
|
|
|
return render_template('index.html', background_url=background_url)
|
|
|
|
@main.route('/gallery')
|
|
@login_required
|
|
def gallery():
|
|
page = request.args.get('page', 1, type=int)
|
|
per_page = request.args.get('per_page', 35, type=int)
|
|
|
|
|
|
images = ImageRecord.query.order_by(
|
|
desc(func.coalesce(ImageRecord.taken_at, ImageRecord.imported_at))
|
|
).paginate(page=page, per_page=per_page)
|
|
|
|
return render_template(
|
|
'gallery.html',
|
|
images=images,
|
|
has_next=images.has_next,
|
|
next_page_url=url_for('main.gallery', page=images.next_num, per_page=per_page) if images.has_next else '',
|
|
has_prev=images.has_prev,
|
|
prev_page_url=url_for('main.gallery', page=images.prev_num, per_page=per_page) if images.has_prev else ''
|
|
)
|
|
|
|
|
|
@main.route('/images/<path:filename>')
|
|
def serve_image(filename):
|
|
return send_from_directory('/images', filename)
|
|
|
|
@main.route('/artist/<artist_name>')
|
|
@login_required
|
|
def artist_gallery(artist_name):
|
|
page = request.args.get('page', 1, type=int)
|
|
per_page = request.args.get('per_page', 35, type=int)
|
|
|
|
images = ImageRecord.query.filter_by(artist=artist_name).order_by(
|
|
desc(func.coalesce(ImageRecord.taken_at, ImageRecord.imported_at))
|
|
).paginate(page=page, per_page=per_page)
|
|
|
|
return render_template(
|
|
'artist_gallery.html',
|
|
images=images,
|
|
artist=artist_name,
|
|
has_next=images.has_next,
|
|
next_page_url=url_for('main.artist_gallery', artist_name=artist_name, page=images.next_num, per_page=per_page) if images.has_next else '',
|
|
has_prev=images.has_prev,
|
|
prev_page_url=url_for('main.artist_gallery', artist_name=artist_name, page=images.prev_num, per_page=per_page) if images.has_prev else ''
|
|
)
|
|
|
|
|
|
@main.route('/artists')
|
|
@login_required
|
|
def artist_list():
|
|
from app.models import ImageRecord
|
|
from sqlalchemy import func
|
|
|
|
# Define how many images to show per artist card
|
|
images_per_artist = 3 # Change this value to tweak
|
|
|
|
# Get all unique artist names
|
|
artists = db.session.query(ImageRecord.artist).distinct().all()
|
|
artist_data = []
|
|
|
|
for (artist,) in artists:
|
|
images = (
|
|
ImageRecord.query
|
|
.filter_by(artist=artist)
|
|
.order_by(ImageRecord.taken_at.desc().nullslast(), ImageRecord.imported_at.desc())
|
|
.limit(images_per_artist)
|
|
.all()
|
|
)
|
|
artist_data.append({'name': artist, 'images': images})
|
|
|
|
return render_template('artist_list.html', artist_data=artist_data, images_per_artist=images_per_artist)
|
|
|
|
@main.route('/settings')
|
|
@login_required
|
|
def settings():
|
|
if not current_user.is_admin:
|
|
abort(403)
|
|
return render_template('settings.html')
|
|
|
|
@main.route('/trigger-thumbnail-generation', methods=['POST'])
|
|
@login_required
|
|
def trigger_thumbnail_generation():
|
|
if not current_user.is_admin:
|
|
flash("You do not have permission to perform this action.", "danger")
|
|
return redirect(url_for('main.settings'))
|
|
|
|
try:
|
|
with open('/import/thumbnail.flag', 'w') as f:
|
|
f.write("trigger")
|
|
flash("Thumbnail regeneration has been triggered.", "success")
|
|
except Exception as e:
|
|
flash(f"Failed to trigger thumbnail regeneration: {e}", "danger")
|
|
|
|
return redirect(url_for('main.settings'))
|
|
|
|
@main.route('/import-images')
|
|
@login_required
|
|
def trigger_image_import():
|
|
with open('/import/trigger.flag', 'w') as f:
|
|
f.write('start')
|
|
flash("Image import triggered.", "success")
|
|
return redirect(url_for('main.settings'))
|
|
|
|
@main.route('/reset-db', methods=['POST'])
|
|
@login_required
|
|
def reset_db():
|
|
if not current_user.is_admin:
|
|
abort(403)
|
|
|
|
ImageRecord.query.delete()
|
|
db.session.commit()
|
|
flash("Database has been reset. All image records removed.", "info")
|
|
return redirect(url_for('main.settings')) |