diff --git a/gallery/auth.py b/gallery/auth.py index dd11190..0ab360b 100644 --- a/gallery/auth.py +++ b/gallery/auth.py @@ -28,7 +28,7 @@ def login(): username = request.form['username'].strip() password = request.form['password'].strip() - remember = True if request.form.get('remember') else False + remember = bool(request.form['remember']) user = db_session.query(db.Users).filter_by(username=username).first() diff --git a/gallery/db.py b/gallery/db.py index a302855..525b3bb 100644 --- a/gallery/db.py +++ b/gallery/db.py @@ -3,8 +3,8 @@ OnlyLegs - Database models and functions for SQLAlchemy """ from uuid import uuid4 import os -import platformdirs from datetime import datetime as dt +import platformdirs from sqlalchemy import create_engine, Column, Integer, String, DateTime, ForeignKey, PickleType from sqlalchemy.orm import declarative_base, relationship diff --git a/gallery/views/routing.py b/gallery/views/routing.py index 105454d..a398e54 100644 --- a/gallery/views/routing.py +++ b/gallery/views/routing.py @@ -83,20 +83,20 @@ def profile(): Profile overview, shows all profiles on the onlylegs gallery """ user_id = request.args.get('id', default=None, type=int) - + # If there is no userID set, check if the user is logged in and display their profile if not user_id: if current_user.is_authenticated: user_id = current_user.id else: abort(404, 'You must be logged in to view your own profile!') - + # Get the user's data user = db_session.query(db.Users).filter(db.Users.id == user_id).first() - + if not user: abort(404, 'User not found :c') - + images = db_session.query(db.Posts).filter(db.Posts.author_id == user_id).all() - + return render_template('profile.html', user=user, images=images) diff --git a/gallery/views/settings.py b/gallery/views/settings.py index 6475132..94926e9 100644 --- a/gallery/views/settings.py +++ b/gallery/views/settings.py @@ -2,7 +2,7 @@ OnlyLegs - Settings page """ from flask import Blueprint, render_template -from flask_login import login_required, current_user +from flask_login import login_required blueprint = Blueprint('settings', __name__, url_prefix='/settings') diff --git a/run.py b/run.py index 2f5ac66..38f4698 100644 --- a/run.py +++ b/run.py @@ -28,10 +28,10 @@ if DEBUG: else: from setup.runner import OnlyLegs # pylint: disable=C0412 import sys - + # Stop Gunicorn from reading the command line arguments as it causes errors sys.argv = [sys.argv[0]] - + options = { 'bind': f'{ADDRESS}:{PORT}', 'workers': WORKERS,