entry: Adjust entry parsing to handle nested entries (BEGIN & END)

[pycalendar.git] / pycalendar / aggregator.py

diff --git a/pycalendar/aggregator.py b/pycalendar/aggregator.py
index 8df064b47300de492b51b2da0332eedd1db22422..07c5b3a72c0a6a2bc67a682bed10a933d3f27efd 100644 (file)
--- a/pycalendar/aggregator.py
+++ b/pycalendar/aggregator.py
@@ -80,6 +80,12 @@ class Aggregator (list):
         stream.write('VERSION:{}\r\n'.format(_text.escape(self.version)))
         stream.write('PRODID:{}\r\n'.format(_text.escape(self.prodid)))
         for feed in self:
-            for entry in feed:
-                entry.write(stream=stream)
+            for key in [
+                    'VEVENT',
+                    'VFREEBUSY',
+                    'VJOURNAL',
+                    'VTODO',
+                    ]:
+                for entry in feed.get(key, []):
+                    entry.write(stream=stream)
         stream.write('END:VCALENDAR\r\n')