from selenium import webdriver
import random
import time
import pytz
from datetime import datetime
import re
import pandas as pd


user_agent = [
    "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; .NET4.0C; .NET4.0E; .NET CLR 2.0.50727; .NET CLR 3.0.30729; .NET CLR 3.5.30729; InfoPath.3; rv:11.0) like Gecko",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)",
    "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
    "Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
    "Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
    "Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
    "Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
    "Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
    "UCWEB7.0.2.37/28/999",
    "NOKIA5700/ UCWEB7.0.2.37/28/999",
    "Openwave/ UCWEB7.0.2.37/28/999",
    "Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999",
    # iPhone 6：
	"Mozilla/6.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/536.26 (KHTML, like Gecko) Version/8.0 Mobile/10A5376e Safari/8536.25",

    ]
voteDict ={}
df0 = pd.DataFrame(columns=['日期', '第一', '票数', '新增'])
df1 = pd.DataFrame(columns=['日期', '第二', '票数', '新增'])
df2 = pd.DataFrame(columns=['日期', '第三', '票数', '新增'])
df3 = pd.DataFrame(columns=['日期', '第四', '票数', '新增'])
while True:
    option = webdriver.ChromeOptions()
    header = random.choice(user_agent)
    option.add_argument(f'user-agent={header}')
    browser = webdriver.Chrome(options=option)

    browser.implicitly_wait(50)
    browser.get("https://sinanews.sina.cn/ent/weibo_night_sublist_2019.d.html?wm=3049_0107")

    i = 0
    htmls = browser.find_elements_by_class_name('survey_answer')
    currentChineseTime = datetime.now().astimezone(pytz.timezone('Asia/Shanghai')).strftime('%Y-%m-%d %H:%M:%S')
    print(currentChineseTime)

    while i != 4:
        info = htmls[i].text.split('投票')[0]
        voteNumber = re.findall(r"\d+",info)
        for v in voteNumber:
            vote = v
        outputName = info.split(vote)[0]
        if outputName not in voteDict:
            voteDict[outputName] = 0
        newIncrese = int(vote)-voteDict[outputName]
        outputVote = int(vote)
        
        if i == 0:
            dfNewRow = pd.Series({'日期':currentChineseTime, '第一':outputName,'票数':outputVote,'新增':newIncrese})
            df0 = df0.append(dfNewRow,ignore_index=True)
            df0.to_csv('/Users/lanzhou/Desktop/test/needToGo/yingxiangli1.csv', index = False, mode='w',encoding="utf_8_sig")
        if i == 1:
            dfNewRow = pd.Series({'日期':currentChineseTime, '第二':outputName,'票数':outputVote,'新增':newIncrese})
            df1 = df1.append(dfNewRow,ignore_index=True)
            df1.to_csv('/Users/lanzhou/Desktop/test/needToGo/yingxiangli2.csv', index = False, mode='w',encoding="utf_8_sig")
        if i == 2:
            dfNewRow = pd.Series({'日期':currentChineseTime, '第三':outputName,'票数':outputVote,'新增':newIncrese})
            df2 = df2.append(dfNewRow,ignore_index=True)
            df2.to_csv('/Users/lanzhou/Desktop/test/needToGo/yingxiangli3.csv', index = False, mode='w',encoding="utf_8_sig")
        if i == 3:
            dfNewRow = pd.Series({'日期':currentChineseTime, '第四':outputName,'票数':outputVote,'新增':newIncrese})
            df3 = df3.append(dfNewRow,ignore_index=True)
            df3.to_csv('/Users/lanzhou/Desktop/test/needToGo/yingxiangli4.csv', index = False, mode='w',encoding="utf_8_sig")
       
        voteDict[outputName] = int(vote)
        i = i+1
    browser.close()
    time.sleep(300)









