Python数据库工具类MySQLdb使用

MySQLdb模块用于连接mysql数据库。

基本操作

#  -*- coding: utf-8 -*-      
# mysqldb      
import time, MySQLdb      
     
# 连接      
conn=MySQLdb.connect(host= " localhost ",user= " root ",passwd= " root ",db= " test ",charset= " utf8 ")    
cursor = conn.cursor()      
 
# 删除表  
sql =  " drop table if exists user "  
cursor.execute(sql)  
 
# 创建  
sql =  " create table if not exists user(name varchar(128) primary key, created int(10)) "  
cursor.execute(sql)  
 
# 写入      
sql =  " insert into user(name,created) values(%s,%s) "    
param = ( " aaa ",int(time.time()))      
n = cursor.execute(sql,param)      
print  ' insert ',n      
     
# 写入多行      
sql =  " insert into user(name,created) values(%s,%s) "    
param = (( " bbb ",int(time.time())), ( " ccc ",33), ( " ddd ",44) )  
n = cursor.executemany(sql,param)      
print  ' insertmany ',n      
 
# 更新      
sql =  " update user set name=%s where name='aaa' "    
param = ( " zzz ")      
n = cursor.execute(sql,param)      
print  ' update ',n      
     
# 查询      
n = cursor.execute( " select * from user ")      
for row  in cursor.fetchall():      
     print row  
     for r  in row:      
         print r      
     
# 删除      
sql =  " delete from user where name=%s "    
param =( " bbb ")      
n = cursor.execute(sql,param)      
print  ' delete ',n      
 
# 查询      
n = cursor.execute( " select * from user ")      
print cursor.fetchall()      
 
cursor.close()      
     
# 提交      
conn.commit()  
# 关闭 
conn.close()
     

 

封装类操作

此处ConfigUtils工具类为Python配置工具类ConfigParser使用提供。在使用数据库连接时，建议每次调用利用try finally机制，做好资源回收。在对于异常处理时，其实不建议像如此处理。

异常处理，不要对大段代码捕获exception，而是要处理具体的异常，以便提高程序健壮性。

  class  DButils(object):
     def  __init__(self,filename,section):
        super(DButils, self). __init__()
         # read config
        cfg = ConfigUtils(filename).config
        self.cfg = cfg
        self.section = section
         # init mysql connection
        self.conn= MySQLdb.connect(
            host=cfg.get(section, ' host '),
            port = cfg.getint(section, ' port '),
            user=cfg.get(section, ' user '),
            passwd=cfg.get(section, ' passwd '),
            db=cfg.get(section, ' db '),
            connect_timeout=cfg.getint(section, ' connect_timeout ')
        )
        self.cur = self.conn.cursor()    
     
     def fetchmany(self,sql):
        sql = sql.replace( ' {$db} ',self.cfg.get(self.section, ' db '))
         try:
             return self.cur.fetchmany(self.cur.execute(sql))
         except Exception, e:
             print traceback.print_exc()
             print sql
   
     def fetchone(self,sql):
        sql = sql.replace( ' {$db} ',self.cfg.get(self.section, ' db '))
         try:
            self.cur.execute(sql)
             return self.cur.fetchone()
         except Exception, e:
             print traceback.print_exc()
             print sql

     def create(self,sql):
         try:
            self.cur.execute(sql)
            self.conn.commit()
         except Exception, e:
             print traceback.print_exc()

     def is_table_exit(self,tableName):
        show_sql =  ' show tables; '
         try:
             return tableName  in self.cur.fetchmany(self.cur.execute(show_sql))
         except Exception,e:
             print traceback.print_exc()
     def close_db(self):
        self.cur.close()
        self.conn.close()

db    = DButils( ' ini.cfg ', ' src_db ')
try:
    db.fetchone( 'select * from table limit 1')
finally:
    db.close_db()

 

转载于:https://www.cnblogs.com/snifferhu/p/4369184.html