API return objects are now iterable (#38)

Detrous · Jan 14, 2020 · 9cb5c64 · 9cb5c64
1 parent c99a18d
commit 9cb5c64
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/darksky/base.py b/darksky/base.py
@@ -20,9 +20,13 @@ def __init__(self, summary=None, icon=None, data=None, timezone=None):
 
     def __repr__(self):
         return '%s([%d])' % (self.__class__.__name__, len(self.data))
+
+    def __iter__(self):
+        return iter(self.data)
 
 
 class AutoInit:
+
     def __init__(self, **params):
         try:
             timezone = pytz.timezone(params.pop('timezone', None))
@@ -42,5 +46,8 @@ def __init__(self, **params):
             else:
                 setattr(self, field, None)
 
+    def __iter__(self):
+        return iter(self.__dict__.items())
+
     def __repr__(self):
         return '%s(%s)' % (self.__class__.__name__, getattr(self, 'time', ''))