reddit: Add note to banning and friend json.

This commit is contained in:
Andre D
2013-05-21 00:07:04 -07:00
committed by Roger Ostrander
parent bd8fe10405
commit 03c5b6a37d

View File

@@ -589,6 +589,20 @@ class UserTableItemJsonTemplate(ThingJsonTemplate):
def thing_attr(self, thing, attr):
return ThingJsonTemplate.thing_attr(self, thing.user, attr)
def inject_note(self, thing, d):
if (thing.type in ("banned", "wikibanned") or
(c.user.gold and thing.type == "friend")):
d["note"] = getattr(thing.rel, 'note', '')
return d
def rendered_data(self, thing):
d = ThingJsonTemplate.rendered_data(self, thing)
return self.inject_note(thing, d)
def raw_data(self, thing):
d = ThingJsonTemplate.raw_data(self, thing)
return self.inject_note(thing, d)
def render(self, thing, *a, **kw):
return ObjectTemplate(self.data(thing))