feat(avatar): add user token accounting

2026-08-25 13:24:02 +08:00
parent 7a0199e685
commit 699bbbde57
15 changed files with 798 additions and 52 deletions
@@ -8,6 +8,8 @@ from models import (
    TakeoverCursor,
    TakeoverMessage,
    TakeoverReplyTask,
+    TokenAccount,
+    TokenUsage,
    User,
 )

@@ -107,6 +109,13 @@ def authorization_context():
        db.query(Avatar).filter(Avatar.id.in_(avatar_ids)).delete(
            synchronize_session=False
        )
+        user_ids = [owner.id, other.id]
+        db.query(TokenUsage).filter(TokenUsage.user_id.in_(user_ids)).delete(
+            synchronize_session=False
+        )
+        db.query(TokenAccount).filter(TokenAccount.user_id.in_(user_ids)).delete(
+            synchronize_session=False
+        )
        db.query(User).filter(User.id.in_([owner.id, other.id])).delete(
            synchronize_session=False
        )