From c6bf35b392a4fa256f39c42a1c39f8e196634eab Mon Sep 17 00:00:00 2001
From: Aman Agrawal <amanagr@zulip.com>
Date: Wed, 24 Sep 2025 11:59:42 +0530
Subject: [PATCH] registration: Use a loop to set extra content value.

---
 zerver/views/registration.py | 41 ++++++++++++------------------------
 1 file changed, 13 insertions(+), 28 deletions(-)

diff --git a/zerver/views/registration.py b/zerver/views/registration.py
index dec74511e7..357d2c9029 100644
--- a/zerver/views/registration.py
+++ b/zerver/views/registration.py
@@ -632,34 +632,19 @@ def registration_helper(
                 form.cleaned_data["how_realm_creator_found_zulip"]
             ]
             how_realm_creator_found_zulip_extra_context = ""
-            if (
-                how_realm_creator_found_zulip
-                == RealmAuditLog.HOW_REALM_CREATOR_FOUND_ZULIP_OPTIONS["other"]
-            ):
-                how_realm_creator_found_zulip_extra_context = form.cleaned_data[
-                    "how_realm_creator_found_zulip_other_text"
-                ]
-            elif (
-                how_realm_creator_found_zulip
-                == RealmAuditLog.HOW_REALM_CREATOR_FOUND_ZULIP_OPTIONS["ad"]
-            ):
-                how_realm_creator_found_zulip_extra_context = form.cleaned_data[
-                    "how_realm_creator_found_zulip_where_ad"
-                ]
-            elif (
-                how_realm_creator_found_zulip
-                == RealmAuditLog.HOW_REALM_CREATOR_FOUND_ZULIP_OPTIONS["existing_user"]
-            ):
-                how_realm_creator_found_zulip_extra_context = form.cleaned_data[
-                    "how_realm_creator_found_zulip_which_organization"
-                ]
-            elif (
-                how_realm_creator_found_zulip
-                == RealmAuditLog.HOW_REALM_CREATOR_FOUND_ZULIP_OPTIONS["review_site"]
-            ):  # nocoverage
-                how_realm_creator_found_zulip_extra_context = form.cleaned_data[
-                    "how_realm_creator_found_zulip_review_site"
-                ]
+            extra_context_options = {
+                "other": "how_realm_creator_found_zulip_other_text",
+                "ad": "how_realm_creator_found_zulip_where_ad",
+                "existing_user": "how_realm_creator_found_zulip_which_organization",
+                "review_site": "how_realm_creator_found_zulip_review_site",
+            }
+            for option, field_name in extra_context_options.items():
+                if (
+                    how_realm_creator_found_zulip
+                    == RealmAuditLog.HOW_REALM_CREATOR_FOUND_ZULIP_OPTIONS[option]
+                ):
+                    how_realm_creator_found_zulip_extra_context = form.cleaned_data[field_name]
+                    break
 
             realm = do_create_realm(
                 string_id,