From 247fdc19ad256c7599f043aa49d3125d3a1f4356 Mon Sep 17 00:00:00 2001
From: Adrian Holovaty <adrian@holovaty.com>
Date: Tue, 26 Dec 2006 22:56:53 +0000
Subject: [PATCH] newforms: Implemented RadioFieldRenderer.__getitem__(), which
 allows for index lookup on radio fields

git-svn-id: http://code.djangoproject.com/svn/django/trunk@4238 bcc190cf-cafb-0310-a4f2-bffc1f526a37
---
 django/newforms/widgets.py           |  4 ++++
 tests/regressiontests/forms/tests.py | 19 +++++++++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/django/newforms/widgets.py b/django/newforms/widgets.py
index 0dd0692e29c9b..996e353775d07 100644
--- a/django/newforms/widgets.py
+++ b/django/newforms/widgets.py
@@ -189,6 +189,10 @@ def __iter__(self):
         for i, choice in enumerate(self.choices):
             yield RadioInput(self.name, self.value, self.attrs.copy(), choice, i)
 
+    def __getitem__(self, idx):
+        choice = self.choices[idx] # Let the IndexError propogate
+        return RadioInput(self.name, self.value, self.attrs.copy(), choice, idx)
+
     def __unicode__(self):
         "Outputs a <ul> for this set of radio fields."
         return u'<ul>\n%s\n</ul>' % u'\n'.join([u'<li>%s</li>' % w for w in self])
diff --git a/tests/regressiontests/forms/tests.py b/tests/regressiontests/forms/tests.py
index a3445d7176c45..6bf64422fbe6c 100644
--- a/tests/regressiontests/forms/tests.py
+++ b/tests/regressiontests/forms/tests.py
@@ -514,6 +514,25 @@
 beatle J G George False
 beatle J R Ringo False
 
+A RadioFieldRenderer object also allows index access to individual RadioInput
+objects.
+>>> w = RadioSelect()
+>>> r = w.render('beatle', 'J', choices=(('J', 'John'), ('P', 'Paul'), ('G', 'George'), ('R', 'Ringo')))
+>>> print r[1]
+<label><input type="radio" name="beatle" value="P" /> Paul</label>
+>>> print r[0]
+<label><input checked="checked" type="radio" name="beatle" value="J" /> John</label>
+>>> r[0].is_checked()
+True
+>>> r[1].is_checked()
+False
+>>> r[1].name, r[1].value, r[1].choice_value, r[1].choice_label
+('beatle', u'J', 'P', 'Paul')
+>>> r[10]
+Traceback (most recent call last):
+...
+IndexError: list index out of range
+
 # CheckboxSelectMultiple Widget ###############################################
 
 >>> w = CheckboxSelectMultiple()