快捷導(dǎo)航

Python 項目轉(zhuǎn)化為so文件實例

更新時間：2019年12月23日 11:39:13 作者：gmHappy

今天小編就為大家分享一篇Python 項目轉(zhuǎn)化為so文件實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

思路是先將py轉(zhuǎn)換為c代碼，然后編譯c為so文件，所以要安裝以下內(nèi)容：

python 安裝:cython

pip install cython

linux 安裝:python-devel,gcc

yum install python-devel
yum install gcc

初步編譯

新建Test.py文件，內(nèi)容如下

class test:
  
  def __init__(self):
    print('init')

  def say(self):
    print ('hello')

新建setup.py，內(nèi)容如下

from distutils.core import setup
from Cython.Build import cythonize

setup(ext_modules = cythonize(["Test.py"]))

在bash中執(zhí)行

python setup.py build_ext

運行后會生成build文件夾，如下

現(xiàn)在so文件就可以像普通py文件一樣導(dǎo)入了

集成編譯

做了以下內(nèi)容：

1.文件夾編譯

2.刪除編譯出的.c文件

3.刪除編譯的temp文件夾

將需要編譯的目錄和setup.py放在同一層級,執(zhí)行python setup.py，so文件在build目錄下

setup.py代碼如下：

'''
Created on 2019年3月27日

@author: hylink
'''
#-* -coding: UTF-8 -* -

"""
執(zhí)行前提：
  系統(tǒng)安裝python-devel 和 gcc
  Python安裝cython

編譯整個當(dāng)前目錄：
  python py-setup.py
編譯某個文件夾：
  python py-setup.py BigoModel

生成結(jié)果：
  目錄 build 下

生成完成后：
  啟動文件還需要py/pyc擔(dān)當(dāng)，須將啟動的py/pyc拷貝到編譯目錄并刪除so文件

"""

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 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)
      elif ffile not in excepts and os.path.splitext(fname)[1] not in('.pyc', '.pyx'):
        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),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))
if os.path.exists(build_tmp_dir): shutil.rmtree(build_tmp_dir)
print ("complate! time:", time.time()-starttime, 's')

注意問題

1.編譯后執(zhí)行需要相同的python版本和編碼

2.py中使用__file__內(nèi)置變量的文件編譯后調(diào)用時會出問題，暫時沒有解決，還需要使用pyc代替

3.使用時注意權(quán)限控制

以上這篇Python 項目轉(zhuǎn)化為so文件實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: