用python 將 pymysql操作封裝成類

覺得代碼啰嗦的可以把logging日志刪掉,但是工程中時刻要記得寫日志

import pymysql
import logging
import sys

# 加入日志
#獲取logger實例
logger = logging.getLogger("baseSpider")
# 指定輸出格式
formatter = logging.Formatter('%(asctime)s\
              %(levelname)-8s:%(message)s')
# 文件日志
file_handler = logging.FileHandler("baseSpider.log")
file_handler.setFormatter(formatter)
# 控制台日志
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(formatter)

# 為logge添加具體的日志處理器
logger.addHandler(file_handler)
logger.addHandler(console_handler)

logger.setLevel(logging.INFO)

class DBHelper:
    # 構造函數
    def __init__(self, host='127.0.0.1', user='root',
        pwd='123456', db='testdb'):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.db = db
        self.conn = None
        self.cur = None


    # 連接數據庫
    def connectDatabase(self):
        try:
            self.conn = pymysql.connect(self.host, self.user,
                           self.pwd, self.db,charset='utf8')
        except:
            logger.error("connectDatabase failed")
            return False
        self.cur = self.conn.cursor()
        return True


    # 關閉數據庫
    def close(self):
        # 如果數據打開，則關閉；否則沒有操作
        if self.conn and self.cur:
            self.cur.close()
            self.conn.close()
        return True
    
    # 執行數據庫的sq語句,主要用來做插入操作
    def execute(self, sql, params=None):
        # 連接數據庫
        self.connectDatabase()
        try:
            if self.conn and self.cur:
                # 正常邏輯，執行sql，提交操作
                self.cur.execute(sql, params)
                self.conn.commit()
        except:
            logger.error("execute failed: "+sql)
            logger.error("params: "+params)
            self.close()
            return False
        return True

    # 用來查詢表數據
    def fetchall(self, sql, params=None):
        self.execute(sql,params)
        return self.cur.fetchall()

if __name__ == '__main__':
    dbhelper = DBHelper()
    # 創建數據庫的表
    sql = "create table maoyan('id'varchar(8),\
            'title'varchar(50),\
            'star'varchar(200), \
            'time'varchar(100),primary key('id'));"
    result = dbhelper.execute(sql,None)
    if result:
        logger.info("maoyan　table創建成功")
    else:
        logger.error("maoyan　table創建失敗")