python except无法捕获同名异常

看到这个标题的人估计和我遇到这个问题的时候一样蒙蔽。简而言之，就是在try里面抛出A异常，但是except A却捕获不到

场景

custom_exception.py 自定义异常KnownException用来抛出给用户的各种提示

class KnownException(Exception):
    pass

test_module.py 逻辑里面抛出了KnownException

from custom_exception import KnownException


def run():
    raise KnownException("测试抛出异常")

test_module2.py 调用test_module.py里的run方法，并捕获KnownException异常

from test_package.test_module import run
from test_package.custom_exception import KnownException


try:
    run()
except KnownException:
    print("捕获成功!")

整体文件树如下：

└─testException
    │  test_module2.py
    │
    └─test_package
            custom_exception.py
            test_module.py
            __init__.py

结果运行test_moduels2.py之后KnownException异常仍然抛出来了

分析一波

python在导入新模块的时候会向sys.modules进行注册，下次导入时就会直接找到对应的模块。在test_module2.py里面如下内容，打印一下执行环境里的注册字典

import sys
from test_package.test_module import run
from test_package.custom_exception import KnownException

for module in sys.modules:
    print(sys.modules.get(module))
try:
    print("try打印：{}".format(KnownException))
    run()
except KnownException:
    print("捕获成功!")

通过观察发现字典里面存了两个KnownException，原来python通过包路径访问模块，即使是相同的文件相同的代码，引入的路径不一样，就会产生不同的加载。

瞬间我就反思下自己之前写过的代码：数据库访问模块是否有重复加载过？配置文件是否有重复加载过？静态文件是否有重复加载过？

python的包管理机制是个简单、容易让人忽略的问题，值得记录一下~~

解决办法

解决办法就是将test_module.py里

from custom_exception import KnownException
改成
from .custom_exception import KnownException