python项目(.pyd或者.so)代码加密
准备工作
Linux安装：
用到的库
sudo apt-get install python-dev gcc
#windows只需安装下面这两个就可以
pip install pycrypto Cython
在项目文件夹下新建两个文件setup.py, rename.py
加密脚本setup.py的代码如下
# coding:utf-8
import sys, os, shutil, time
from distutils.core import setup
from Cython.Build import cythonize
starttime = time.time()
currdir = os.path.abspath(‘.’)
parentpath = sys.argv[1] if len(sys.argv) > 1 else “”
setupfile = os.path.join(os.path.abspath(‘.’), __file__)
build_dir = “build” # 项目加密后位置
build_tmp_dir = build_dir + “/temp”
def getpy(basepath=os.path.abspath(‘.’), parentpath=”, name=”, excepts=(), copyOther=False, delC=False):
“””
获取py文件的路径
:param basepath: 根路径
:param parentpath: 父路径
:param name: 文件/夹
:param excepts: 排除文件(当前文件不是目标文件)
:param copy: 是否copy其他文件
:return: py文件的迭代器
“””
# 返回所有文件夹绝对路径
fullpath = os.path.join(basepath, parentpath, name)
# 返回指定的文件夹包含的文件或文件夹的名字的列表
for fname in os.listdir(fullpath):
ffile = os.path.join(fullpath, fname)
# print(‘fname’,fname)
# print(“ffile”, ffile)
# print basepath, parentpath, name,file
# 是文件夹 且不以.开头 不是 build ，不是迁移文件
if os.path.isdir(ffile) and fname != build_dir and not fname.startswith(‘.’) and fname != “migrations”:
# 循环遍历文件夹
for f in getpy(basepath, os.path.join(parentpath, name), fname, excepts, copyOther, delC):
yield f
elif os.path.isfile(ffile):
# 筛选出 .c 文件
ext = os.path.splitext(fname)[1]
if ext == “.c”:
# 显示文件 “ffile” 信息,st_mtime: 最后一次修改的时间。
if delC and os.stat(ffile).st_mtime > starttime:
# 删除 .c 文件
os.remove(ffile)
# 文件不是排除文件 且不是’.pyc’ ‘.pyx’文件
elif ffile not in excepts and os.path.splitext(fname)[1] not in (‘.pyc’, ‘.pyx’):
# manage.py文件不编译
if os.path.splitext(fname)[1] in (‘.py’, ‘.pyx’) and not fname.startswith(
‘__’) and fname != “manage.py” and fname != “test.py”:
# 返回要加密的文件(加到module_list中成为一个列表)
yield os.path.join(parentpath, name, fname)
elif copyOther:
dstdir = os.path.join(basepath, build_dir, parentpath, name)
# 判断文件夹是否存在，不存在则创建
if not os.path.isdir(dstdir):
os.makedirs(dstdir)
print(‘dstdir的是’,dstdir)
# 复制文件到新文件夹下
print(‘ffile的是’,ffile)
shutil.copyfile(ffile, os.path.join(dstdir, fname))
else:
pass
# 获取py列表
module_list = list(getpy(basepath=currdir, parentpath=parentpath, excepts=(setupfile)))
print(module_list)
try:
setup(ext_modules=cythonize(module_list), script_args=[“build_ext”, “-b”, build_dir, “-t”, build_tmp_dir])
except Exception as e:
print(e)
else:
module_list = list(getpy(basepath=currdir, parentpath=parentpath, excepts=(setupfile), copyOther=True))
module_list = list(getpy(basepath=currdir, parentpath=parentpath, excepts=(setupfile), delC=True))
# shutil.rmtree() 表示递归删除文件夹下的所有子文件夹和子文件
# 删除build_tmp_dir临时文件夹
if os.path.exists(build_tmp_dir):
shutil.rmtree(build_tmp_dir)
print(“complate! time:”, time.time() – starttime, ‘s’)
方法：将需要加密的代码放到列表里，然后在终端执行 python setup.py
1、如果报错不能找到vcvasall.bat说明没有按照vc++安装之后再执行!
2、如果安装了vc++还是报错的话，这时候需要配置一下路径请看这里
# coding:utf-8
import os
from distutils.core import setup
from Cython.Build import cythonize
import shutil
# filter=[“.py”,”.so”,’.c’] linux的过滤条件
filter=[“.py”,”.pyd”,’.c’] #设置过滤后的文件类型 当然可以设置多个类型
def all_path(dirname):
result = [] # 所有的文件
for maindir, subdir, file_name_list in os.walk(dirname):
for filename in file_name_list:
apath = os.path.join(maindir, filename)#合并成一个完整路径
ext = os.path.splitext(apath)[1] # 获取文件后缀 [0]获取的是除了文件名以外的内容
# 筛选出.py 和 不是__init__的文件
if ext in filter and os.path.splitext(filename)[0] not in(‘__init__’,’manage’,’test’):
result.append(apath)
else:
pass
return result
def rename(list):
for i in list:
if i.__contains__(“.pyd”):
re_name = i.split(“.”)[0] + ‘.pyd’ #(linux的是.so文件)
print(i.split(‘.’))
os.rename(i, re_name)
elif i.__contains__(“.c”):
os.remove(i)
if __name__ == ‘__main__’:
path = os.getcwd()
list = all_path(path)
print(list)
print(len(list))
rename(list)
再执行一遍rename.py就得到修改后的pyd文件了
至此代码加密完成 加密完的项目在build文件夹中
再次在终端执行python manage.py runserver 运行项目成功
说明加密成功