v2 抽奖代码

之前说了要抽两本书的，今天搞完论文的事情，刚去吃了散伙饭回来。草草写了个抽奖脚本，抽帖子 https://www.v2ex.com/t/457319 和 https://www.v2ex.com/t/459938 里的人。写好之后发现可以用楼层数抽就好了....尴尬。贴上渣代码和抽奖结果。不知道如何在 v2 上优雅的放代码哈，随意排版了下。 def get_html(url):

cookies = ''

re = request.Request(url)

re.add_header('cookie', cookies)

re.add_header('User-Agent', 'Mozilla/5.0 (Windows NT 6.1; WOW64) '
                            'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36')
                            
return request.urlopen(re).read().decode('utf-8')

def get_user_by_id(id):

page = etree.HTML(get_html('https://www.v2ex.com/t/' + str(id))) 
message_num = page.xpath('//*[@id="Main"]/div[4]/div[1]/span/text()[1]')[0].split(' ')[0]  # 回复数 确定翻页次数
print('共有', message_num, '条回复')
page = math.ceil(int(message_num) / 100)
print('共有', page, '页')
user_list = []
for i in range(1, page + 1):
    url = 'https://www.v2ex.com/t/' + str(id) + '?p=' + str(i)
    print(url)
    divs = etree.HTML(get_html(url))
    res = divs.xpath('//*[@id="Main"]/div[4]//a[@class="dark"]')
    for info in res:
        user_list.append(info.text)
        print(info.text)
    print(len(user_list))
return user_list

def main():

id_list = [459938, 457319]
user_list = []
for id in id_list:
	user_list.extend(get_user_by_id(id))
print('共有', len(user_list), '个人回复')
print('去重后还有', len(set(user_list)), '个人')
prizer = random.sample(set(user_list), 2)
print(prizer)

共有 322 个人回复去重后还有 298 个人 ['JohnChiu', 'Keepall']

没有特殊要求我就按顺序给你们发第一第二本了如果有的话你们自行商量哈。

vx:cTM5ODU1NjA1Mw==

ps 代码在我机器上运行是没问题的哈[狗头]