有没有办法使代码与"import"一起使用？

Question

我从一个脚本中发现了一个代码，阻止它用import打开

def begin(stdscr):
stdscr.clear()
curses.init_pair(1, curses.COLOR_GREEN, curses.COLOR_BLACK)
curses.init_pair(2, curses.COLOR_RED, curses.COLOR_BLACK)
init_maze()
generate_maze()
mainloop()
if __name__ == '__main__':
curses.setupterm()
stdscr = curses.initscr()
curses.curs_set(False)
stdscr.keypad(True)
cols = curses.tigetnum('cols')
lines = curses.tigetnum('lines')
while width < cols and height + 5 < lines:
curses.wrapper(begin)
width += 10
height += 6

主要是if __name__ == '__main__':

谁能告诉我如何使游戏可以没有if __name__ == '__main__':和import工作?

编辑:如果我删除if __name__ == '__main__':，脚本不再工作

Answer 1

只要您的模块与您试图导入模块的代码位于同一目录中，我相信这应该可以工作:

def begin(stdscr):
stdscr.clear()
curses.init_pair(1, curses.COLOR_GREEN, curses.COLOR_BLACK)
curses.init_pair(2, curses.COLOR_RED, curses.COLOR_BLACK)
init_maze()
generate_maze()
mainloop()
def main():
curses.setupterm()
stdscr = curses.initscr()
curses.curs_set(False)
stdscr.keypad(True)
cols = curses.tigetnum('cols')
lines = curses.tigetnum('lines')
while width < cols and height + 5 < lines:
curses.wrapper(begin)
width += 10
height += 6
if __name__ == "__main__":
main()

从你导入的脚本中，你可以这样做:

import my_script.py
my_script.main()