回答

收藏

if __name__ == "__main__": 怎么办?

技术问答 技术问答 294 人阅读 | 0 人回复 | 2023-09-11

给出以下代码有什么作用?if __name__ == "__main__":?
* p$ O4 w6 s7 w( ]
    # Threading exampleimport time,threaddef myfunction(string,sleeptime,lock,*args):    while True:        lock.acquire()        time.sleep(sleeptime)        lock.release()        time.sleep(sleeptime)if __name__ == "__main__":    lock = thread.allocate_lock()    thread.start_new_thread(myfunction,("Thread #: 1",2,lock))    thread.start_new_thread(myfunction,("Thread #: 2",2,lock))
    1 Y5 r2 y! z  r+ {7 k, W
               
( h+ f/ X) L2 m( _2 |" h- B- j4 \    解决方案:                                                                4 i8 {, |9 K: U9 U: t
                                                                简答它是模型代码,可以防止用户无意中调用脚本。以下是脚本中省略守卫的一些常见问题:1 w/ t3 W6 e! o$ P. ^6 i
如果你在另一个脚本(例如(例如)import my_script_without_a_name_eq_main_guard)导入无保护脚本时,第二个脚本将触发第一个脚本在导入时运行并使用第二个脚本的命令行参数。这几乎总是一个错误。( ]) G5 E3 G. D, ]& j
如果你在guardless 脚本中有一个自定义类,并将其保存到pickle 如果在另一个脚本中取消触发guardless 脚本的引入与上一个项目符号中概述的问题相同。
长答案为了更好地理解为什么以及它有多重要,我们需要退一步来理解 Python 如何初始化脚本,如何与模块导入机制交互。4 S+ k) J! y1 ^* f% W
每当 Python 当解释器读取源文件时,它会做两件事:4 @$ |: o0 j0 T1 H  ?
它设置了一些特殊的变量,如__name__,然后
2 i0 r# K% q; z- Y/ J, X在执行文件中找到的所有代码。
让我们看看它是如何工作的,以及它和它__name__我们在 Python 脚本中经常看到的检查相关问题有什么关系?
, H0 |" Z# g/ Z代码示例让我们用稍微不同的代码示例来探索导入和脚本的工作方法。假设以下内容在名称中foo.py.& T& L8 ?5 L, G
    # Suppose this is foo.py.print("before import")import mathprint("before functionA")def functionA():    print("Function A")print("before functionB")def functionB():    print("Function B {}".format(math.sqrt(100)))print("before __name__ guard")if __name__ == '__main__    functionA()    functionB()print("after __name__ guard")
    3 n. t- K- c$ }- @) d: q
特殊变量当 Python 当解释器读取源文件时,它首先定义了一些特殊的变量。在这种情况下,我们关心__name__变量。$ e9 \' u+ [+ J$ W( A! Z: z" k
当你的模块是主程序时
' ]1 |8 }, t/ ?9 z" p% }! L# ]1 g5 N以模块(源文件)为主程序,例如) P4 A3 s6 w; Q0 C7 I
    python foo.py
    . O. X. z; u+ c4 N
解释器将硬编码字符串赋值"__main__"给__name__变量,即
4 F8 f. W: q$ n7 x9 Y8 s5 ]  ]/ ]* B
    # It's as if the interpreter inserts this at the top# of your module when run as the main program.__name__ = "__main__"
    - d7 @3 I4 i- Q% {1 X  }# N3 h+ o
当您的模块被另一个导入时
; O) D! l" B: e4 g+ P0 o另一方面,假设其他模块是主程序,并导入您的模块。这意味着主程序或主程序导入的其他模块中有这样的句子:6 R3 @/ }* z' w5 s, }2 M) w
    # Suppose this is in some other main program.import foo; n' o; M/ ^( L* H2 H. L. J
解释器会搜索你的foo.py文件(以及搜索其他一些变体),它将在执行该模块之前"foo"导入语句中的名称分配给__name__变量,即, O  a7 n' A9 q5 C1 T
    # It's as if the interpreter inserts this at the top# of your module when it's imported from another module.__name__ = "foo"3 j# b4 L- H  ~7 r, G" u9 C
