use peer.library in some places

This commit is contained in:
j 2016-02-11 21:25:16 +05:30
commit 9a9185d3d5
2 changed files with 15 additions and 2 deletions

View file

@ -372,6 +372,12 @@ class List(db.Model):
return self.public_id
def items_count(self):
if self.user_id != settings.USER_ID:
peer = utils.get_peer(self.user_id)
if self.name:
return len(peer.info['lists'].get(self.name, []))
else:
return len(peer.library)
return self.get_items().count()
def json(self):