diff --git a/analytics/management/commands/populate_analytics_db.py b/analytics/management/commands/populate_analytics_db.py index 2b8d6661e0..47d4cdeadd 100644 --- a/analytics/management/commands/populate_analytics_db.py +++ b/analytics/management/commands/populate_analytics_db.py @@ -1,6 +1,8 @@ +import os from datetime import timedelta from typing import Any, Dict, List, Mapping, Type, Union +from django.core.files.uploadedfile import UploadedFile from django.core.management.base import BaseCommand from django.utils.timezone import now as timezone_now @@ -18,8 +20,10 @@ from analytics.models import ( from zerver.actions.create_realm import do_create_realm from zerver.actions.users import do_change_user_role from zerver.lib.create_user import create_user +from zerver.lib.storage import static_path from zerver.lib.stream_color import STREAM_ASSIGNMENT_COLORS from zerver.lib.timestamp import floor_to_day +from zerver.lib.upload import upload_message_image_from_request from zerver.models import Client, Realm, Recipient, Stream, Subscription, UserGroup, UserProfile @@ -113,6 +117,13 @@ class Command(BaseCommand): ] Subscription.objects.bulk_create(subs) + # Create an attachment in the database for set_storage_space_used_statistic. + IMAGE_FILE_PATH = static_path("images/test-images/checkbox.png") + file_info = os.stat(IMAGE_FILE_PATH) + file_size = file_info.st_size + with open(IMAGE_FILE_PATH, "rb") as fp: + upload_message_image_from_request(UploadedFile(fp), shylock, file_size) + FixtureData = Mapping[Union[str, int, None], List[int]] def insert_fixture_data( diff --git a/analytics/views/stats.py b/analytics/views/stats.py index cd50756b3b..d2f9ebd4cf 100644 --- a/analytics/views/stats.py +++ b/analytics/views/stats.py @@ -59,6 +59,7 @@ def render_stats( data_url_suffix=data_url_suffix, for_installation=for_installation, remote=remote, + upload_space_used=request.user.realm.currently_used_upload_space_bytes(), ) request_language = get_and_set_request_language( diff --git a/static/js/stats/stats.js b/static/js/stats/stats.js index 1d93a4f4fa..bcca580f17 100644 --- a/static/js/stats/stats.js +++ b/static/js/stats/stats.js @@ -17,6 +17,21 @@ const font_14pt = { let last_full_update = Number.POSITIVE_INFINITY; +// Copied from attachments_ui.js +function bytes_to_size(bytes, kb_with_1024_bytes = false) { + const kb_size = kb_with_1024_bytes ? 1024 : 1000; + const sizes = ["B", "KB", "MB", "GB", "TB"]; + if (bytes === 0) { + return "0 B"; + } + const i = Number.parseInt(Math.floor(Math.log(bytes) / Math.log(kb_size)), 10); + let size = Math.round(bytes / Math.pow(kb_size, i)); + if (i > 0 && size < 10) { + size = Math.round((bytes / Math.pow(kb_size, i)) * 10) / 10; + } + return size + " " + sizes[i]; +} + // TODO: should take a dict of arrays and do it for all keys function partial_sums(array) { let accumulator = 0; @@ -153,6 +168,13 @@ function get_thirty_days_messages_sent(data) { $("#id_thirty_days_messages_sent").closest("summary-stats").show(); } +function set_storage_space_used_statistic(upload_space_used) { + const space_used = bytes_to_size(upload_space_used, true); + + $("#id_storage_space_used").text(space_used); + $("#id_storage_space_used").closest("summary-stats").show(); +} + // PLOTLY CHARTS function populate_messages_sent_over_time(data) { if (data.end_times.length === 0) { @@ -1113,3 +1135,5 @@ get_chart_data( {chart_name: "messages_read_over_time", min_length: "10"}, populate_messages_read_over_time, ); + +set_storage_space_used_statistic(page_params.upload_space_used); diff --git a/templates/analytics/stats.html b/templates/analytics/stats.html index 52161ae8d9..207f7b8e74 100644 --- a/templates/analytics/stats.html +++ b/templates/analytics/stats.html @@ -28,6 +28,7 @@