selenium抖音

from selenium import webdriver
import time
import pymysql


def login():
    '''
    模拟登录（不是）
    '''
    driver.get(
        'https://www.douyin.com/search/%E5%A4%AE%E8%A7%86%E7%96%AB%E8%8B%97?publish_time=0&sort_type=1&source=normal_search&type=video')
    driver.maximize_window()
    time.sleep(10)
    login = driver.find_element_by_css_selector('#_285c63f4da53bd5cedc023b4fdd71412-scss > button')  # 找到安全登录按钮
    login.click()
    time.sleep(15)  # 手动扫码登录


def Transfer(driver):
    '''
    下拉滑动条，实现翻页功能
    '''
    try:
        driver.execute_script("window.scrollBy(0,document.body.scrollHeight)", "")  ## 移动到页面最底部
        time.sleep(2)
        driver.execute_script("window.scrollBy(0,document.body.scrollHeight)", "")
        time.sleep(2)
        driver.execute_script("window.scrollBy(0,document.body.scrollHeight)", "")
    except:
        pass
    return "Transfer successfully \n"


def get_news(i):
    '''
        获取新闻发布者，新闻标题，以及新闻热度
        建立当前新闻的数据库
        '''
    try:
        news_user = driver.find_element_by_css_selector(
            f'#root > div > div:nth-child(2) > div > div.cdfb2696b192e707e529d93274fe5752-scss > div:nth-child(2) > ul > li:nth-child({i}) > div > div > a > p > span > span > span > span > span').text
        news_title