qzone.py

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import time
import requests
import demjson
import re
import datetime
import getpass


qq = ''
pwd = ''


def print_time():
    print(datetime.datetime.now(), end=' ')


def get_gtk(p_skey):
    hash=5381
    for i in p_skey:
        hash += (hash << 5)+ord(i)

    print_time()
    print('生成gtk')
    return hash & 0x7fffffff


def change_cookie(cookie):
    s = ''
    for c in cookie:
        s = s + c['name'] + '=' + c['value'] + '; '

    return s


def check_time():
    now = datetime.datetime.now()
    hour = str(now)[11:13]
    minute = str(now)[14:16]
    second = str(now)[17:19]

    if 0 == int(hour) % 6 and minute == '00' and int(second) < 30:
        return True
    else:
        return False


def get_cookie():
    chrome_options = Options()
    chrome_options.add_argument('--headless')
    driver = webdriver.Chrome(chrome_options=chrome_options)

    driver.get('https://qzone.qq.com/')

    driver.switch_to.frame('login_frame')

    driver.find_element_by_id('switcher_plogin').click()
    driver.find_element_by_id('u').clear()
    driver.find_element_by_id('u').send_keys(qq)
    driver.find_element_by_id('p').clear()
    driver.find_element_by_id('p').send_keys(pwd)
    driver.find_element_by_id('login_button').click()

    time.sleep(1)

    driver.find_element_by_id('QZ_Body').click()

    cookie = driver.get_cookies()

    # print(cookie)

    driver.close()
    driver.quit()

    print_time()
    print('提取cookie')

    return cookie


def get_args():
    cookie = get_cookie()

    for c in cookie:
        if c['name'] == 'p_skey':
            p_skey = c['value']
            break

    cookie = change_cookie(cookie)

    # print(p_skey)

    gtk = get_gtk(p_skey)

    return cookie, gtk


def do_like(d, gtk, headers):
    url = 'https://user.qzone.qq.com/proxy/domain/w.qzone.qq.com/cgi-bin/likes/internal_dolike_app?g_tk=' + str(gtk)

    body = {
        'qzreferrer': 'http://user.qzone.qq.com/' + qq,
        'opuin': qq,
        'from': 1,
        'active': 0,
        'fupdate': 1
    }

    try:
        html = d['html']

        # print(html)
        # unikey = re.search(r'data-unikey=\"http:[^"]*\"', html).group(0)
        # curkey = re.search(r'data-curkey=\"http:[^"]*\"', html).group(0)
        # print(unikey, curkey)

        temp = re.search('data-unikey="(http[^"]*)"[^d]*data-curkey="([^"]*)"[^d]*data-clicklog=("like")[^h]*href="javascript:;"', html);

        if temp is None:
            return

        unikey = temp.group(1);
        curkey = temp.group(2);

        # print(unikey, curkey)

        body['unikey'] = unikey
        body['curkey'] = curkey
        body['appid'] = d['appid']
        body['typeid'] = d['typeid']
        body['fid'] = d['key']

        r = requests.post(url, data=body, headers=headers)

        if 200 == r.status_code:
            print_time()
            print('给 ' +  d['nickname'] + ' 点赞')

    except:
        return


def get_content(headers, gtk):
    try:
        r = requests.get('http://ic2.s8.qzone.qq.com/cgi-bin/feeds/feeds3_html_more?uin=0924761163&scope=0&view=1&daylist=&uinlist=&gid=&flag=1&filter=all&applist=all&refresh=0&aisortEndTime=0&aisortOffset=0&getAisort=0&aisortBeginTime=0&pagenum=1&externparam=offset%3D6%26total%3D97%26basetime%3D1470323193%26feedsource%3D0&firstGetGroup=0&icServerTime=0&mixnocache=0&scene=0&begintime=0&count=10&dayspac=0&sidomain=cnc.qzonestyle.gtimg.cn&useutf8=1&outputhtmlfeed=1&getob=1&g_tk=' + str(gtk), headers=headers)

        r = r.text[10:-2]

        r = demjson.decode(r)

        data = r['data']['data']

        print_time()
        print('获取了 ' + str(len(data)) + ' 条说说')

        return data
    except:
        return []


def main():

    print_time()
    print('程序运行...')

    global qq
    global pwd

    qq = input('QQ:')
    pwd = getpass.getpass('Password:')

    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36'
    }

    cookie, gtk = get_args()
    headers['Cookie'] = cookie

    while True:
        time.sleep(1)

        if check_time():
            cookie, gtk = get_args()
            headers['Cookie'] = cookie

            print_time()
            print('更新了 cookie 和 gtk')

        data = get_content(headers, gtk)

        for d in data:
            do_like(d, gtk, headers)


if __name__ == '__main__':
    main()