私有化部署时实现类似软件注册码的机制

import rsa
from .utils import get_sys_key


class ClientSecure:
    """客户端需要做的就是加密本地信息，解密服务端信息"""
    # 系统ID
    sys_id = ''
    # 客户端秘钥文件
    client_privkey_f = ''
    # 服务端加解密公钥
    server_pubkey_f = ''

    def __init__(self, sys_id, client_privkey_path, server_pubkey_path):
        self.client_privkey_f = client_privkey_path
        self.server_pubkey_f = server_pubkey_path

    def decode(self, server_reg_msg):
        """使用客户端私钥解密服务器的信息
        每次本地服务启动时，都进行一次判断
        :param server_reg_msg 文件内容, open(finename, 'rb').read()
        """
        with open(self.client_privkey_f, 'rb') as r:
            p = r.read()
            client_privkey = rsa.PrivateKey.load_pkcs1(p)

        server_msg = rsa.decrypt(server_reg_msg, client_privkey)
        return str(server_msg, encoding='utf8')

    def check(self, server_msg):
        """判断解密之后的服务器端信息是否有效"""
        # TODO 可以在这里进行有效期校验
        # TODO 测试信息是否正确
        sys_msg = get_sys_key(self.sys_id)
        if server_msg != sys_msg:
            return False
        return True

    def encode(self):
        """使用服务端公钥加密本地信息
        如果是离线模型，通常只需要在第一次启动前生成
        """
        with open(self.server_pubkey_f, 'rb') as r:
            p = r.read()
            server_pubkey = rsa.PublicKey.load_pkcs1(p)

        # print('加密')
        msg = get_sys_key(self.sys_id)
        msg = bytes(msg, encoding="utf8")
        crypto = rsa.encrypt(msg, server_pubkey)
        return crypto