python使用pymysql访问数据库

约定不等于承诺〃 2024-04-17 06:28 75阅读 0赞

*  python使用pymysql访问数据库

import pymysql
    
    class SQL:
        # 记录访问数据库的账户信息
        __account = { 
            'user': '',
            'password': ''
        }
        
        def __init__(self, _account: dict):
            if 'user' in _account and 'password' in _account:
                self.__account['user'] = _account['user']
                self.__account['password'] = _account['password']
    
        # 连接数据库
        def __get_connect(self):
            # 返回连接对象
            return pymysql.connect(
                host='************.**.tencentcdb.com',  # 此处为腾讯云数据库地址
                										# 也可填写数据库IP
                port=10069,
                user=self.__account['user'],
                password=self.__account['password'],
                database='db',							# 要访问的数据库
                charset='utf8'
            )
    
        # 查询数据
        def query_data(self, sql):
            connect = self.__get_connect()
            # try块，finally保证connect可以正确关闭
            try:
                # 连接上可重用的执行SQL的对象
                cursor = connect.cursor(pymysql.cursors.DictCursor)  
                				# DictCursor: 返回的每条数据是字典格式，存放在一个列表中
                cursor.execute(sql)
                return cursor.fetchall()
            finally:
                connect.close()
    
        # 更新数据，可以执行数据的增、删、改
        def update_data(self, sql):
            connect = self.__get_connect()
            try:
                cursor = connect.cursor()
                cursor.execute(sql)
                connect.commit()  # 更新数据库需要commit
            finally:
                connect.close()
                
    # 测试
    if __name__ == '__main__':
        account = { 
            'user': 'root',
            'password': 'root123456'
        }
        mysql = SQL(account)
        try:
            sql = "select * from table1 where att1 = '{ATT1}'".format(ATT1="att1")
        	result = mysql.query_data(query)
        except Exception as e:
            print(e.args[0])		# 响应码
            print(e.args[1])		# message

*  python使用pymysql包访问数据库时，当前保留的与数据库的连接 是 程序最后一次创建的连接。当访问数据库的操作涉及账户切换时，每次以该账号执行所有访问操作，都需要重新登陆账户。避免交替更换账户出现权限不一致问题。（**需要用哪个账户，就实时建立哪个连接**）

import func.sql as sql	# sql的SQL类中封装了访问数据库的操作
    """ root有访问Log表的权限 user无访问Log表的权限 """
    
    # 错误操作，涉及账户切换时，不能事先创建两个数据库连接对象
    account1 = { 
        'user': 'root',
        'password': 'root123456'
    }
    mysql1 = sql.SQL(account1)
    
    account2 = { 
        'user': 'user1',
        'password': 'user1123456'
    }
    mysql2 = sql.SQL(account2)
    
    try:
        print(mysql1.query_data("select * from Log"))
    except Exception as e:
        print(e)
    	# 这时会print错误信息：(1142, "SELECT command denied to user 'user1'@'223.104.1.250' for table 'Log'")
    
    
    # 正确操作，需要用哪个账户，就实时建立哪个连接
    account1 = { 
        'user': 'root',
        'password': 'root123456'
    }
    account2 = { 
        'user': 'user1',
        'password': 'user1123456'
    }
    
    try:
        # 需要用哪个账户，就实时建立哪个连接
        mysql = sql.SQL(account1)
        print(mysql.query_data("select * from Log"))
    except Exception as e:
        print(e)