代码执行模块设置特殊变量后,解释器执行模块中的所有代码,一次一句。您可能希望在带有代码示例的一侧打开另一个窗口,以便您可以按照此说明操作。
% P( m2 z% K5 I7 c# X4 V) G9 A总是. P0 e1 ~- r9 a
[ol]它打印字符串"before import"(无引号)。
4 K) M' o; B. |1 q, ?; p它加载math将模块分配给名称math. 这相当于替换import math以下内容(请注意)__import__是 Python 接受字符串并触发实际导入的低级函数:[/ol]
    * H+ z1 x( D3 p# A
    # Find and load a module given its string name,"math",# then assign it to a local variable called math.math = __import__("math")/ a- T! I& e) \6 z
[ol]它打印字符串"before functionA"。
! Y, N* w' N7 @* F它执行def块,创建函数对象,然后将函数对象分配给一个名为 的变量functionA。
6 q. g1 ^6 t% [) I: m. Y! p它打印字符串"before functionB"。" G3 B, ?4 _+ X9 {) C# E, F' F
它执行第二个def块,创建另一个函数对象,然后分配给 变量functionB。
4 Y% Y9 u9 i8 Z! Y  C它打印字符串"before __name__ guard"。[/ol]只有当你的模块是主程序时
0 h0 z# h4 i# R) _6 Y# M[ol]假如你的模块是主程序,它会看到的__name__确实设置为"__main__"并调用两个函数打印字符串"Function A"和"Function B 10.0".[/ol]只有当您的模块被另一个导入时* L2 z7 }" P, H6 n; g
[ol](相反)假如你的模块不是主程序,而是由另一个程序导入,那么__name__will "foo",not "__main__",它将跳过if句子的主体。[/ol]总是
& J. `8 A+ U1 d[ol]它会"after __name__ guard"在两种情况下打印字符串。[/ol]概括  \, k+ d( o$ J- n7 {! t
总之,在两种情况下打印以下内容:
+ _# I' E- I; o- ~- h
    # What gets printed if foo is the main programbefore importbefore functionAbefore functionBbefore __name__ guardFunction AFunction B 10.0after __name__ guard# What gets printed if foo is imported as a regular modulebefore importbefore functionAbefore functionBbefore __name__ guardafter __name__ guard
    " @+ n6 k! ~& ]  C
为什么这样工作?你自然会想知道为什么有人想要这个。嗯,有时候你想写一个。.py其他程序和/或模块都可以用作模块,也可以用作主程序本身。5 w4 B" Z* a  y, ^
您的模块是一个库,但您想要一个脚本模式,它运行一些单元测试或演示。
! u, `' x$ v3 a0 i+ h- \) D您的模块仅用作主程序,但它有一些单元测试,通过导入测试框架.py脚本和其他文件并运行特殊的测试功能来工作。您不希望它尝试运行脚本,因为它正在导入模块。
- g$ @2 h* C: K3 J. M您的模块主要用作主程序,但也为高级用户提供了对程序员友好的 API。
除此之外, Python 操作脚本只设置一些魔法变量并导入脚本,非常优雅。操作脚本是导入脚本模块的副作用。* y2 N. P2 [; Y! B1 l/ S3 Z
问答问:我能有多个__name__答:这很奇怪,但语言不会阻止你。
7 D$ \6 n) j- l+ z假设以下内容是foo2.py. 如果你python foo2.py说命令行会发生什么?为什么?

    4 D) a% D( o! ]0 G3 [2 V5 m# Suppose this is foo2.py.import os,sys; sys.path.insert(0,os.path.dirname(__file__)) # needed for some interpretersdef functionA():    print("a1")    from foo2 import functionB    print("a2")    functionB()    print("a3")def functionB():    print("b")print("t1")if __name__ == "__main__":    print("m1")    functionA()    print("m2")print("t2")5 B+ q2 O. n" R4 z' p, r
现在,如果你取消取消了它__name__签入会发生什么?foo3.py:
    ; W0 Q1 a6 q- p2 ^2 B0 _
    # Suppose this is foo3.py.import os,sys; sys.path.insert(0,os.path.dirname(__file__)) # needed for some interpretersdef functionA():    print("a1")    from foo3 import functionB    print("a2")    functionB()    print("a3")def functionB():    print("b")print("t1")print("m1")functionA()print("m2")print("t2")
    0 z7 q; A$ S) F* m4 Z3 |1 e( O  {
当它被用作脚本时,它会做什么?导入模块时?
    / {( a6 B* G. X6 |  b) a
    # Suppose this is in foo4.py__name__ = "__main__"def bar():    print("bar")print("before __name__ guard")if __name__ == "__main__":    bar()print("after __name__ guard")
    $ g) O6 N  W, w% l
分享到:
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则