Py学习  »  Python

多个 Python 项目怎么调用我自己的工具函数?

进击的Coder • 2 年前 • 198 次点击  
这是「进击的Coder」的第 604 篇技术分享
作者:kingname
来源:未闻 Code

阅读本文大概需要 4 分钟。


在多年写代码的过程中,我总结了不少常用的工具函数。这些工具函数有的能够实现快速重试网络请求,有的可以把任意格式的时间转成标准格式,还有的可以自动生成正则表达式。

我把这些工具函数分别放到多个.py文件中。然后把这些.py文件放在一个叫做my_awesome_util的文件夹中。

由于我没有把这些代码上传到 Pypi 或者 Github,因此我每次在新的项目中要使用时,都要把my_awesome_util文件夹复制到新的项目中,非常麻烦,而且会形成大量的重复代码。

有没有什么办法,能让新开的项目直接就能导入这些工具函数呢?就像导入官方模块import time一样,如果我要使用时间相关的工具函数,我只需要import time_util就可以了。

首先要排除的方法,就是手动在代码里面,通过sys.path添加文件夹。因为这个方法需要在每个项目的入口文件中增加两行代码,无法做到全自动

今天的方法,是在 Python 的site-packages文件夹中做文章。相信很多同学都知道,任何放到site-packages文件夹里面的.py文件或者文件夹,都可以在 Python 中直接导入:

看到这里,肯定有同学要问,那是不是把my_awesome_util文件夹整个复制到site-packages文件夹中呢?

当然不是的。因为我的工具包里面的代码是在持续演进的,我会经常更新里面的代码,放到site-packages里面以后,代码修改起来很不方便。

我们要用到的,是 Python 的一个特性,叫做.pth文件。在site-packages文件夹中,任意创建一个.pth文件。文件名可以随便取,只要不跟已有的模块冲突就可以了。例如叫做xxxx.pth。这个文件里面只有一行,就是my_awesome_util文件夹的绝对路径。例如:

现在,我在任何一个文件夹中启动 Python,直接就可以导入my_awesome_util文件夹中的任何.py文件:

.pth文件可以有很多行,每行一个文件夹的绝对路径,就可以让 Python 同时导入这些文件夹里面的所有.py文件。

有时候,我们的电脑上有多个虚拟环境。A 环境有 pandas,没有 pymongo;B 环境有 pymongo,但没有 pandas。现在我在 B 环境里面要写一段代码,既需要 pandas,又需要 pymongo 怎么办呢?如果不想重新安装,我可以把 A 环境的site-packages文件夹的绝对路径,写入到一个.pth文件中,然后把这个文件存放到 B 环境的site-packages文件夹中。这样,我就可以在B环境里面直接使用 A 环境中的所有第三方库了。

有同学会问,怎么知道site-packages文件夹在哪里呢?其实只需要执行下面两行代码就能查询到:

>>> import sysconfig
>>> print(sysconfig.get_path('purelib'))

运行效果如下图所示:



End

崔庆才的新书《Python3网络爬虫开发实战(第二版)》已经正式上市了!书中详细介绍了零基础用 Python 开发爬虫的各方面知识,同时相比第一版新增了 JavaScript 逆向、Android 逆向、异步爬虫、深度学习、Kubernetes 相关内容,‍同时本书已经获得 Python 之父 Guido 的推荐,目前本书正在七折促销中!

内容介绍:《Python3网络爬虫开发实战(第二版)》内容介绍



扫码购买




好文和朋友一起看~


Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/131863
 
198 次点击