#mysql数据库的查询等

import pymysql

from xctest_tools.get_ReadTxt import *

class mysql:

    def __init__(self):

        # 读取配置文件进行连接

        self.db=ReadTxt().read_txt("./db_config")

        self.con=pymysql.connect(host=self.db[0],port=eval(self.db[1]),user=self.db[2],password=self.db[3],database=self.db[4],charset="utf8")

        self.cur=self.con.cursor()

    def select_all(self,sql):

        a=self.cur.execute(sql)

        self.con.commit()

        #查询到的所有信息，返回二维元组

        r1=self.cur.fetchall()

        return r1

    def select_one(self,sql):

        a=self.cur.execute(sql)

        self.con.commit()

        #查询到的第一条信息，返回一维元组

        r2 = self.cur.fetchone()

        return r2

    def select_any(self,sql,number):

        a=self.cur.execute(sql)

        self.con.commit()

        #一次指定取number条记录

        r3 = self.cur.fetchmany(number)

        return r3

    def select_number(self,sql):

        #查询到的条数

        a=self.cur.execute(sql)

        self.con.commit()

        return a

    def __del__(self):

        self.cur.close()

        self.con.close()