From e104dd02c43ed2c64410f344fafbb150efaf3005 Mon Sep 17 00:00:00 2001 From: Arjun Choudhary Date: Mon, 22 Jul 2024 14:06:41 +0530 Subject: [PATCH] fix: handle null value --- drive/api/storage.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/drive/api/storage.py b/drive/api/storage.py index 3b149826..44cbd115 100644 --- a/drive/api/storage.py +++ b/drive/api/storage.py @@ -16,7 +16,7 @@ def get_max_storage(): """ plan_limit = frappe.conf.get("plan_limit") if plan_limit: - limit = plan_limit['max_storage_usage'] + limit = plan_limit["max_storage_usage"] limit = int(limit) return int(limit * 1024**2) @@ -51,7 +51,8 @@ def total_storage_used_by_file_kind(): def get_storage_allowed(): limit = get_max_storage() usage = total_storage_used() - return limit - usage[0].total_size + usage = usage[0].total_size or 0 + return limit - usage def total_disk_storage_used():