作為一個乙方的算法研究員菇晃,疲于應(yīng)付各種項(xiàng)目蕉毯,每天至少要寫幾百行腳本蹦疑,雖然絕大多數(shù)都是一次性代碼西雀,但是也有少量有復(fù)用價(jià)值的代碼,所以準(zhǔn)備把日常用到的所有代碼整理成一個可以隨時(shí)調(diào)用的庫歉摧。
今天是2022年10月7日艇肴,記錄一下怎么把Python代碼封裝成庫腔呜。
1. 項(xiàng)目結(jié)構(gòu):
項(xiàng)目結(jié)構(gòu)
每個模塊下面都有一個init.py文件
2. 編寫setup.py
from setuptools import setup, find_packages
setup(
name='dl',
version='0.1',
description='deeplearning toolkit',
author='dalalaa',
packages=find_packages()
)
3. 編寫__init__.py
dl/__init__.py
__all__ = ['module1', 'module2']
dl/module1/__init__.py
__all__ = ['add']
dl/module2/__init__.py
__all__ = ['minus']
4. 打包成whl
python setup.py bdist_wheel
5. 安裝whl
pip install dist/dl-0.1-py3-none-any.whl
6. 使用
經(jīng)過上述步驟,這個dl庫就可以在任意目錄中使用了
image.png