首页 > 编程 > Python > 正文

Python3实现的Mysql数据库操作封装类

2020-02-15 21:40:13
字体:
来源:转载
供稿:网友

本文实例讲述了Python3实现的Mysql数据库操作封装类。分享给大家供大家参考,具体如下:

#encoding:utf-8#name:mod_db.py'''''使用方法:1.在主程序中先实例化DB Mysql数据库操作类。   2.使用方法:db=database() db.fetch_all("sql")'''import MySQLdbimport MySQLdb.cursorsimport mod_configimport mod_loggerDB = "database"LOGPATH = mod_config.getConfig('path', 'logpath') + 'database.log'DBNAME = mod_config.getConfig(DB, 'dbname')DBHOST = mod_config.getConfig(DB, 'dbhost')DBUSER = mod_config.getConfig(DB, 'dbuser')DBPWD = mod_config.getConfig(DB, 'dbpassword')DBCHARSET = mod_config.getConfig(DB, 'dbcharset')DBPORT = mod_config.getConfig(DB, "dbport")logger = mod_logger.logger(LOGPATH)#数据库操作类class database:#注,python的self等于其它语言的this  def __init__(self, dbname=None, dbhost=None):    self._logger = logger    #这里的None相当于其它语言的NULL    if dbname is None:      self._dbname = DBNAME    else:      self._dbname = dbname    if dbhost is None:      self._dbhost = DBHOST    else:      self._dbhost = dbhost    self._dbuser = DBUSER    self._dbpassword = DBPWD    self._dbcharset = DBCHARSET    self._dbport = int(DBPORT)    self._conn = self.connectMySQL()    if(self._conn):      self._cursor = self._conn.cursor()  #数据库连接  def connectMySQL(self):    conn = False    try:      conn = MySQLdb.connect(host=self._dbhost,          user=self._dbuser,          passwd=self._dbpassword,          db=self._dbname,          port=self._dbport,          cursorclass=MySQLdb.cursors.DictCursor,          charset=self._dbcharset,          )    except Exception,data:      self._logger.error("connect database failed, %s" % data)      conn = False    return conn  #获取查询结果集  def fetch_all(self, sql):    res = ''    if(self._conn):      try:        self._cursor.execute(sql)        res = self._cursor.fetchall()      except Exception, data:        res = False        self._logger.warn("query database exception, %s" % data)    return res  def update(self, sql):    flag = False    if(self._conn):      try:        self._cursor.execute(sql)        self._conn.commit()        flag = True      except Exception, data:        flag = False        self._logger.warn("update database exception, %s" % data)    return flag  #关闭数据库连接  def close(self):    if(self._conn):      try:        if(type(self._cursor)=='object'):          self._cursor.close()        if(type(self._conn)=='object'):          self._conn.close()      except Exception, data:        self._logger.warn("close database exception, %s,%s,%s" % (data, type(self._cursor), type(self._conn)))

这段代码需要注意几个地方:

1.数据库配置文件变量放在类外面,只在模块被载入的时候读一次,在使用数据库类时,无需再读取。这样有一个好处就是减少了IO的操作。如果是脚本程序需要不停的扫数据库,放在类初始化的时候读数据库配置,那么很有可能造成IO错误,或者读到的section为空,从而导致程序down掉。不好的地方就是如果修改了config,无法及时更新。不过一般程序部署后,不会随意修改config文件,或者在修改config文件后,重新启动程序。不过从程序的健壮性考虑,可以在可能会抛出异常的地方try,catch一下,更好的办法是,在mod_config.py模块中判断一下该section和key是否存在,然后再读取。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表