tea/browsertrix
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Recalculate Organization.storedBytes in migration 0017 (#1220)

Browse Source
This commit is contained in:
Tessa Walsh 2023-09-28 14:22:10 -04:00 committed by GitHub
parent 9b125bc2c6
commit 1f74f03447
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/btrixcloud/migrations/migration_0017_storage_by_type.py
View File

@ -54,11 +54,16 @@ class Migration(BaseMigration):
if profile_file: if profile_file:
bytes_stored_profiles += profile_file.get("size", 0) bytes_stored_profiles += profile_file.get("size", 0)
org_total_bytes = (
bytes_stored_crawls + bytes_stored_uploads + bytes_stored_profiles
)
try: try:
res = await mdb_orgs.find_one_and_update( res = await mdb_orgs.find_one_and_update(
{"_id": oid}, {"_id": oid},
{ {
"$set": { "$set": {
"bytesStored": org_total_bytes,
"bytesStoredCrawls": bytes_stored_crawls, "bytesStoredCrawls": bytes_stored_crawls,
"bytesStoredUploads": bytes_stored_uploads, "bytesStoredUploads": bytes_stored_uploads,
"bytesStoredProfiles": bytes_stored_profiles, "bytesStoredProfiles": bytes_stored_profiles,