社区编辑申请
注册/登录
一日一技:如何让自己的工具函数在Python全局可用?
开发 前端
在Python里面,所有内置的函数或者类都在builtins​模块中,因此在代码里面可以直接使用而不用导入。现在只需要把我们自定义的工具函数注册到builtins模块中,就能让它跟内置函数有相同的效果。

我们在开发Python项目的时候,经常会写一些工具函数。为了在项目里面多个.py文件中使用这个工具函数,就不得不在多个地方都导入它,非常麻烦。

例如下面这个例子:

图片

在A.py和C.py​文件都要使用clean_msg​这个工具函数,那么他们就都要从util.py​中导入clean_msg。这似乎理所当然。

但今天我在看icecream/builtins.py[1]源代码的时候,突然发现了一个高级用法,可以让我们使用工具函数的时候,就像使用Python的print函数一样,不用导入,而是直接使用。

我们先来看看效果:

图片

大家注意A.py和C.py​,我并没有导入clean_msg而是直接使用了这个函数。并且运行完全正常。

关键原理就在入口文件main.py,被我框住的3行:

import builtins

from util import clean_msg
setattr(builtins, 'clean_msg', clean_msg)

在Python里面,所有内置的函数或者类都在builtins​模块中,因此在代码里面可以直接使用而不用导入。现在只需要把我们自定义的工具函数注册到builtins模块中,就能让它跟内置函数有相同的效果。

如果想把一个工具函数注册成为内置函数,只需要在入口文件中导入它,然后使用setattr​把它设置成builtins模块的一个属性。第二个参数是你想全局调用的时候的名字,第三个参数就是你需要注册的工具函数。名字可以跟工具函数的名字不相同,只要不跟已有的内置函数重复就好了。

注册完成以后,在这个项目的整个运行时中,任何.py文件里面都可以直接通过注册的名字调用这个工具函数,就跟调用内置函数一样。

参考资料

[1] icecream/builtins.py: https://github.com/gruns/icecream/blob/master/icecream/builtins.py​

责任编辑:武晓燕 来源: 未闻Code
相关推荐

2022-07-06 12:07:06

Python函数式编程

2022-04-11 20:59:48

工具函数Python

2022-03-30 08:37:32

2020-06-24 07:44:12

Python数据技术

2010-02-03 15:40:37

Python函数

2021-03-17 09:59:26

Python函数调用

2010-03-15 09:32:56

Python函数

2021-04-26 05:35:22

Python内置函数

2019-07-18 16:32:06

Python函数数据

2021-11-09 07:34:34

2021-05-15 10:16:14

2019-02-18 15:05:16

Python内置函数索引

2017-04-05 12:04:17

2018-02-01 14:15:00

2021-09-15 09:20:37

2021-07-10 10:01:37

2019-11-06 15:14:16

PythonNumbersSquare

2020-05-11 09:38:52

Python开发函数

2020-04-16 09:50:14

Python 开发效率

2020-09-25 16:20:21

Python内置函数字符串

同话题下的热门内容

如何创建一个无代码的自助客户聊天机器人Java 服务 Docker 容器化优秀实践程序员不得不知道的 API 接口常识如何提高 TypeScript 的代码质量?聊聊 13 种锁的实现方式11个 ES2022(ES13)中惊人的 JavaScript 新特性使用 Vite 和 TypeScript 带你从零打造一个属于自己的 Vue3 组件库Hooks时代,如何写出高质量的react和vue组件?

编辑推荐

太厉害了,终于有人能把TCP/IP协议讲的明明白白了!牛人5次面试腾讯不成功的经验HBase原理–所有Region切分的细节都在这里了Javascript如何监听页面刷新和关闭事件如何搭建一个HTTPS服务端
我收藏的内容
点赞
收藏

51CTO技术栈公众号