python - 将信息传递给导入的 Python 模块

Question

我希望我可以让我的模块根据主程序中选择的选项以两种不同的方式进行初始化。如意代码:

主程序:

<magic variable> = True
import mymodule

我的模块.py:

def start1():
    ...
def start2():
    ...
if <magic variable>: start1()
else: start2()

有什么方法可以将魔法变量放在导入模块可以找到的地方吗？不创建另一个模块？

Answer 1

我要解决这个问题的方法是在导入的模块中创建一个用于初始化的通用函数。然后用户将传入标志以选择某种初始化样式:

我的模块.py

def init(flag=True):
    if flag:
        start1()
    else:
        start2()

def start1():
    print('start1 called')

def start2():
    print('start2 called')

主.py

import mymodule
mymodule.init() # Or False could be passed in, depending on the users choice.