#coding:utf-8
#!/usr/bin/python3
from selenium import webdriver
import time
import re
import importlib2
import sys
importlib2.reload(sys)
def startSpider():
  driver = webdriver.Chrome('/Users/zachary/zachary/chromedriver.exe') #這個是chormedriver的地址
  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號') #這里填寫你的QQ號
  driver.find_element_by_id('p').clear()
  driver.find_element_by_id('p').send_keys('QQ密碼') #這里填寫你的QQ密碼
  driver.find_element_by_id('login_button').click()
  time.sleep(2)
  #設置爬取內容保存路徑
  f = open('/Users/zachary/Documents/shuoshuo.txt','w')
  #---------------獲得g_qzonetoken 和 gtk
  html = driver.page_source
  '''g_qzonetoken=re.search('window\.g_qzonetoken = \(function\(\)\{ try\{return (.*?);\} catch\(e\)',html)#從網(wǎng)頁源碼中提取g_qzonetoken'''
  g_qzonetoken = "e794139a284d6ea9e0b26826e541b55df37d0667a3544f534de25aebdb64628d3ab75e1d7104bbb22a"
  cookie = {}#初始化cookie字典
  for elem in driver.get_cookies():#取cookies
    cookie[elem['name']] = elem['value']
  gtk=getGTK(cookie)#通過getGTK函數(shù)計算gtk
  #print(g_qzonetoken)
  #print(gtk)
  #--------------獲得好友列表  注意下面的鏈接
  driver.get('https://user.qzone.qq.com/proxy/domain/r.qzone.qq.com/cgi-bin/tfriend/friend_hat_get.cgi?hat_seed=1&uin=你的QQ號fupdate=1&g_tk='+str(gtk)+'&qzonetoken='+str(g_qzonetoken)+'&g_tk='+str(gtk))
  friend_list = driver.page_source
  friend_list = str( friend_list )
  abtract_pattern = re.compile('\"(.\d*)\":\{\\n"realname":"(.*?)"}',re.S)
  QQ_name_list = re.findall(abtract_pattern,str(friend_list)) #數(shù)組
  print(QQ_name_list)
  numList=dict()# numList => (QQnum:QQname) #列表
  for i in QQ_name_list:
    numList[str(i[0])]=str(i[1])
  begin = 0
  last_source = ""
  tag = 1
  first = 0
  firstTime=""
  #如果要爬取自己的說說，手動添加自己的qq號
  #numList['你的qq號']='你的名字'
  #print(numList)
  for key in numList.keys():
    QQnum = key
    QQname = numList[QQnum]
    if QQnum == "好友qq號": #根據(jù)qq號查找指定好友說說
      count = 1
      begin = 0
      while tag==1 :
        #-------------進入好友說說頁面                                    #'+QQnum+'       '+str(begin)+'
        #print("Begin:"+str(begin))
        driver.get('https://user.qzone.qq.com/proxy/domain/taotao.qq.com/cgi-bin/emotion_cgi_msglist_v6?uin='+QQnum+'&ftype=0&sort=0&pos='+str(begin)+'&num=40&replynum=200&g_tk='+str(gtk)+'&callback=_preloadCallback&code_version=1&format=jsonp&need_private_comment=1&qzonetoken='+str(g_qzonetoken)+'&g_tk='+str(gtk))
        try:
          msg_list_json = driver.page_source
        except:
          begin = begin + 40
          continue
        msg_list_json = str(msg_list_json)
        if last_source==msg_list_json :
          break
        else:
          last_source=msg_list_json
        #檢測是否沒有權限訪問
        abtract_pattern = re.compile(',"message":"(.*?)","name":',re.S)
        message = re.findall(abtract_pattern,str(msg_list_json))
        if message!=[]:
          if str(message[0])=='對不起,主人設置了保密,您沒有權限查看':#對不起,主人設置了保密,您沒有權限查看
            break
        #print(msg_list_json)
        #解析JSON
        #webDriver沒有現(xiàn)成的JSON解析器，所以采用獲取源碼的方式，然后使用正則表達式獲取具體細節(jié)
        msg_list_json = msg_list_json.split("msglist")[1]#拆分json，縮小范圍，也能加快解析速度
        msg_list_json = msg_list_json.split("smoothpolicy")[0]
        msg_list_json = msg_list_json.split("commentlist")[1:]
        #說說動態(tài)分4種：1、文字說說（或帶有配圖的文字說說）
        #       2、只有圖片的說說
        #       3、轉發(fā)，并配有文字
        #       4、轉發(fā)，不配文字
        for text in msg_list_json:
          # 1、先檢查說說，用戶是否發(fā)送了文字，如果沒有文字，正則表達式匹配無效
          abtract_pattern = re.compile('\}\],"content":"(.*?)","createTime":"(.*?)","created_time":(.*?),"',re.S)
          msg_time = re.findall(abtract_pattern,str(text))
          if msg_time!=[]:
            # 2、如果作者說說有文字，那么檢查是否有轉發(fā)內容
            msg = str(msg_time[0][0])
            sendTime = str(msg_time[0][1])
            abtract_pattern = re.compile('\}\],"content":"(.*?)"},"rt_createTime":"(.*?)","',re.S)
            text = text.split("created_time")[1]
            msg_time2 = re.findall(abtract_pattern,str(text))
            #合并發(fā)送內容 格式：評論+轉發(fā)內容
            if msg_time2!=[]:
              msg = msg +" 轉發(fā)內容:"+str(msg_time2[0][0])
          else:
            # 3、說說內容為空，檢查是否為 =>只有圖片的說說 or 轉發(fā)，不配文字
            #獲取正文發(fā)送時間 （發(fā)送時間分為：正文發(fā)送時間 or 轉發(fā)時間）
            abtract_pattern = re.compile('"conlist":null,"content":"","createTime":"(.*?)",',re.S)
            msgNull_time = re.findall(abtract_pattern,str(text))
            if msgNull_time!=[]:
              #如果有正文發(fā)送時間，那么就是這條說說僅含有圖片 =>只有圖片的說說
              msg = "圖片"
              sendTime = str(msgNull_time[0])
            else:
              #如果沒有正文發(fā)送時間，那么就是說這條說為 =>轉發(fā)，不配文字
              abtract_pattern = re.compile('\}\],"content":"(.*?)"},"rt_createTime":"(.*?)","',re.S)
              msg_time = re.findall(abtract_pattern,str(text))
              msg =" 轉發(fā)內容:"+str(msg_time[0][0])
              sendTime = str(msg_time[0][1])
          #寫入本地文件
          #f.write('{},{},{},{}\n'.format(str(QQname),str(QQnum),sendTime,msg))
          print(str(count)+" : "+str(QQname)+" : "+str(QQnum)+" : "+sendTime+" : "+msg)
          count = count + 1
        begin = begin + 40
def getGTK(cookie):
  hashes = 5381
  for letter in cookie['p_skey']:
    hashes += (hashes << 5) + ord(letter)
  return hashes & 0x7fffffff
startSpider()
print("爬取結束")