import time

ONE_WEEK_IN_SECONDS = 7 * 24 * 60 * 60

VOTE_SCORE = 432

ARTICLES_PER_PAGE = 25

'''

用户对文章进行投票（目前没有考虑事务问题）

用户尝试投票时，使用 ZSCORE 命令检查记录文章发布时间的有序集合，判断文章的发布时间是否未超过一周，

如果仍可投票，使用 SADD 命令，尝试将用户添加到 记录文章已投票用户名单的集合里面，

如果成功，说明此用户是第一次为这个文章投票，此时使用 ZINCRBY 命令为文章添加 432 的评分数，

并使用 HINCRBY 命令对散列记录的文章投票数量进行更新

'''

def article_vote(conn, user, article):

    # 计算投票的截止时间

    cutoff = time.time() - ONE_WEEK_IN_SECONDS

    if conn.zscore('time:', article) < cutoff:

        # 超过投票截止时间则直接返回

        return

    # 获取文章 ID 部分

    article_id = article.partition(':')[-1]

    # 尝试将投票用户添加到文章已投票用户集合中

    if conn.sadd('voted:' + article_id, user):

        # 为文章添加评分

        conn.zincrby('score:', article, VOTE_SCORE)

        # 更新文章的得票数(+1)

        conn.hincrby(article, 'votes', 1)

'''

发布文章

通过对一个计数器执行 INCR 命令来生成文章id,

接着使用 SADD 将文章发布者的id 添加到记录文章已投票用户名单的

集合里面（这个集合需要使用 EXPIRE 命令设置为一周的过期时间，因为只有一周内的文章可以投票）,

之后使用 HMSET 命令来存储文章的相关信息，并执行两个 ZADD 命令，

将文章的初始评分和发布时间分别添加到两个相应的有序集合里面

'''

def post_article(conn, user, title, link):

    # 通过自增方式生成文章 ID

    article_id = str(conn.incr('article:'))

    # 生成并添加对文章投票的用户信息，同时设置过期时间

    voted = 'voted:' + article_id

    conn.sadd(voted, user)

    conn.expire(voted, ONE_WEEK_IN_SECONDS)

    # 存储文章信息

    now = time.time()

    article = 'article:' + article_id

    conn.hmset(article, {

        'title': title,

        'link': link,

        'poster': user,

        'time': now,

        'votes': 1,

    })

    # 保存文章发布时间及评分信息

    conn.zadd('score:', article, now + VOTE_SCORE)

    conn.zadd('time:', article, now)

    return article_id

'''

获取指定排序后的文章信息

使用 ZREVRANGE 取出多个文章id, 再对每个文章id 执行一次 HGETALL 来取出文章的详细信息

'''

def get_articles(conn, page, order='score:'):

    start = (page - 1) * ARTICLES_PER_PAGE

    end = start + ARTICLES_PER_PAGE - 1

    # 逆序(从大到小)获取数据

    ids = conn.zrevrange(order, start, end)

    articles = []

    for id in ids:

        article_data = conn.hgetall(id)

        article_data['id'] = id

        articles.append(article_data)

    return articles