python连接加密数据库的封装代码

连接加密数据库的Python封装代码与连接非加密数据库的代码类似，只需要在连接数据库时传入相应的加密参数即可。下面是一个使用PyMySQL连接加密的MySQL数据库的封装代码示例：

import os
import pymysql

class EncryptedDatabase:
    def __init__(self, host, user, password, database, port=3306, charset='utf8mb4', ssl_ca=None, ssl_cert=None, ssl_key=None):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.port = port
        self.charset = charset
        self.ssl_ca = ssl_ca
        self.ssl_cert = ssl_cert
        self.ssl_key = ssl_key
        self.conn = None
        self.cursor = None

    def connect(self):
        self.conn = pymysql.connect(host=self.host, user=self.user, password=self.password, database=self.database, port=self.port, charset=self.charset, ssl={'ca': self.ssl_ca, 'cert': self.ssl_cert, 'key': self.ssl_key})
        self.cursor = self.conn.cursor()

    def close(self):
        if self.cursor:
            self.cursor.close()
            self.cursor = None
        if self.conn:
            self.conn.close()
            self.conn = None

    def execute(self, sql, args=None):
        try:
            if not self.conn or not self.cursor:
                self.connect()
            self.cursor.execute(sql, args)
            self.conn.commit()
            return self.cursor.fetchall()
        except Exception as e:
            self.conn.rollback()
            raise e
        finally:
            self.close()

这个封装代码使用了EncryptedDatabase类来连接加密的MySQL数据库。在类的构造函数中，我们传入了数据库连接的相关参数，包括主机名、用户名、密码、数据库名、端口号、字符集以及ssl_ca、ssl_cert、ssl_key等加密参数。在connect()方法中，我们使用了ssl参数来连接加密的数据库。在execute()方法中，我们先检查是否已经连接数据库，如果没有则先进行连接，然后执行SQL语句，并提交事务，最后关闭数据库连接。如果执行SQL语句时出现异常，则回滚事务，然后抛出异常。