from selenium import webdriver
import time
import json
from selenium.webdriver.common.by import By

#創(chuàng)建webdriver 對象，指明使用chrome 瀏覽器驅(qū)動
wd = webdriver.Chrome()

wd.implicitly_wait(10)

#調(diào)用webdriver 對象的get方法，可以讓瀏覽器打開指定網(wǎng)址
wd.get('https://zhidao.baidu.com/activity/iknowduck/level?actId=47')
input('網(wǎng)頁端登錄百度賬號后，請按回車鍵')
cookie= wd.get_cookies()
# #將獲得cookie 的信息打印
print(cookie)
with open('baiducookies.txt','w') as f:
    # 將cookies保存為json格式
    f.write(json.dumps(wd.get_cookies()))
    f.close()

wd.add_cookie(cookie)

from selenium import webdriver
import time
import json
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
from chatgpt_demo import chatgpt

#創(chuàng)建webdriver 對象，指明使用chrome 瀏覽器驅(qū)動
wd= webdriver.Chrome()
wd.implicitly_wait(10)

#調(diào)用webdriver 對象的get方法，可以讓瀏覽器打開指定網(wǎng)址
wd.get('https://zhidao.baidu.com/activity/iknowduck/level?actId=47')
# 首先清除由于瀏覽器打開已有的cookies
wd.delete_all_cookies()
time.sleep(10)
#打開cookie文本，使用已保存的cookie登錄
with open('baiducookies.txt','r') as f:
    # 使用json讀取cookies 注意讀取的是文件 所以用load而不是loads
    cookies_list = json.load(f)
    for cookie in cookies_list:
        wd.add_cookie(cookie)
wd.refresh() #刷新頁面
wd.refresh() #刷新頁面
time.sleep(6)
#獲取當前窗口的title
First_handle = wd.current_window_handle
j = 0
while j < 60:
#try:
    for i in range(9):
        n = 1
        #點擊第一個問題
        questions = wd.find_elements(By.CSS_SELECTOR, 'div.q-item > div.q-title > span:nth-child(2) ')
        for question in questions:
            print('第{}個問題：'.format(n), question.text)
            n += 1

            #將第一個問題，輸入catgpt在線智能回答
            message = chatgpt(question.text)
            print(message)
            #with open("./output/{}.doc".format(prompt), "w") as of:
            #    of.write(message)
            titles = question.text + '_百度知道'
            time.sleep(1)
            #點擊“回答”，按鈕
            # answer = wd.find_element(By.CSS_SELECTOR,'.goto-anwser-btn')
            # print('點擊:',answer.text)
            wd.add_cookie(cookie) #帶cookie
            question.click() #點擊問題
            time.sleep(3)
            #切換窗口，點擊“去回答”按鈕時，打開了新的窗口，但WebDriver對象對應的 還是老窗口。這里要跟隨跳轉(zhuǎn)
            for handle in wd.window_handles:
                # 先切換到該窗口
                wd.switch_to.window(handle)
                # 得到該窗口的標題欄字符串，判斷是不是我們要操作的那個窗口
                if 'titles' in wd.title:
                    # 如果是，那么這時候WebDriver對象就是對應的該該窗口，正好，跳出循環(huán)，
                    break
            #print(wd.title)

            #新窗口下，不管是否已有其他回答，都點擊“我來答按鈕”
            element = wd.find_element(By.CSS_SELECTOR, '#answer-bar')
            element.click()#點擊“我來答按鈕”
            #輸入文本
            #新窗口，切換到第一級iframe框下，正文內(nèi)容
            wd.switch_to.frame('ueditor_0')
            element = wd.find_element(By.CSS_SELECTOR,'body > p')
            print('正在輸入答案')
            element.send_keys(message)
            #先返回到主html,點擊提交按鈕
            wd.switch_to.default_content()
            time.sleep(50)
            element = wd.find_element(By.CSS_SELECTOR,'div.addons.line > a')
            print(element.text)
            wd.add_cookie(cookie)  # 帶cookie
            element.click()
            time.sleep(5)
            #做完一系列操作后關閉school_handle
            wd.close()
            # 切換窗口會第一個窗口
            wd.switch_to.window(First_handle)
#except:
    print("出錯{}次,正在重新運行程序。".format(j))
    j += 1
    wd.switch_to.window(First_handle)
    time.sleep(5)
    wd.refresh()  # 刷新頁面
else:
    print('出錯太多次啦，程序已結(jié)束')