mysql獲取某個表的所有屬性名及其數據

MYSQL類實現從數據庫相應的表中獲取所有屬性及其數據，數據為元組類型。返回結果存放在字典中

import pymysql


class MYSQL:
    def __init__(self):
        pass

    def __del__(self):
        self._cursor.close()
        self._connect.close()

    def connectDB(self):
        """
        連接數據庫
        :return:
        """
        try:
            self._connect = pymysql.Connect(
                host='localhost',
                port=3306,
                user='root',
                passwd='123456',
                db='test',
                charset='utf8'
            )

            return 1
        except:
            return 0

    def readObject(self, target):
        """
        讀取評價對象
        :return: list對象:所有的評價對象及其數據,數據的行數
        """
        self._cursor=self._connect.cursor()
        result = {}  # {字段：該字段的所有值列表（包含字段）}
        name = []    # target表的所有字段的列表
        data = ()
        sql = "select COLUMN_NAME from information_schema.COLUMNS where table_name = '%s'"
        name_sql = "select %s from %s"
        self._cursor.execute(sql % target)
        results = self._cursor.fetchall()
        for row in results:
            name.append(row[0])

        # print(name)

        for i in name:
            self._cursor.execute(name_sql % (i, target))
            data = self._cursor.fetchall()
            # Python 字典 setdefault() 函數
            # 如果字典中包含有給定鍵，則返回該鍵對應的值，否則返回為該鍵設置的值
            result.setdefault(i, data)

        return result, len(data)


if __name__ == '__main__':

    mysql = MYSQL()
    flag = mysql.connectDB()
    if flag == 0:
        print('數據庫連接失敗')
    else:
        print('數據庫連接成功')
        data, row_count = mysql.readObject('employee')
        print(data)
        print(row_count)

 

輸出結果：

數據庫連接成功
{'FIRST_NAME': (('Mac',), ('Marry',), ('Bob',)), 'LAST_NAME': (('Mohan',), ('Mohan',), ('Mohan',)), 'AGE': ((20,), (32,), (21,)), 'SEX': (('M',), ('M',), ('F',)), 'INCOME': ((2000.0,), (3000.0,), (4000.0,))}
3