作為一個乙方的算法研究員,疲于應付各種項目,每天至少要寫幾百行腳本,雖然絕大多數都是一次性代碼,但是也有少量有復用價值的代碼,所以準備把日常用到的所有代碼整理成一個可以隨時調用的庫。
今天是2022年10月7日,記錄一下怎么把Python代碼封裝成庫。
1. 項目結構:

項目結構
每個模塊下面都有一個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. 使用
經過上述步驟,這個dl庫就可以在任意目錄中使用了

image.png