Python主脚本导入的库如何让子模块直接复用，无需重复导入？

针对你不想重复导入库、不想传递大量参数，同时希望子模块能直接使用主脚本资源的需求，这里提供几个实用且优雅的方案：

方案1：传递主模块全局命名空间给子模块初始化 #

通过子模块的初始化函数接收主脚本的全局命名空间，让子模块直接引用主脚本的资源。

修改 library/login.py： #

# 存储主模块的全局命名空间
_main_globals = None

def init(main_globals):
    global _main_globals
    _main_globals = main_globals

def login():
    # 从主模块命名空间中获取所需资源
    color = _main_globals['color']
    getpass = _main_globals['getpass']
    debug = _main_globals['debug']
    
    username = input(color.green + "Enter your username" + color.reset)
    if debug:
        print("Debug - Username entered:", username)
    password = getpass.getpass(color.red + "Now enter your password" + color.reset)
    if debug: 
        print("Debug - Password entered:", password)
    return [username, password]

修改 main.py： #

import os
import sys
debug = True

import requests
import getpass 

libdir = os.path.join(os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__))),"library")
sys.path.append(libdir)

import login
import color

# 初始化login模块，传入主脚本的全局命名空间
login.init(globals())

session = login.login()

方案2：创建共享上下文模块（推荐） #

专门维护一个共享上下文模块，统一管理所有需要共享的库和全局变量，所有模块从这里导入资源，结构更清晰易维护。

在 library 目录下新建 context.py： #

# 统一导入需要共享的第三方库
import requests
import getpass

# 定义共享的全局变量
debug = False  # 初始值可在main中修改

修改 main.py： #

import os
import sys

libdir = os.path.join(os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__))),"library")
sys.path.append(libdir)

# 导入共享上下文，并设置全局变量
from library import context
context.debug = True

import login
import color

session = login.login()

修改 library/login.py： #

from library import context
import color  # 本地库也可放入context统一管理

def login():
    username = input(color.green + "Enter your username" + color.reset)
    if context.debug:
        print("Debug - Username entered:", username)
    password = context.getpass.getpass(color.red + "Now enter your password" + color.reset)
    if context.debug: 
        print("Debug - Password entered:", password)
    return [username, password]

方案3：直接合并主模块全局命名空间到子模块（极简但需谨慎） #

如果不想修改子模块代码，可在主脚本中直接将自身全局命名空间合并到子模块中，子模块无需任何改动即可直接使用主脚本资源。

修改 main.py： #

import os
import sys
debug = True

import requests
import getpass 

libdir = os.path.join(os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__))),"library")
sys.path.append(libdir)

import login
import color

# 将主脚本的全局资源合并到login模块的命名空间
login.__dict__.update(globals())

session = login.login()

⚠️ 注意：此方法会覆盖子模块中与主脚本同名的变量/函数，使用前需确保无命名冲突。

内容的提问来源于stack exchange，提问作者HeartOfGermany