Python 中的 if __name__ == '__main__'

對于很多編程語言來說,程序都必須要有一個入口,比如 C,C++,以及完全面向?qū)ο蟮木幊陶Z言 Java,C# 等。如果你接觸過這些語言,對于程序入口這個概念應該很好理解,C 和 C++ 都需要有一個 main 函數(shù)來作為程序的入口,也就是程序的運行會從 main 函數(shù)開始。同樣,Java 和 C# 必須要有一個包含 Main 方法的主類來作為程序入口。

而 Python 則有不同,它屬于腳本語言,不像編譯型語言那樣先將程序編譯成二進制再運行,而是動態(tài)的逐行解釋運行。也就是從腳本第一行開始運行,沒有統(tǒng)一的入口。

一個 Python 源碼文件除了可以被直接運行外,還可以作為模塊(也就是庫)被導入。不管是導入還是直接運行,最頂層的代碼都會被運行(Python 用縮進來區(qū)分代碼層次)。而實際上在導入的時候,有一部分代碼我們是不希望被運行的。

舉一個例子來說明一下,假設我們有一個 const.py 文件,內(nèi)容如下:

PI=3.14defmain():print"PI:",PImain()

我們在這個文件里邊定義了一些常量,然后又寫了一個 main 函數(shù)來輸出定義的常量,最后運行 main 函數(shù)就相當于對定義做一遍人工檢查,看看值設置的都對不對。然后我們直接執(zhí)行該文件(python const.py),輸出:

PI:3.14

現(xiàn)在,我們有一個 area.py 文件,用于計算圓的面積,該文件里邊需要用到 const.py 文件中的 PI 變量,那么我們從 const.py 中把 PI 變量導入到 area.py 中:

fromconstimportPIdefcalc_round_area(radius):returnPI*(radius**2)defmain():print"round area: ",calc_round_area(2)main()

運行 area.py,輸出結(jié)果:

PI: 3.14

round area:? 12.56

可以看到,const 中的 main 函數(shù)也被運行了,實際上我們是不希望它被運行,提供 main 也只是為了對常量定義進行下測試。這時,if __name__ == '__main__'就派上了用場。把 const.py 改一下:

PI=3.14defmain():print"PI:",PIif__name__=="__main__":main()

然后再運行 area.py,輸出如下:

round area:? 12.56

再運行下 const.py,輸出如下:

PI:3.14

這才是我們想要的效果。

if __name__ == '__main__'?就相當于是 Python?模擬的程序入口。Python 本身并沒有規(guī)定這么寫,這只是一種編碼習慣。由于模塊之間相互引用,不同模塊可能都有這樣的定義,而入口程序只能有一個。到底哪個入口程序被選中,這取決于?__name__?的值。

__name__

__name__?是內(nèi)置變量,用于表示當前模塊的名字,同時還能反映一個包的結(jié)構。來舉個例子,假設有如下一個包:

a

├── b

│? ├── c.py

│? └── __init__.py

└── __init__.py

目錄中所有 py 文件的內(nèi)容都為:

print __name__

我們執(zhí)行?python -c "import a.b.c",輸出結(jié)果:

a

a.b

a.b.c

由此可見,__name__?可以清晰的反映一個模塊在包中的層次。其實,所謂模塊名就是 import 時需要用到的名字,例如:

importtornadoimporttornado.web

這里的 tornado 和 tornado.web 就被稱為模塊的模塊名。

如果一個模塊被直接運行,則其沒有包結(jié)構,其?__name__?值為?__main__。例如在上例中,我們直接運行 c.py 文件(python a/b/c.py),輸出結(jié)果如下:

__main__

所以,if __name__ == '__main__'?我們簡單的理解就是:?如果模塊是被直接運行的,則代碼塊被運行,如果模塊是被導入的,則代碼塊不被運行

實際上,這個問題還可以衍生出其他的一些知識點,例如?__main__.py?文件與 Python 的?-m?參數(shù)。

__main__.py 文件與 python -m

Python 的?-m?參數(shù)用于將一個模塊或者包作為一個腳本運行,而?__main__.py?文件則相當于是一個包的”入口程序“。

首先我們需要來看看?python xxx.py?與?python -m xxx.py?的區(qū)別。兩種運行 Python 程序的方式的不同點在于,一種是直接運行,一種是當做模塊來運行。

先來看一個簡單的例子,假設有一個 Python 文件 run.py,其內(nèi)容如下:

importsysprintsys.path

我們用直接運行的方式啟動(python run.py),輸出結(jié)果(為了說明問題,輸出結(jié)果只截取了重要部分,下同):

['/home/huoty/aboutme/pythonstudy/main',...]

然后以模塊的方式運行(python -m run.py):

['',...]/usr/bin/python:Nomodulenamedrun.py

由于輸出結(jié)果只列出了關鍵的部分,應該很容易看出他們之間的差異。直接運行是把 run.py 文件所在的目錄放到了 sys.path 屬性中。以模塊方式運行是把你輸入命令的目錄(也就是當前工作路徑),放到了 sys.path 屬性中。以模塊方式運行還有一個不同的地方是,多出了一行?No module named run.py?的錯誤。實際上以模塊方式運行時,Python 先對 run.py 執(zhí)行一遍 import,所以?print sys.path被成功執(zhí)行,然后 Python 才嘗試運行 run.py 模塊,但是,在 path 變量中并沒有 run.py 這個模塊,所以報錯。而正確的運行方式,應該是?python -m run.

這個例子并不能明顯的說明問題。接著我們來看看?__main__.py?的作用。

仍然先看例子,有如下一個包:

package├──__init__.py└──__main__.py

__init__.py

importsysprint"__init__"printsys.path

__main__.py

importsysprint"__main__"printsys.path

用?python -m package?運行結(jié)果:

__init__

['', ...]

__main__

['', ...]

用?python package?運行結(jié)果:

__main__

['package', ...]

然后我們來總結(jié)一下:

1、?加上 -m 參數(shù)時會把當前工作目錄添加到 sys.path 中,而不加時則會把腳本所在目錄添加到 sys.path 中

2、?加上 -m 參數(shù)時 Python 會先將模塊或者包導入,然后再執(zhí)行

3、?__main__.py 文件是一個包或者目錄的入口程序。不管是用?python package?還是用?python -m package?運行時,__main__.py 文件總是被執(zhí)行。

后序

我試圖使用長篇大論來闡述,在 Python 中如何理解?if __name__ == '__main__'?這個問題,不知道我有沒有描述得足夠的明白。Python 的確是簡單的,優(yōu)雅的,但也有很多問題是不太容易理解的,例如很多高級的特性,像元類、生成器表達式、描述符、協(xié)程等。Python 并沒有在太多的地方規(guī)定要如何如何,很多的用法只是慣用法,例如 self 和本文討論的內(nèi)容。這些用法或是為了讓代碼看起來更優(yōu)雅,或是前人的經(jīng)驗。使用 Python 是有無限可能的,你可以寫出很多簡潔優(yōu)雅的代碼。

參考資料

?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 227,572評論 6 531
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,071評論 3 414
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,409評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,569評論 1 307
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,360評論 6 404
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 54,895評論 1 321
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 42,979評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,123評論 0 286
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,643評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,559評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,742評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,250評論 5 356
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 43,981評論 3 346
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,363評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,622評論 1 280
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,354評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,707評論 2 370

推薦閱讀更多精彩內(nèi)容