【mysql数据库之python链接mysql_封装版】

from pymysql import connect
from server.config import *


class DB(object):
    '''数据库'''

    def __init__(self):
        '''初始化数据库'''
        # 建立连接
        self.conn = connect(
            host=DB_HOST,
            port=DB_PORT,
            database=DB_NAME,
            user=DB_USER,
            password=DB_PASS,
            charset='utf8')
        # 获取游标
        self.cursor = self.conn.cursor()

    def close(self):
        '''释放数据库资源'''
        self.cursor.close()
        self.conn.close()

    def get_one(self, sql):
        '''使用sql语句查询一条数据'''
        # 1.执行sql
        self.cursor.execute(sql)
        # 2.获取查询结果
        query_result = self.cursor.fetchone()
        # 3.判断是否有结果
        if not query_result:
            return None
        # 4.获取字段
        fileds = [filed[0] for filed in self.cursor.description]
        # 5.使用字段和结果数据合成字典，供返回
        return_data = {}
        for filed,value in zip(fileds, query_result):
            return_data[filed] = value
        return return_data
    
    def get_all(self, sql):
        '''使用sql语句查询多条数据'''
        pass


if __name__ == '__main__':
    db = DB()
    data = db.get_one('select * from users where user_name = "user2"')
    print(data)
    db.close()