提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
- 前言
- 一、python项目转化为so文件
- 1.引入库
- 总结
前言
提示:这里可以添加本文要记录的大概内容:
提示:以下是本篇文章正文内容,下面案例可供参考
一、python项目转化为so文件
示例:。
1.引入库
代码如下(示例):
# -*- coding:utf-8 -*-
# author: cbj
# datetime: 2024/7/8 9:02
# @File: setup3.py
# -*- coding:utf-8 -*-
# author: cbj
# datetime: 2024/7/5 16:45
# @File: 134打包python项目so文件3.py.py
# -* -coding: UTF-8 -* -
"""
执行前提:
系统安装python-devel 和 gcc
Python安装cython
编译整个当前目录:
python py-setup.py
编译某个文件夹:
python py-setup.py BigoModel
生成结果:
目录 build 下
生成完成后:
启动文件还需要py/pyc担当,须将启动的py/pyc拷贝到编译目录并删除so文件
"""
import sys, os, shutil, time
from distutils.core import setup
from Cython.Build import cythonize
from pathlib import Path
starttime = time.time()
currdir = os.path.abspath('.') # 打包文件的地址
# currdir = os.path.abspath(r'D:\data1\20240705\data_augmentation20230914\data_augmentation')
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"
ignore_file = ["main.py", "clean.py", "setup.py", "setup2.py", "forward_main.py", "utils.py"]
ignore_dir_name = ['migrations', '__pycache__', 'static', '.idea']
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):
if fname in ignore_dir_name:
continue
ffile = os.path.join(fullpath, fname)
# print basepath, parentpath, name,file
if os.path.isdir(ffile) and fname != build_dir and not fname.startswith('.'):
for f in getpy(basepath, os.path.join(parentpath, name), fname, excepts, copyOther, delC):
yield f
elif os.path.isfile(ffile):
ext = os.path.splitext(fname)[1] # 取后缀
if ext == ".c":
if delC and os.stat(ffile).st_mtime > starttime:
# os.remove(ffile)
pass
elif ffile not in excepts and os.path.splitext(fname)[1] not in ('.pyc', '.pyx'):
if fname in ignore_file:
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, fname))
continue
if os.path.splitext(fname)[1] in ('.py', '.pyx') and not fname.startswith('__'):
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)
shutil.copyfile(ffile, os.path.join(dstdir, fname))
else:
pass
# 获取py列表
module_list = list(getpy(basepath=currdir, parentpath=parentpath, excepts=(setupfile)))
try:
setup(ext_modules=cythonize(module_list, language_level='3'),
script_args=["build_ext", "-b", build_dir, "-t", build_tmp_dir])
pass
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))
if os.path.exists(build_tmp_dir): shutil.rmtree(build_tmp_dir)
print("complate! time:", time.time() - starttime, 's')
# if __name__ == '__main__':
#
# module_list = list(getpy(basepath=currdir, parentpath=parentpath, excepts=(setupfile)))
#
# pass
总结
提示:这里对文章进行总结:
例如:以上就是今天要讲的内容