from flask import abort, current_app from flask_login import current_user, logout_user from threading import Thread from app import db from app.decorators import content_negotiation from app.models import Avatar, User from . import bp @bp.route('/', methods=['DELETE']) @content_negotiation(produces='application/json') def delete_user(user_id): def _delete_user(app, user_id): with app.app_context(): user = User.query.get(user_id) user.delete() db.session.commit() user = User.query.get_or_404(user_id) if not (user == current_user or current_user.is_administrator()): abort(403) thread = Thread( target=_delete_user, args=(current_app._get_current_object(), user.id) ) if user == current_user: logout_user() thread.start() response_data = { 'message': f'User "{user.username}" marked for deletion' } return response_data, 202 @bp.route('//avatar', methods=['DELETE']) @content_negotiation(produces='application/json') def delete_user_avatar(user_id): def _delete_avatar(app, avatar_id): with app.app_context(): avatar = Avatar.query.get(avatar_id) avatar.delete() db.session.commit() user = User.query.get_or_404(user_id) if user.avatar is None: abort(404) if not (user == current_user or current_user.is_administrator()): abort(403) thread = Thread( target=_delete_avatar, args=(current_app._get_current_object(), user.avatar.id) ) thread.start() response_data = { 'message': f'Avatar marked for deletion' } return response_data, 202