Convert approved_by and banned_by fields to redditor objects

praw-dev · Apr 2, 2012 · 1c49c5c · 1c49c5c
1 parent 273bb76
commit 1c49c5c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/reddit/objects.py b/reddit/objects.py
@@ -24,6 +24,8 @@
 from reddit.helpers import (_get_section, _get_sorter, _modify_relationship,
                             _request)
 
+REDDITOR_KEYS = ('approved_by', 'author', 'banned_by', 'redditor')
+
 
 class RedditContentObject(object):
     """Base class that  represents actual reddit objects."""
@@ -72,8 +74,8 @@ def __getattr__(self, attr):
     def __setattr__(self, name, value):
         if value and name == 'subreddit':
             value = Subreddit(self.reddit_session, value, fetch=False)
-        elif value and name in ['redditor', 'author']:
-            if value == '[deleted]':
+        elif value and name in REDDITOR_KEYS:
+            if not value or value == '[deleted]':
                 value = None
             else:
                 value = Redditor(self.reddit_session, value, fetch=False)