fix(user): Ensure user reset removes usage fields correctly

This commit is contained in:
ReturnFI
2025-11-07 08:36:31 +00:00
parent 4e17fe04d0
commit e1894c15b6

View File

@ -2,8 +2,6 @@
import init_paths import init_paths
import sys import sys
import os
from datetime import date
from db.database import db from db.database import db
def reset_user(username): def reset_user(username):
@ -26,15 +24,21 @@ def reset_user(username):
print(f"Error: User '{username}' not found in the database.") print(f"Error: User '{username}' not found in the database.")
return 1 return 1
updates = { result = db.collection.update_one(
'upload_bytes': 0, {'_id': username},
'download_bytes': 0, {
'status': 'Offline', '$set': {
'account_creation_date': date.today().strftime("%Y-%m-%d"), 'status': 'On-hold',
'blocked': False 'blocked': False
} },
'$unset': {
'account_creation_date': "",
'download_bytes': "",
'upload_bytes': ""
}
}
)
result = db.update_user(username, updates)
if result.modified_count > 0: if result.modified_count > 0:
print(f"User '{username}' has been reset successfully.") print(f"User '{username}' has been reset successfully.")
return 0 return 0