SOlve issue #610

collective · Jun 30, 2024 · 4dc2ea2 · 4dc2ea2
1 parent 39fef78
commit 4dc2ea2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/icalendar/prop.py b/src/icalendar/prop.py
@@ -233,6 +233,9 @@ def __init__(self, c_list):
         self.cats = [vText(c) for c in c_list]
         self.params = Parameters()
 
+    def __iter__(self):
+        return iter(vCategory.from_ical(self.to_ical()))
+
     def to_ical(self):
         return b",".join([c.to_ical() for c in self.cats])
 

diff --git a/src/icalendar/tests/prop/test_unit.py b/src/icalendar/tests/prop/test_unit.py
@@ -336,6 +336,9 @@ def test_prop_vCategory(self):
 
         self.assertEqual(v_cat.to_ical(), b'cat 1,cat 2,cat 3')
         self.assertEqual(vCategory.from_ical(v_cat.to_ical()), catz)
+        c = vCategory(vCategory.from_ical("APPOINTMENT,EDUCATION"))
+        cats = list(c)
+        assert cats == ["APPOINTMENT", "EDUCATION"]
 
     def test_prop_TypesFactory(self):
         from icalendar.prop import TypesFactory
@@ -365,3 +368,4 @@ def test_prop_TypesFactory(self):
             factory.from_ical('cn', b'Rasmussen\\, Max M\xc3\xb8ller'),
             'Rasmussen, Max M\xf8ller'
         )
+