# -*-coding:utf-8 -*- #
# ---------------------------------------------------------------------------
# ProjectName:   test64
# FileName:      db.py
# Author:       lao_zhao
# Datetime:     2024/12/26 14:00
# Description:
# 
# ---------------------------------------------------------------------------
import pymysql
from APIAutoTest_v3 import log_decorator
from APIAutoTest_v3.common.read_basic_ini import ReadBasicIni


class DB:
    @log_decorator
    def __init__(self):
        """连接数据库,获取连接对象和游标对象"""
        ini = ReadBasicIni()
        self.conn = pymysql.connect(
            host=ini.get_database_conn_msg("host"),
            port=int(ini.get_database_conn_msg("port")),
            user=ini.get_database_conn_msg("user"),
            password=ini.get_database_conn_msg("pwd"),
            database=ini.get_database_conn_msg("database"),
            charset="utf8"
        )
        self.cursor = self.conn.cursor()


    @log_decorator
    def close(self):
        """关闭游标对象和连接对象"""
        self.cursor.close()
        self.conn.close()

    @log_decorator
    def delete(self,sql_data):
        """执行删除的sql语句"""
        self.cursor.execute(sql_data)
        self.conn.commit()

    @log_decorator
    def select(self, sql_data):
        """执行查询的sql语句"""
        self.cursor.execute(sql_data)
        select_res = self.cursor.fetchall()
        if select_res:
            return select_res[0][0]