db.py 1.21 KB
Newer Older
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 40 41
# -*-coding:utf-8 -*- #
# ---------------------------------------------------------------------------
# ProjectName:   test_56
# FileName:      db.py
# Author:       xxxxxxx
# Datetime:     2023/8/7 15:26
# Description:
# 
# ---------------------------------------------------------------------------
import pymysql
from InterfaceAutoTest.common.read_ini import ReadIni


class DB:
    def __init__(self):
        """链接数据库"""
        read_ini = ReadIni()
        self.conn = pymysql.connect(
            host=read_ini.get_sql_connect_message("host"),
            port=int(read_ini.get_sql_connect_message("port")),
            user=read_ini.get_sql_connect_message("user"),
            password=read_ini.get_sql_connect_message("pwd"),
            database=read_ini.get_sql_connect_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]