delete chat history when deleting account

This commit is contained in:
wangjiacai 2023-04-17 22:55:37 +08:00
parent a4e718067d
commit b7ba1cf4c1

View File

@ -2,7 +2,7 @@ from flask_login import login_user, logout_user
from flask import Blueprint, render_template, redirect, url_for, request, flash from flask import Blueprint, render_template, redirect, url_for, request, flash
from werkzeug.security import generate_password_hash, check_password_hash from werkzeug.security import generate_password_hash, check_password_hash
from flask_login import login_required, current_user, login_manager from flask_login import login_required, current_user, login_manager
from .models import User from .models import User, Conversation
from . import db from . import db
import time import time
@ -119,6 +119,9 @@ def manage_post():
account = User.query.filter_by( account = User.query.filter_by(
id=id, email=email, name=name, role=role, isActivated=isActivated).first() id=id, email=email, name=name, role=role, isActivated=isActivated).first()
if account: if account:
db.session.query(Conversation).filter(
Conversation.userid == id).delete()
db.session.commit()
if db.session.query(User).filter(User.id == id).delete() and not db.session.commit(): if db.session.query(User).filter(User.id == id).delete() and not db.session.commit():
time.sleep(0.05) time.sleep(0.05)
return "success" return "success"