python对mysql数据库操作之代码优化(二)
python对mysql数据库操作之代码优化(二)
之前编写了对mysql数据库的增，删，改，查的方法，再见这部分的代码：
import MySQLdb
class MySQLHelper(object):
def __init__(self):
pass
@property
def selectMySQL(self):
try:
conn=MySQLdb.connect(host='localhost',user='root',
passwd='server',db='db',,charset='utf8')
cur=conn.cursor()
except Exception,e:
print u'操作mysql数据库失败'
else:
cur.execute('select * from user;')
data=cur.fetchall()
for d in data:
print d
finally:
cur.close()
conn.close()
@property
def insertMySQL(self):
try:
conn=MySQLdb.connect(host='localhost',user='root',
passwd='server',db='db',,charset='utf8')
cur=conn.cursor()
sql="insert into user(id,name,sex,email) VALUES (%s,%s,%s,%s)"
params=(1,'selenium2','boy','USA@qq.com')
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'插入后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
@property
def updateMySQL(self):
try:
conn=MySQLdb.connect(host='localhost',user='root',
passwd='server',db='db',,charset='utf8')
cur=conn.cursor()
sql="update user set NAME=%s where id=%s"
params=('selenium webdriver',1)
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'修改后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
@property
def deleteMySQL(self):
try:
conn=MySQLdb.connect(host='localhost',user='root',
passwd='server',db='db',,charset='utf8')
cur=conn.cursor()
sql="delete from USER WHERE id=%s"
params=(1)
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'删除后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
发现在增，删，改，查方法中，都有MySQLdb.connect(host='localhost',user='root',passwd='server',db='db',,charset='utf8')，这意味着什么？意味着对mysql操作的方法，都得写MySQLdb.connect(host='localhost',user='root',passwd='server',db='db',,charset='utf8')，有多少个方法，就得写多少个，这样明显是不合理的，很明显是重复造轮子，而且也不符合python的简单而优雅的设计理念，那么开始重构这部分的代码,把一个配置文件config.py或者写一个类Config，把连接mysql的部分写成一个静态方法，有操作mysql的方法，直接调用即可。这里创建config.py的模块，见config.py模块的源码：
#coding:utf-8
conn=dict(host='127.0.0.1',user='root',passwd='server',db='db',charset='utf8')
见重构后的MySQLHelper类的源码：
#coding:utf-8
import MySQLdb
import config
class MySQLHelper(object):
def __init__(self):
self.__conn=config.conn
@property
def selectMySQL(self):
try:
conn=MySQLdb.connect(**self.__conn)
cur=conn.cursor()
except Exception,e:
print u'操作mysql数据库失败'
else:
cur.execute('select * from user;')
data=cur.fetchall()
for d in data:
print d
finally:
cur.close()
conn.close()
@property
def insertMySQL(self):
try:
conn=MySQLdb.connect(**self.__conn)
cur=conn.cursor()
sql="insert into user(id,name,sex,email) VALUES (%s,%s,%s,%s)"
params=(1,'selenium2','boy','USA@qq.com')
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'插入后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
@property
def updateMySQL(self):
try:
conn=MySQLdb.connect(**self.__conn)
cur=conn.cursor()
sql="update user set NAME=%s where id=%s"
params=('selenium webdriver',1)
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'修改后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
@property
def deleteMySQL(self):
try:
conn=MySQLdb.connect(**self.__conn)
cur=conn.cursor()
sql="delete from USER WHERE id=%s"
params=(1)
cur.execute(sql,params)
conn.commit()
except Exception,e:
print u'操作mysql数据库失败'
else:
print u'删除后表的数据为:'
self.selectMySQL
finally:
cur.close()
conn.close()
注解：把连接mysql的部分写在了config.py模块中，在该模块中，conn=dict(host='127.0.0.1',user='root',passwd='server',db='db',charset='utf8'),conn是一个字典，在类MySQKHelper中，初始化self.__conn=config.conn为私有字段，直接调用，这样连接mysql 的方法就直接即可。