批量将python 文件编译成so系统文件

from distutils.core import setup
from Cython.Build import cythonize
import os
import shutil
# import argparse
# parser = argparse.ArgumentParser() ###和自带参数build_ext --inplace冲突，暂时无法使用
# parser.add_argument("--dir",type=str,default="./test",help="root dir",)
# args = parser.parse_args()
root_dir='XGB' ##目标根目录
res=[] ##存放文件树下所有py文件地址
for root, dirs, fs in os.walk(root_dir):
'''
root 表示当前正在访问的文件夹路径
dirs 表示该文件夹下的子目录名list
fs 表示该文件夹下的文件list
'''
# # 遍历所有的文件夹
# for d in dirs:
# d=os.path.join(root, d)
# print(d)
# 遍历文件
for f in fs:
f=os.path.join(root, f)
if f.endswith('py') and '__init__' not in f:
res.append(f)
#####################################################
#generate .so files
xx=setup(ext_modules = cythonize(res))
#####################################################
work_path=os.path.split(res[0])[0]
end=''
for j in os.listdir(work_path):###获取.so文件前缀，只用执行一次
if j.endswith('so'):
end=j.split('.')[1:]
end='.'.join(end)
break
#####################################################
for i in res:
os.remove(i) ##删除*.py
os.remove(i.replace('py','c')) ##删除*.c