fix(user): Ensure user reset removes usage fields correctly
This commit is contained in:
@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user