class UserTag(mongopie.Model):
......
@classmethod
def add_tag(cls, vote):
ut = cls.find_one(user=vote.votee,
tag=vote.tag)
if not ut:
ut = cls()
ut.user = vote.votee
ut.tag = vote.tag
ut.count += 1
ut.save()
return ut
class Vote(mongopie.Model):
......
def on_created(self):
UserTag.add_tag(self)
这样每当一个投票产生后,UserTag对应的count会加一.
def make_vote(voter, votee, tag):
v = Vote()
v.voter = voter
v.votee = votee
v.tag = tag
v.save()
return v
def test():
# Tom 给Jack的hacking 投了一票
make_vote('Tom', 'Jack', 'Hacking')
# Jerry 给Jack的Food 投了一票
make_vote('Jerry', 'Jack', 'Food')
# Jerry 也给Jack的Hacking投了一票
make_vote('Jerry', 'Jack', 'Hacking')
for ut in UserTag.find(user='Jack'): # 找出Jack所有的投票
print ut.get_dict()
#输出结果是
{'count': 2, '_id': ObjectId('4eecbce5347307034e000001'), 'tag': u'Hacking', 'user': u'Jack'}
{'count': 1, '_id': ObjectId('4eecbce5347307034e000003'), 'tag': u'Food', 'user': u'Jack'}