from selenium import webdriver

from selenium.webdriver.common.keys import Keys

from bs4 import BeautifulSoup

from selenium.webdriver.chrome.options import Options

import time

import selenium

import random

# from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

def driver_open():

    # dcap = dict(DesiredCapabilities.PHANTOMJS)

    # dcap["phantomjs.page.settings.userAgent"] = (r"Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3100.0 Safari/537.36")

    options = Options()

    options.add_argument('--log-level=3')

    driver = webdriver.Chrome(executable_path='E:\chromedriver\chromedriver.exe', chrome_options=options)

    return driver

def next_windows(driver,current_window):

    window_handles = driver.window_handles

    for window_handle in window_handles:

        if current_window != window_handle:

            handleb = window_handle

    driver.switch_to.window(handleb)

    return  driver

def next_windows_handle(driver,current_window):

    window_handles = driver.window_handles

    for window_handle in window_handles:

        if current_window != window_handle:

            handleb = window_handle

    return handleb

if __name__ == '__main__':

    learn_dict = {}

    driver = driver_open()

    driver.maximize_window()

    driver.implicitly_wait(10)  # seconds

    # # Theoretical study

    theoretical_study_url = 'xxxxx'

    driver.get(theoretical_study_url)

    time.sleep(20)

    current_window = driver.current_window_handle

    print(current_window)

    # 浏览6篇文章

    article = 6

    elements = driver.find_elements_by_xpath('//span[contains(@class,"text")]')

    for element in elements:

        element.click()

        driver = next_windows(driver,current_window)

        client_height = driver.execute_script("return document.body.clientHeight;")

        # scroll_height = driver.execute_script("return document.body.scrollWidth;")

        # print(scroll_height)

        init_position = 0

        while (client_height > init_position):

            init_position = init_position + 200

            js_string = "window.scrollTo(0," + str(init_position) + ");"

            driver.execute_script(js_string)

            sleep_time = random.randint(2, 4)

            time.sleep(sleep_time)

        time.sleep(3)

        driver.close()

        driver.switch_to.window(current_window)

        article = article - 1

        if(article == 0):

            break

    driver.get('xxxxxx')

    time.sleep(3)

    window_handles = driver.window_handles

    for window_handle in window_handles:

        current_window = window_handle

    driver.switch_to.window(current_window)

    # 浏览6个视频

    videos_count = 6

    videos = driver.find_elements_by_xpath('//div[contains(@class,"innerPic")]')

    for video in videos:

        driver.switch_to.window(current_window)

        video.click()

        driver = next_windows(driver, current_window)

        js_string = "window.scrollTo(0,400);"

        driver.execute_script(js_string)

        time_sleep = random.randint(10,50)

        time.sleep(time_sleep)

        driver.close()

        videos_count = videos_count - 1

        if(videos_count == 0):

            break

    driver.quit()