Как выполнить только определенную функцию из другого файла

Question

У меня есть два файла:

test2.py:

def fun(): 
    print "from fun" 
print "from test2" 

test.py:

from test2 import fun 
print "in text" 
fun() 

Я хочу, чтобы выполнить только функцию fun от test2.py, но я также получил from test2.

Как использовать только функцию, которую я хочу?

Answer 1

В test2.py вам нужно убедиться, что «из test2» напечатан только тогда, когда программа запускается сама по себе.

Вы можете сделать это следующим образом:

test2.py

def fun(): 
    print "from fun" 

if __name__ == "__main__": 
    print "from test2"