课程文字课时topic vote.pdfVIP

#encoding:utf-8

fromredisimportWatchError

defmake_vote_up_key(topic_id):

returnbbstopic+str(topic_id)+vote_up

defmake_vote_down_key(topic_id):

returnbbstopic+str(topic_id)+vote_down

classTopicVote:

对帖子进行投票，可以投支持票也可以投票。

definit(self,client,topic_id):

设置要被投票的帖子。

self.client=client

self.vote_up_set=make_vote_up_key(topic_id)

self.vote_down_set=make_vote_down_key(topic_id)

defis_voted(self,user_id):

检查用户是否已经对这个帖子投过票。

returnself.client.sismember(self.vote_up_set,user_id)or\

self.client.sismember(self.vote_down_set,user_id)

defvote_up(self,user_id):

用户对这个帖子投支持票。

ifnotself.is_voted(user_id):

self.client.sadd(self.vote_up_set,user_id)

defvote_down(self,user_id):

用户对这个帖子投票。

ifnotself.is_voted(user_id):

self.client.sadd(self.vote_down_set,user_id)

defundo(self,user_id):

撤销用户的投票。

self.client.srem(self.vote_up_set,user_id)

self.client.srem(self.vote_down_set,user_id)

defvote_up_count(self):

返回帖子目前获得的支持票数量。

returnself.client.scard(self.vote_up_set)

defvote_down_count(self):

返回帖子目前获得的