from urllib.request import urlretrieve

from selenium import webdriver
#要想使用selenium的webdriver 里的函数，首先把包导进来

#from bs4 import BeautifulSoup

import time

from selenium.webdriver import ActionChains
#导入鼠标操作

def login(login_qq,password, business_qq):
    '''
    登陆
    :param login_qq: 登陆用的QQ
    :param password: 登陆的QQ密码
    :param business_qq: 业务QQ
    :return: driver
    '''
    driver = webdriver.Chrome()

    driver.get('https://user.qzone.qq.com/')  # URL
    driver.implicitly_wait(10)
    # 隐示等待，为了等待充分加载好网址
    driver.find_element_by_id('login_div')
    #定位到登陆框的最外层DIV
    driver.switch_to.frame('login_frame')
    # 切到输入账号密码的frame
    driver.find_element_by_id('switcher_plogin').click()
    ##点击‘账号密码登录’
    driver.find_element_by_id('u').clear()
    ##清空账号栏
    driver.find_element_by_id('u').send_keys(login_qq)
    #  定位元素U,输入账号
    driver.find_element_by_id('p').clear()
    # 清空密码栏
    driver.find_element_by_id('p').send_keys(password)
    # 输入密码
    driver.find_element_by_id('login_button').click()
    # 点击‘登录’
    driver.switch_to.default_content()
    #切到frame中之后，我们若像继续操作主文档的元素，则需切回主文档
    driver.implicitly_wait(10)

    #等待
    time.sleep(5)

    try:
        driver.find_element_by_id('QM_OwnerInfo_Icon')
        #找到头像元素，代表页面成功
        return driver
    except:
        print('不能访问' + business_qq)
        return None



def get_photo(driver):

    # 照片下载路径
    photo_path = "C:/Users/Administrator/Desktop/photo/{}.jpg"

    # 相册索引
    photoIndex = 1

    while True:
        # 回到主文档
        driver.switch_to.default_content()
        # driver.switch_to.parent_frame()
        # 点击头部的相册按钮
        driver.find_element_by_xpath('//*[@id="menuContainer"]/div/ul/li[3]/a').click()
        #等待加载
        driver.implicitly_wait(10)
        time.sleep(3)
        # 切换 frame

        #重改的地方
        #driver.switch_to.frame('app_canvas_frame')
        # 改为ID定位
        driver.switch_to.frame('tphoto')
        # 各个相册的超链接
        a = driver.find_elements_by_class_name('album-cover')

        # 单个相册
        a[photoIndex].click()

        driver.implicitly_wait(10)
        time.sleep(3)
        # 相册的第一张图
        p = driver.find_elements_by_class_name('item-cover')[0]
        p.click()
        time.sleep(3)

        # 相册大图在父frame，切换到父frame
        driver.switch_to.parent_frame()
        # 循环相册中的照片
        while True:
            # 照片url地址和名称
            img = driver.find_element_by_id('js-img-disp')
            src = img.get_attribute('src').replace('&t=5', '')
            name = driver.find_element_by_id("js-photo-name").text

            # 下载
            urlretrieve(src, photo_path.format(name))

            # 取下面的 当前照片张数/总照片数量
            counts = driver.find_element_by_xpath('//*[@id="js-ctn-infoBar"]/div/div[1]/span').text

            counts = counts.split('/')
            # 最后一张的时候退出照片浏览
            if int(counts[0]) == int(counts[1]):
                # 右上角的 X 按钮
                driver.find_element_by_xpath('//*[@id="js-viewer-main"]/div[1]/a').click()
                break
            # 点击 下一张，网页加载慢，所以10次加载
            for i in (1, 10):
                if driver.find_element_by_id('js-btn-nextPhoto'):
                    n = driver.find_element_by_id('js-btn-nextPhoto')
                    ActionChains(driver).click(n).perform()
                    break
                else:
                    time.sleep(5)

        # 相册数量比较，是否下载了全部的相册
        photoIndex = photoIndex + 1
        if len(a) <= photoIndex:
            break
if __name__ == '__main__':

    driver = login('***', '***', '2222222')
    if driver:
        #get_shuoshuo(driver)
        get_photo(driver)