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
