Python 阻止导入ipynb执行导入的文件

我有两个

.ipynb

，我正在使用import

import\u ipynb

从文件

a.ipynb

中的文件

B.ipynb

导入一个函数，但是当我使用

从导入某个东西时，会运行文件
a.ipynb
，并打印我在该文件中打印的内容：

文件
A.ipynb
：

for i in range(0, 5):
    print(i)
    
def something():
    return 'something'

import import_ipynb
from A import something

print(something())

文件
B.ipynb
：

for i in range(0, 5):
    print(i)
    
def something():
    return 'something'

import import_ipynb
from A import something

print(something())

当我运行
B.ipynb
时，输出为：

importing Jupyter notebook from A.ipynb
0
1
2
3
4
something

但是我刚刚导入了
something
函数，我不希望打印函数在文件
A.ipynb
中执行，只需将
\uuuuu main\uuu==“\uu名称”
添加到文件A中，这样会在导入时阻止它运行

if __name__ == "__main__":
    for i in range(0, 5):
       print(i)
    
def something():
    return 'something'