db.py 1.28 KB
Newer Older
rio committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
# -*-coding:utf-8 -*- #
# ---------------------------------------------------------------------------
# ProjectName:   pythonProject
# FileName:      db.py
# Author:       xxxxxxx
# Datetime:     2023/8/7 18:36
# Description:
# 
# ---------------------------------------------------------------------------
import pymysql

from APIAutoTest_004.common.read_ini import ReadIni


class DB:
    def __init__(self):
        read_ini = ReadIni()
        self.conn = pymysql.connect(host=read_ini.get_sql_message("host"),
                                    port=int(read_ini.get_sql_message("port")),
                                    user=read_ini.get_sql_message("user"),
                                    password=read_ini.get_sql_message("pwd"),
                                    database=read_ini.get_sql_message("database"),
                                    charset="utf8"
                                    )
        self.cursor = self.conn.cursor()

    def close(self):
        self.cursor.close()
        self.conn.close()

    def delete(self, sql_sentence):
        self.cursor.execute(sql_sentence)
        self.conn.commit()

    def select(self, sql_sentence):
        self.cursor.execute(sql_sentence)
        result = self.cursor.fetchall()
        if result:
            return result[0][0]