From 2d5757af0e196c5890c06357a5134e402f5eb188 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@netboxlabs.com>
Date: Tue, 13 Feb 2024 13:23:20 -0500
Subject: [PATCH] Expose option context on ObjectVar for custom scripts

---
 docs/customization/custom-scripts.md | 1 +
 netbox/extras/scripts.py             | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/docs/customization/custom-scripts.md b/docs/customization/custom-scripts.md
index 96423a94b..2e0e983c9 100644
--- a/docs/customization/custom-scripts.md
+++ b/docs/customization/custom-scripts.md
@@ -304,6 +304,7 @@ A particular object within NetBox. Each ObjectVar must specify a particular mode
 
 * `model` - The model class
 * `query_params` - A dictionary of query parameters to use when retrieving available options (optional)
+* `context` - A custom dictionary mapping template context variables to fields, used when rendering `<option>` elements within the dropdown menu (optional)
 * `null_option` - A label representing a "null" or empty choice (optional)
 
 To limit the selections available within the list, additional query parameters can be passed as the `query_params` dictionary. For example, to show only devices with an "active" status:
diff --git a/netbox/extras/scripts.py b/netbox/extras/scripts.py
index 40301e5c9..3a7b361da 100644
--- a/netbox/extras/scripts.py
+++ b/netbox/extras/scripts.py
@@ -193,16 +193,19 @@ class ObjectVar(ScriptVariable):
 
     :param model: The NetBox model being referenced
     :param query_params: A dictionary of additional query parameters to attach when making REST API requests (optional)
+    :param context: A custom dictionary mapping template context variables to fields, used when rendering <option>
+        elements within the dropdown menu (optional)
     :param null_option: The label to use as a "null" selection option (optional)
     """
     form_field = DynamicModelChoiceField
 
-    def __init__(self, model, query_params=None, null_option=None, *args, **kwargs):
+    def __init__(self, model, query_params=None, context=None, null_option=None, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
         self.field_attrs.update({
             'queryset': model.objects.all(),
             'query_params': query_params,
+            'context': context,
             'null_option': null_option,
         })