python selenium抓取网页数据

在db文件夹中dbmysql.py,在db文件夹中新增__init__.py,内容为空

import pymysql

class DBMysql:
    host = localhost
    username = root
    password = root
    db_name = test

    @classmethod
    def connect_mysql(self):
        connect = pymysql.connect(host=self.host, user=self.username, password=self.password, database=self.db_name)
        return connect

    @classmethod
    def insert_table(self, sql):
        db = self.connect_mysql()
        cur = db.cursor()
        insert_id = 0
        try:
            cur.execute(sql)
            insert_id = int(db.insert_id())
            db.commit()
        except pymysql.Error as e:
            print("创建数据库连接失败|Mysql Error %d: %s" % (e.args[0], e.args[1]))
            print(sql)
            db.rollback()
        db.close()
        return insert_id

    @classmethod
    def find_sql(self,sql):
        db = self.connect_mysql()
        cur = db.cursor()
        cur.execute(sql)
        obj = cur.fetchone()
        db.close()
        return obj

在sele文件夹中index.py

经验分享 程序员 微信小程序 职场和发展