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