python - 接口自動化測試 - MysqlUtil - 數據庫操作封裝

 

 

# -*- coding:utf-8 -*-

'''
@project: ApiAutoTest
@author: Jimmy
@file: mysql_util.py
@ide: PyCharm Community Edition
@time: 2018-12-22 10:30
@blog: https://www.cnblogs.com/gotesting/

'''

'''
1. 連接數據庫
2. 查詢
3. 建立游標
4. 執行
'''
import pymysql
import traceback
from Common.read_config import ReadConfig


class MysqlUtil:

    # 讀取配置文件中的數據庫配置信息，連接數據庫
    def __init__(self):
        read_config = ReadConfig()
        host = read_config.get_config_str('mysql','host')
        port = read_config.get_config_int('mysql','port')
        user = read_config.get_config_str('mysql','user')
        password = read_config.get_config_str('mysql','pwd')
        try:
            self.mysql = pymysql.connect(host=host,port=port,user=user,password=password,cursorclass=pymysql.cursors.DictCursor)
        except Exception as e:
            print('mysql connect failed , please check')
            raise e


    # 查詢sql數據，並返回
    def fetch_one(self,sql):
        db = self.mysql
        cursor = db.cursor() # 建立游標
        try:
            cursor.execute(sql) # 執行sql語句
            result = cursor.fetchone() # 返回一條數據
        except:
            traceback.print_exc() # 輸出異常信息
            db.rollback()
        return result

    def fetch_all(self,sql):
        db = self.mysql
        cursor = db.cursor()
        try:
            cursor.execute(sql)
            result = cursor.fetchall()
        except:
            traceback.print_exc()
        return result