redis使用bitmap实现签到

import redis
import datetime
import calendar
​
r = redis.Redis(
    host="127.0.0.1",
    port=6379,
    db=1,
    password=""
)
​
​
def check_input(func):
    def wrapper(*args, **kwargs):
        if not isinstance(args[1], int):
            raise ValueError(f"User_id must be int, and your input is {type(args[1])}")
        return func(*args, **kwargs)
​
    return wrapper
​
​
class RedisCheckIn:
​
    _private_key = "_check_in_"
​
    def __init__(self):
        pass
​
    @check_input
    def sign(self, user_id: int) -> int:
        # 用户签到
        today = str(datetime.datetime.now())[:10]
        return r.setbit(self._private_key+today, user_id, 1)
​
    @check_input
    def sign_status(self, user_id: int) -> int:
        # 用户今日签到状态
        today = str(datetime.datetime.now())[:10]
        return r.getbit(self._private_key+today, user_id)
​
    @check_input
    def week_sign_status(self, user_id: int) -> list:
        # 求出这个周的签到状况
        now = datetime.datetime.now()
        # 周一是1 周日是7 now.weekday()则是周一是0，周日是6
        weekday = now.isoweekday()
        with r.pipeline(transaction=False) as p:
            for d in range(weekday):
                check_day = str(now-datetime.timedelta(days=1)*d)[:10]
                p.getbit(self._private_key + check_day, user_id)
            data = p.execute()[::-1]
        # 比如周三的时候我们只查3次getbit，然后剩下补0
        data.extend([0] * (7 - len(data)))
        return data
​
    @check_input
    def month_sing_status(self, user_id: int) -> list:
        # 求出这个月的某个用户签到状况
        now = datetime.datetime.now()
        day = now.day
        with r.pipeline(transaction=False) as p:
            for d in range(day):
                check_day = str(now - datetime.timedelta(days=1) * d)[:10]
                p.getbit(self._private_key + check_day, user_id)
            data = p.execute()[::-1]
        # 获取当月天数，还没到的天数补0
        month_range = calendar.monthrange(now.year, now.month)
        data.extend([0] * (month_range[1] - len(data)))
        return data
​
    @check_input
    def week_sign_num(self, user_id: int) -> int:
        # 求出这个周的签到次数
        return sum(self.week_sign_status(user_id))
​
    @check_input
    def month_sign_num(self, user_id: int) -> int:
        # 求出这个月的签到次数
        return sum(self.month_sing_status(user_id))
​
    @check_input
    def today_sign_all_num(self) -> int:
        # 求出当天有多少用户签到
        today = str(datetime.datetime.now())[:10]
        return r.bitcount(self._private_key + today)
​
​
redis_sign_in = RedisCheckIn()