python接口自動化第五篇--封裝python連接數據庫（mysql）框架

#數據庫操作封裝
import pymysql

class DBHandler:
    def __init__(self,host=None,port=8111,user='root',password='',charset='utf8',database=None,**kw):
        '''初始化參數'''
        self.conn = pymysql.connect(host = host,#域名
                               port = port,#端口號
                               user = user,#數據庫登陸的用戶名
                               password = password,#數據庫登陸的用戶名
                               charset = charset,#不能是utf-8
                               database = database,#你要連接的那個數據庫
                               **kw,
        )
        self.cursor = self.conn.cursor()
    def query_one(self,sql,args=None):
        '''查詢一條數據庫語句'''
        self.cursor.execute(sql,args)
        return self.cursor.fetchone()

    def query_all(self,sql,args=None):
        '''查詢所有數據庫語句'''
        self.cursor.execute(sql,args)
        return self.cursor.fetchall()

#如果我們不知道查詢一個還是多個，就可以把這兩個合併起來
    def query(self,sql,args= None,one=True):
        '''主體查詢函數'''
        if one:
            return self.query_one()
        return self.query_all()

    def close(self):
        '''關閉'''
        self.cursor.close()
        self.conn.close()