main函數(shù)與import用法

在python中，如果自己封裝的庫文件中有函數(shù)外的語句，那么這些語句是會在庫函數(shù)被導(dǎo)入時自動執(zhí)行的，這樣就會出現(xiàn)一些奇怪的問題，一個例子如下所示。

目錄結(jié)構(gòu)如下

• A.py
• B.py

B.py中的內(nèi)容如下：

def print_B():
? ? print("B")

print("starting B")

A.py中使用import導(dǎo)入print_B()方法：

from B import print_B
print_B()

結(jié)果如下：

starting B
B

可以發(fā)現(xiàn)B.py中的函數(shù)外語句是會自動先執(zhí)行的。但有時候我們確實希望直接運行封裝的庫能夠執(zhí)行某些功能，但又不希望這些功能在被import的時候被觸發(fā)，這個時候，就可以把這部分功能顯式的放在main方法中。

修改后的B.py如下所示：

def print_B():
? ? print("B")

if __name__ == "__main__":
? ? print("starting B")

此時運行A.py，結(jié)果如下：

B

main函數(shù)的作用以及解釋

python 中name == ‘__main__’ 的作用

• 經(jīng)典的英文解釋：Make a script both importable and executable
• 中文解釋：使腳本可以被調(diào)用import并且也可以直接運行

1、直接運行

# cat test_fun.py
def fun():
? ? print(__name__)
? ? print('this is fun')


if __name__ == '__main__':
? ? fun()
? ? print('this is main')

python test_fun.py
__main__
this is fun
this is main

2、被調(diào)用import

>>> import test_fun
>>> test_fun.fun()
test_fun
this is fun

調(diào)用導(dǎo)入時：此處輸出沒有顯示”main“,也就是說模塊name = ‘main’ 下面的代碼并未執(zhí)行,main函數(shù)沒有執(zhí)行。

這個功能還有一個用處：

調(diào)試代碼的時候，在”if name == ‘main‘“中加入一些我們的調(diào)試代碼，我們可以讓外部模塊調(diào)用的時候不執(zhí)行我們的調(diào)試代碼，但是如果我們想排查問題的時候，直接執(zhí)行該模塊文件，調(diào)試代碼能夠正常運行！