python selenium入门（三）打开网页爱奇艺，获取影视名，创建数据库，把获取的数据写入数据库。

打开网页爱奇艺，获取影视名，创建数据库，把获取的数据写入数据库。
新增数据的时候做个判断，查询一下数据库有没有这个数据，没有就新增。有就不做任何处理。

from time import sleep

from selenium import webdriver

from sqlalchemy import create_engine

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy import Column,Integer,String

from sqlalchemy.orm import sessionmaker


driver = webdriver.Chrome()

engine=create_engine(
    "mysql+pymysql://root:byb123456!@localhost:3306/test?charset=utf8",
    echo=True)

Base=declarative_base()
class mytablemove(Base):
    __tablename__='mytablemove'
    id=Column(Integer,primary_key=True)
    name=Column(String(50))

Base.metadata.create_all(engine)

DBSession=sessionmaker(bind=engine)
session=DBSession()

driver.get('https://www.iqiyi.com/')
sleep(2)

driver.find_element_by_xpath(
    '/html/body/div[1]/div/div/div[1]/div[3]/div[2]/div/div[4]/div[2]/div/div[1]/h2/div/ul/li/div/a').click()
sleep(2)

handles = driver.window_handles  # 获取当前打开的所有窗口的句柄
print(handles)

driver.switch_to.window(handles[1])  # 切换到第二个窗口的句柄

submit = driver.find_elements_by_class_name('title-link')
for i in submit:
    # print(i.text)
    d = session.query(mytablemove).filter_by(name=i.text).first()
    if not d:
        new_data = mytablemove(name=str(i.text))
        session.add(new_data)
session.commit()
session.close()

sleep(2)

driver.quit()