#!usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author:DOULIHANG
@file: db_operation.py
@time: 2020/07/01
"""
import json
import pymysql
class DbOperation():
def __init__(self, host, user, password, database, port=3306):
"""
host：数据库服务器地址
user：登录数据库的用户名
password：登录数据库的密码
database：要连接的数据库名称
port：连接数据库的端口
"""
self.host = host
self.user = user
self.password = password
self.database = database
self.port = port
# 连接数据库
self.conn = pymysql.connect(host=self.host,
user=self.user,
password=self.password,
database=self.database,
port=self.port)
def select(self, sql):
try:
# 使用cursor操作游标
cursor = self.conn.cursor()
# 执行sql
cursor.execute(sql)
# 获取所有记录，返回格式为元组
results = cursor.fetchall()
return results
except:
print("SQL语句: {} 无法查询".format(sql))
def update(self, sql):
try:
# 使用cursor操作游标
cursor = self.conn.cursor()
# 执行sql
cursor.execute(sql)
# 提交到数据库执行
self.conn.commit()
except:
# 发生错误时回滚
self.conn.rollback()
def delete(self, sql):
try:
# 使用cursor操作游标
cursor = self.conn.cursor()
# 执行sql
cursor.execute(sql)
# 提交到数据库执行
self.conn.commit()
except:
# 发生错误时回滚
self.conn.rollback()
def insert(self, sql):
try:
# 使用cursor操作游标
cursor = self.conn.cursor()
# 执行sql
cursor.execute(sql)
# 提交到数据库执行
self.conn.commit()
except:
# 发生错误时回滚
self.conn.rollback()
def run_stored_procedure(self, args):
"""
调用存储过程
"""
try:
cur = self.conn.cursor()
# 调用存储过程，QSP_Clean_Data为存储过程名称，args为存储过程要传入的参数，格式为元组
cur.callproc('QSP_Clean_Data', args)
self.conn.commit()
except:
# 发生错误时回滚
self.conn.rollback()
def close(self):
# 关闭数据库连接
self.conn.close()
