.py文件批量转.pyd文件

环境准备

pip install cython

代码

create_pyd.py

import os
import shutil
import time
import sys

def func(path):
    folder_path = os.path.dirname(path)
    file_path = os.path.split(path)[1]
    os.chdir(folder_path)
    with open('setup.py', 'w') as f:
        f.write('from setuptools import setup\n')
        f.write('from Cython.Build import cythonize\n')
        f.write('setup(\n')
        f.write("name='test',\n")
        f.write("ext_modules=cythonize('%s')\n" % file_path)
        f.write(")\n")
    os.system('python setup.py build_ext --inplace')
    filename = file_path.split('.py')[0]
    time.sleep(2)
    # 这里的cp37-win_amd64需要注意一下，这个是依据python解释器类型以及windows版本生成的，建议是单个生成一个pyd文件然后相应修改一下
    os.rename('%s\\%s.cp37-win_amd64.pyd' % (folder_path, filename), '%s\\%s.pyd' % (folder_path, filename))
    # 这个是删除py源文件，测试的时候可以先注释掉查看效果
    os.remove('%s.c' % filename)
    build_folder_path = os.path.join(folder_path, 'build')
    # 删除掉生成的build文件夹
    shutil.rmtree(build_folder_path)
    os.remove('setup.py')
    os.remove(file_path)

def get_all_file(path):
    for root, dirs, files in os.walk(path):
        for name in files:
            if name.endswith(".py"):
                file_path = os.path.join(root, name)
                func(file_path)

paths = sys.argv[1]
get_all_file(paths)

用法

python create_pyd.py D:\python_demo\demo  ##文件夹的绝对路径