修改 python3 的 import 函数,从而实现挂钩hook或更改其行为

原始想法是突破python对顶层包的限制,从而允许在当前目录或父目录使用相对位置加载模块。

方法发现自 pycharm 的python挂钩代码 pydev_import_hook.py
该文件可以在pycharm的 PyCharm\plugins\python-ce\helpers\pydev_pydev_bundle 目录中找到

原理,修改 builtins.import 函数即可实现目的
本修改对整个python程序生效

下面是一个例子,阻止 collections 模块加载

import builtins

ori_import = builtins.__import__


def my_import(name, globals=None, locals=None, fromlist=(), level=0):
    if name == 'collections':
        print('不允许加载 collections 模块', flush=True)
        raise ImportError()
    else:
        print(f'即将加载 {name} 模块', flush=True)
    r = ori_import(name, globals, locals, fromlist, level)
    print(f'成功加载 {name} 模块', flush=True)
    return r

builtins.__import__ = my_import

# 加载 typing 模块,将会成功
import typing

# 加载collections模块,将会失败
import collections

参考:
https://www.programcreek.com/python/example/58937/builtins.import

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值