python3对MySQL数据库操作进行封装
(2020-06-19 22:27:18)
标签:
it |
分类: python |
一、写一个配置文件,便于维护程序配置信息
建立一个python package名为 config的包,在下面建立名为db_config的python file,内容如下:
二、读取配置文件信息,连接MySQL数据库进行数据库操作
建立一个python package名为util的包,在下面建立DBHelper的python file,内容如下:
建立一个python package名为 config的包,在下面建立名为db_config的python file,内容如下:
db_config = {
'host':'127.0.0.1',
'port':3306,
'username':'root',
'password':'root',
'database':'gaiisf0426',
'charset':'utf8'
}
二、读取配置文件信息,连接MySQL数据库进行数据库操作
建立一个python package名为util的包,在下面建立DBHelper的python file,内容如下:
import pymysql
import logging
import sys
from config.db_config import db_config
# 加入日志
# 获取logger实例
logger = logging.getLogger("baseSpider")
# 指定输出格式
formatter = logging.Formatter('%(asctime)s\
%(levelname)-8s:%(message)s')
# 文件日志
file_handler = logging.FileHandler("operation_database.log")
file_handler.setFormatter(formatter)
# 控制台日志
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(formatter)
# 为logge添加具体的日志处理器
logger.addHandler(file_handler)
logger.addHandler(console_handler)
logger.setLevel(logging.INFO)
class DBHelper():
# 构造函数,初始化数据库连接
def __init__(self,sql,params=None):
self.sql = sql
self.params = params
self.conn = None
self.cur = None
def connectiondatabase(self):
print(db_config['host'],db_config['username'],db_config['password'],db_config['database'],db_config['charset'])
try:
self.conn = pymysql.connect(db_config['host'],db_config['username'],
db_config['password'],db_config['database'],charset=db_config['charset'])
except:
logger.error("connectDatabase failed")
return False
self.cur = self.conn.cursor()
return True
# 关闭数据库
def closedatabase(self):
# 如果数据打开,则关闭;否则没有操作
if self.conn and self.cur:
self.cur.close()
self.conn.close()
return True
# 执行数据库的sq语句,主要用来做插入操作
def execute(self):
self.connectiondatabase()
try:
if self.conn and self.cur:
# 正常逻辑,执行sql,提交操作
self.cur.execute(self.sql,self.params)
self.conn.commit()
except:
logger.error("execute failed: " + self.sql)
logger.error("params: " + self.params)
self.closedatabase()
return False
return True
# 用来查询表数据
def select(self):
self.connectiondatabase()
self.cur.execute(self.sql,self.params)
result = self.cur.fetchall()
print(result)
return result
三、调用方法
在主运行文件中代码如下:
from util.DBHelper import DBHelper
sql='select * from t_s_base_user'
t=DBHelper(sql)
t.select()
t.closedatabase()