linux python项目加密混淆

# author : @sj
# -*- coding:utf-8 -*-
# time : 2023/4/25 16:37
import os
import time
import shutil
from Cython.Distutils import build_ext
from distutils.core import Extension, setup
from Cython.Build import cythonize
from Cython.Compiler import Options

start_time = time.time()
curr_dir = os.path.abspath('.')
# parent_path = sys.argv[1] if len(sys.argv) > 1 else ""
parent_path = 'project_name'
setup_file = os.path.join(os.path.abspath('.'), __file__)

build_dir = "{}_build".format(parent_path)
build_tmp_dir = build_dir + "/{}_temp".format(parent_path)

# 环境 ubuntu
# 依赖
# sudo apt install python3-dev gcc
# pip3 install cython

# Python文件加密 parent_path 项目的目录名字，例如 parent_path = project_name
# setup.py放下根目录下
# 执行 python3 setup.py build_ext
# project_name_build 覆盖掉project_name 执行命令 mv project_name_build ../project_name


def getpy(basepath=os.path.abspath('.'), parentpath='', name='', excepts=(), copyOther=True,delC=False):
    """
    获取py文件的路径
    :param basepath: 根路径
    :param parentpath: 父路径
    :param name: 文件/夹
    :param excepts: 排除文件
    :param copy: 是否copy其他文件
    :return: py文件的迭代器
    """
    fullpath = os.path.join(basepath, parentpath, name)
    for f_name in os.listdir(fullpath):
        ffile = os.path.join(fullpath, f_name)

        if os.path.isdir(ffile) and f_name != build_dir and not f_name.startswith('.'):
            for f in getpy(basepath, os.path.join(parentpath, name), f_name, excepts, copyOther, delC):
                # f = basepath + '/' + f
                yield f.replace('\\', '/')

        elif os.path.isfile(ffile):
            ext = os.path.splitext(f_name)[1]
            if ext == ".c":
                if delC and os.stat(ffile).st_mtime > start_time:
                    os.remove(ffile)
            elif ffile not in excepts and os.path.splitext(f_name)[1] not in('.pyc', '.pyx'):
                if os.path.splitext(f_name)[1] in('.py', '.pyx') and not f_name.startswith('__'):
                    yield os.path.join(parentpath, name, f_name)
                elif copyOther:
                        dstdir = os.path.join(basepath, build_dir, parentpath, name)
                        if not os.path.isdir(dstdir): os.makedirs(dstdir)
                        shutil.copyfile(ffile, os.path.join(dstdir, f_name))
        else:
            pass


if __name__ == '__main__':
    module_list = list(getpy(basepath=curr_dir,parentpath=parent_path, excepts=(setup_file)))
    try:
        setup(ext_modules=cythonize(module_list),script_args=["build_ext", "-b", build_dir, "-t", build_tmp_dir])
    except Exception as e:
        print(f'error {e}')
    else:
        module_list = list(getpy(basepath=curr_dir, parentpath=parent_path, excepts=(setup_file), copyOther=True))
    module_list = list(getpy(basepath=curr_dir, parentpath=parent_path, excepts=(setup_file), delC=True))
    if os.path.exists(build_tmp_dir):
        shutil.rmtree(build_tmp_dir)
    print('end')