delete upvoted reviews when deleting account

Author: bojieli

app/models/review.py

@@ -29,7 +29,7 @@ class Review(db.Model):
     upvote_count = db.Column(db.Integer, default=0) #点赞数量
     comment_count = db.Column(db.Integer, default=0)
 
-    upvote_users = db.relationship('User', secondary=review_upvotes)
+    upvote_users = db.relationship('User', backref='upvoted_reviews', secondary=review_upvotes)
     comments = db.relationship('ReviewComment', backref='review', lazy='joined', order_by='ReviewComment.publish_time')
 
     author = db.relationship('User', lazy='joined')

app/models/user.py

@@ -92,6 +92,7 @@ class User(db.Model, UserMixin):
             backref=db.backref('followers'))
     # followers: backref to User
     # notifications: backref to Notification
+    # upvoted_reviews: backref to Review
 
     def __init__(self, username, email, password):
         self.username = username

tests/delete_account.py

@@ -12,11 +12,15 @@
 ctx = app.test_request_context()
 ctx.push()
 
-user = User.query.filter(User.email == sys.argv[1]).first()
+if sys.argv[1].isdigit():
+    user = User.query.filter(User.id == sys.argv[1]).first()
+else:
+    user = User.query.filter(User.email == sys.argv[1]).first()
 if not user:
-    print('user ' + sys.argv[1] + ' not found')
+    print('user ' + sys.argv[1] + ' not found (please use User ID or email)')
     sys.exit(1)
 
+
 review_count = Review.query.filter(Review.author == user).count()
 if review_count > 0:
     print(str(user) + ' has ' + str(review_count) + ' reviews')
@@ -32,6 +36,7 @@
 print(str(user) + ' has followed ' + str(len(user.users_following)) + ' users')
 print(str(user) + ' has ' + str(len(user.followers)) + ' followers')
 print(str(user) + ' has ' + str(len(user.notifications)) + ' notifications')
+print(str(user) + ' has ' + str(len(user.upvoted_reviews)) + ' upvoted reviews')
 
 confirm = input('Confirm to delete user ' + str(user) + '? (y/n): ')
 if confirm != 'y' and confirm != 'Y':
@@ -49,6 +54,7 @@
 user.users_following = []
 user.followers = []
 user.notifications = []
+user.upvoted_reviews = []
 
 user.is_deleted = True
 user.username = None