python基础类知识~pymysql封装类

一简介:咱们来介绍下 DBA常用的几个类

二 具体代码

#!/usr/bin/python3
import pymysql
import sys
class DBHelper:
def __init__(self,host,user,pwd,db):
self.host = host
self.user = user
self.pwd = pwd
self.db = db
self.conn = None
self.cur = None

def connectDatabase(self):
try:
self.conn = pymysql.connect(self.host, self.user,self.pwd, self.db,charset='utf8')
except:
return False
self.cur = self.conn.cursor()
return True

def close(self):
if self.conn and self.cur:
self.cur.close()
self.conn.close()
return True

def execute(self, sql, params=None):
print(self.connectDatabase())
try:
if self.conn and self.cur:
self.cur.execute(sql, params)
print("ok")
self.conn.commit()
except:
self.close()
return False
return True

def fetchall(self, sql, params=None):
self.execute(sql,params)
return self.cur.fetchall()

三 调用方法

import  DBHelper

conn=DBHelper(host,user,pwd,db) 建立连接

conn.execute(sql) 执行DML操作

conn.fetchall(sql)查询全部数据

conn.close() 关闭