python操作數據庫---工具封裝

1. 基本操作（增刪改查）封裝

2. 基於python3操作

'''
Created on 2019年
數據庫工具類
@author: Root
'''
import pymysql
import configparser

# 讀取配置文件
cf = configparser.ConfigParser()
cf.read('db.config')
host = cf.get('db', 'host')
port = cf.get('db', 'port')
port = int(port)
user = cf.get('db', 'user')
pwd = cf.get('db', 'passwd')
db = cf.get('db', 'db')
charset = cf.get('db', 'charset')

class DBUtils:
     
    # 獲取數據庫連接
    def getConn(self):
        conn = pymysql.connect(host=host,port=port,user=user,passwd=pwd,db=db,charset=charset)
        return conn
    
    # 查詢所有數據   
    def get_allDate(self,sql):
        # 取得連接
        conn = DBUtils().getConn()
        # 創建游標
        cursor = conn.cursor()
        # 執行SQL
        cursor.execute(sql)
        # 處理結果集
        result = cursor.fetchall()
        # 返回處理結果
        return result
        # 關閉連接
        conn.close()
    
    # 條件查詢
    def get_dateByCount(self,sql,count):
        # 取得連接
        conn = DBUtils().getConn()
        try:
            # 創建游標
            cursor = conn.cursor()
            # 執行SQL
            cursor.execute(sql,count)
            # 處理結果集
            result = cursor.fetchall()
            # 返回處理結果
            return result
        except Exception as e:
            print ('查詢失敗',e)
        # 關閉連接
        conn.close()
    

    # 操作數據
    def operate_dateByCount(self,sql,count):
        # 取得連接
        conn = DBUtils().getConn()
        try:
            # 創建游標
            cursor = conn.cursor()
            # 執行SQL
            if type(count) is tuple:
                cursor.execute(sql,count)
            elif type(count) is list:
                cursor.executemany(sql,count)
            else:
                print ('參數count類型不正確')
            # 執行SQL成功，則提交事務
            conn.commit()
            print (count,'操作成功')
            return 0
        except Exception as e:
            # 執行SQL失敗，則回滾事務
            conn.rollback()
            print ('操作失敗',e)
        # 關閉連接
        conn.close()

配置文件寫法

db.config或者db.ini

[db]
host=localhost
port=3306
user=root
passwd=root
db=xxx
charset=utf8