diff options
| -rw-r--r-- | api/routes/user_routes.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/api/routes/user_routes.py b/api/routes/user_routes.py index 16f6778..b494d1d 100644 --- a/api/routes/user_routes.py +++ b/api/routes/user_routes.py @@ -11,6 +11,8 @@ from api.util.check_password_reqs import check_password_reqs from api.schemas.auth_schemas import User from api.schemas.user_schemas import * from models import User as UserModel +from models import Link as LinkModel +from models import Log as LogModel from api.util.authentication import ( verify_password, get_current_user, @@ -54,6 +56,16 @@ async def delete_user( detail="User not found", ) + # Delete all links and logs associated with the user + links = ( + db.query(LinkModel).filter(LinkModel.owner == current_user.id).all() + ) + for link in links: + db.delete(link) + logs = db.query(LogModel).filter(LogModel.link == link.link).all() + for log in logs: + db.delete(log) + db.delete(user) db.commit() return status.HTTP_204_NO_CONTENT |