github/reddit
1
1
Fork 0
mirror of https://github.com/reddit-archive/reddit.git synced 2026-01-29 16:58:21 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add each_<userrel>() method.

Browse Source
This commit is contained in:
Logan Hanks
2012-12-11 14:54:24 -08:00
parent 13b64364d9
commit 07db3ff6d8
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
r2/r2/lib/db/userrel.py
View File

@@ -63,16 +63,19 @@ class UserRelManager(object):
return self.add(thing, user, **attrs)
def ids(self, thing):
q = self.relation._query(self.relation.c._thing1_id == thing._id,
self.relation.c._name == self.name,
sort='_date')
return [r._thing2_id for r in q]
return [r._thing2_id for r in self.by_thing(thing)]
def reverse_ids(self, user):
q = self.relation._query(self.relation.c._thing2_id == user._id,
self.relation.c._name == self.name)
return [r._thing1_id for r in q]
def by_thing(self, thing):
return self.relation._query(self.relation.c._thing1_id == thing._id,
self.relation.c._name == self.name,
sort='_date')
class MemoizedUserRelManager(UserRelManager):
"""Memoized manager for a relation to users."""
@@ -138,6 +141,7 @@ def UserRel(name, relation, disable_ids_fn=False, disable_reverse_ids_fn=False):
setattr(UR, 'get_' + name, UR._bind(mgr.get))
setattr(UR, 'add_' + name, UR._bind(mgr.add))
setattr(UR, 'remove_' + name, UR._bind(mgr.remove))
setattr(UR, 'each_' + name, UR._bind(mgr.by_thing))
setattr(UR, name + '_permission_class', permission_class)
if not disable_ids_fn:
setattr(UR, mgr.ids_fn_name, UR._bind(mgr.ids))