python项目交付神器：代码混淆、文件加密与配置文件

前言

整体逻辑顺序：制作配置文件-->代码混淆-->文件加密

一、配置文件

1.1 制作.ini的配置文件

python 的配置文件格式为.ini，代码格式为

;cfg.ini

[section0]
ip = 192.168.xx.xx
[section1]
port = 5672

根据不同功能分section来分配要配置的变量。

1.2 读取配置文件

代码如下

#test_ini.py
import configparser
#读取ini方法
def read_ini(inikey,inivaluse):
        config = configparser.ConfigParser()
        config.read("cfg.ini")
        convaluse=config.get(inikey,inivaluse)
        return convaluse
if __name__=='__main__':
        ip = read_ini("section0",'port')
        print(ip)

执行结果：

192.168.xx.xx

读取文件完成。

二、python代码混淆

2.1简介

条件：首先项目的中所有的业务逻辑封装成类或者函数，并用单个文件表示，即main函数与后端逻辑用文件分开。
方法：将类文件-->编译成.so文件

2.2举例

即将以上的test_ini.py打包成test_ini.so文件，编程如下

from distutils.core import setup
from Cython.Build import cythonize
setup(
    ext_modules = cythonize("test_ini.py")
)

完成后，在Terminal上执行

 python setup.py build_ext --inplace

即可生成.so文件，生成的文件带有长后缀，建议不要改动主体文件名，可以删除副后缀，例如此处将.so的文件名改为test_ini.so。
在main函数文件中执行打印配置文件，测试.so文件是否可用。此处若担心test_ini.py的影响，可以先转移或者删除test_ini.py文件。

#main.py
from test_ini import read_ini

ip = read_ini("section0",'port')
print(ip)

2.3 附件

例子中是转单个文件为.so文件，此处附上将文件夹中所有的代码转换为.so文件代码

#-*- coding:utf-8 -*-_
import os
import re 
from distutils.core import Extension, setup
 
from Cython.Build import cythonize
from Cython.Compiler import Options
 
 
# __file__ 含有魔术变量的应当排除，Cython虽有个编译参数，但只能设置静态。
exclude_so = ['__init__.py', 'run.py']
sources = 'test'   #文件夹名
 
 
extensions = []
remove_files = []
for source,dirnames,files in os.walk(sources):
    for dirpath, foldernames, filenames in os.walk(source):
        if 'test' in dirpath:
            break;
        for filename in filter(lambda x: re.match(r'.*[.]py$', x), filenames):
            file_path = os.path.join(dirpath, filename)
            if filename not in exclude_so:
                extensions.append(
                        Extension(file_path[:-3].replace('/', '.'), [file_path], extra_compile_args = ["-Os", "-g0"],
                                  extra_link_args = ["-Wl,--strip-all"]))
                remove_files.append(file_path[:-3]+'.py')
                remove_files.append(file_path[:-3]+'.pyc')

Options.docstrings = False
compiler_directives = {'optimize.unpack_method_calls': False, 'always_allow_keywords': True}
setup(  
        # cythonize的exclude全路径匹配，不灵活，不如在上一步排除。
        ext_modules = cythonize(extensions, exclude = None, nthreads = 20, quiet = True, build_dir = './build',
                                language_level = 2, compiler_directives = compiler_directives))

# 删除py和pyc文件，酌情注释
for remove_file in remove_files:

    if os.path.exists(remove_file):
        os.remove(remove_file)

以上为混淆+配置文件的方法，以上已经完成python的交付，既保护了代码又可让客户通过更改配置文件来配置需要的接口。为什么还需要文件加密呢？这就是为了保护深度学习项目中的模型文件，当然本文的方法适用于保护所有的文件。下面开始说一下文件加密。

三、文件加密

3.1 思路

原本想使用一些对称加密的手段，考虑到深度学习模型的加载需要读取到模型，若使用对称加密，在文件加载时解密，然后执行完后又要加密。步骤繁琐，逻辑也补通畅。交付任务临近，想到一个使用加密压缩包的方法。然后加载模型文件时，通过python的zipfile模块读取加密压缩包即可完成模型加载。步骤如下，

压缩包加密-->zipfile读取压缩文件-->运行文件

3.2. 方法

压缩包加密，windows下直接右键压缩文件添加密码选项，linux加密文件路径下执行命令：

zip -rP passwork filename.zip filename

其中，passwork就是设置的密码，根据文件重要程度，酌情修改。
读取压缩文件python代码如下

import numpy as np
import zipfile

file_name  = zipfile.ZipFile("a.zip")
print(file_name)
file = file_name.open('a.npy',mode='r',pwd='123333'.encode('utf-8'))
a = np.load(file)
print(a)

例子中，在a.zip压缩文件中包含一个a.npy文件，压缩密码为123333，然后加载文件并打印，与原来a.npy的内容一致。请放心使用该方法。

以上，为python项目交付的一些方法，欢迎提出问题，以及提出更好的交付方法。