Bryan Van Deusen
70 lines
2.0 KiB
Python
70 lines
2.0 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():
|
|
return render_template('index.html')
|
|
|
|
@main.route('/gallery')
|
|
@login_required
|
|
def gallery():
|
|
recent_images = 21
|
|
images = ImageRecord.query.order_by(
|
|
desc(func.coalesce(ImageRecord.taken_at, ImageRecord.imported_at))
|
|
).limit(recent_images).all()
|
|
return render_template('gallery.html', images=images)
|
|
|
|
@main.route('/static/images/<path:filename>')
|
|
def serve_image(filename):
|
|
return send_from_directory('/images', filename)
|
|
|
|
@main.route('/artist/<artist_name>')
|
|
def artist_gallery(artist_name):
|
|
per_page=49
|
|
page = request.args.get('page', 1, 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)
|
|
|
|
|
|
@main.route('/settings')
|
|
@login_required
|
|
def settings():
|
|
if not current_user.is_admin:
|
|
abort(403)
|
|
return render_template('settings.html')
|
|
|
|
@main.route('/import-images')
|
|
@login_required
|
|
def trigger_image_import():
|
|
if not current_user.is_admin:
|
|
abort(403)
|
|
source_dir = '/import'
|
|
dest_dir = '/images'
|
|
|
|
result = import_images_task(source_dir, dest_dir)
|
|
flash(f"Image import completed: {result}")
|
|
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')) |