import
redis
class
LikeMessage(
object
):
def
__init__(
self
, msg_id, client):
self
.msg_id
=
msg_id
self
.client
=
client
self
.key
=
"weibo::message::"
+
str
(msg_id)
+
"::like"
def
like(
self
, user_id):
self
.client.sadd(
self
.key, user_id)
def
is_liking(
self
, user_id):
return
self
.client.sismember(
self
.key, user_id)
def
undo(
self
, user_id):
self
.client.srem(
self
.key, user_id)
def
count(
self
):
return
self
.client.scard(
self
.key)
def
get_all_liking_user(
self
):
if
not
self
.client.exists(
self
.key):
return
False
else
:
return
self
.client.smembers(
self
.key)
if
__name__
=
=
"__main__"
:
redis_client
=
redis.StrictRedis()
vote
=
LikeMessage(
65535
, redis_client)
vote.like(
10010
)
vote.like(
10086
)
vote.like(
10000
)
print
(vote.count())
print
(vote.is_liking(
10086
))
vote.undo(
10086
)
print
(vote.get_all_liking_user())