feat(grzctl,grz-common): show size in inbox listing (#446)

Resolves #388

Author: Travis Wrightsman

packages/grz-common/src/grz_common/workers/download.py

@@ -243,6 +243,7 @@ class InboxSubmissionSummary(BaseModel):
     state: InboxSubmissionState
     oldest_upload: datetime.datetime
     newest_upload: datetime.datetime
+    total_size_bytes: int
 
 
 def query_submissions(s3_options: S3Options, show_cleaned: bool) -> list[InboxSubmissionSummary]:
@@ -266,6 +267,8 @@ def query_submissions(s3_options: S3Options, show_cleaned: bool) -> list[InboxSu
         oldest_object = submission_objects_sorted[next(iter(submission_objects_sorted))]
         newest_object = submission_objects_sorted[next(reversed(submission_objects_sorted))]
 
+        total_size_bytes = sum(map(itemgetter("Size"), submission_objects))
+
         cleaning_key = f"{submission_id}/cleaning"
         cleaned_key = f"{submission_id}/cleaned"
         if (cleaning_key in submission_objects_sorted) and (cleaned_key in submission_objects_sorted):
@@ -287,6 +290,7 @@ def query_submissions(s3_options: S3Options, show_cleaned: bool) -> list[InboxSu
             state=state,
             oldest_upload=oldest_object["LastModified"],
             newest_upload=newest_object["LastModified"],
+            total_size_bytes=total_size_bytes,
         )
 
         if state in {InboxSubmissionState.CLEANING, InboxSubmissionState.CLEANED} and (not show_cleaned):

packages/grzctl/src/grzctl/commands/list_submissions.py

@@ -23,6 +23,9 @@
 log = logging.getLogger(__name__)
 
 
+BYTES_PER_GIGABYTE = 1_000_000_000
+
+
 def _get_latest_state_str(submission_db: SubmissionDb, submission_id: str) -> str | None:
     submission_from_db = submission_db.get_submission(submission_id)
     if submission_from_db:
@@ -83,6 +86,7 @@ def _prepare_table(
         table.add_column("Database State", no_wrap=True, justify="center", style="green")
     table.add_column("Upload Duration", overflow="fold", justify="center")
     table.add_column("Newest Upload", overflow="fold")
+    table.add_column("Size (GB)", justify="center")
     for summary in summaries:
         match summary.state:
             case InboxSubmissionState.INCOMPLETE:
@@ -103,6 +107,7 @@ def _prepare_table(
             status_text,
             _format_upload_duration(summary.newest_upload - summary.oldest_upload),
             summary.newest_upload.astimezone().strftime("%Y-%m-%d %H:%M:%S"),
+            f"{summary.total_size_bytes / BYTES_PER_GIGABYTE:.1f}",
         ]
         if database_states:
             latest_state_txt = _get_latest_state_txt(database_states[summary.submission_id])