如何获取Python入口文件依赖的本地模块列表以构建轻量Docker镜像？

一、现成工具推荐 #

• modulegraph：专门用于分析Python模块的依赖关系，能准确识别本地模块和第三方库。安装后可通过API获取依赖文件列表：

  pip install modulegraph
  

  示例代码：

  import os
  from modulegraph import ModuleGraph
  
  mg = ModuleGraph()
  mg.add_script('server.py')
  # 筛选出当前项目目录下的本地依赖文件
  local_files = [
      m.filename for m in mg.flatten() 
      if m.filename and os.path.exists(m.filename) 
      and os.path.abspath(m.filename).startswith(os.getcwd())
  ]
  

• pydeps：生成依赖可视化图谱的同时，可导出依赖文件清单。安装后运行：

  pip install pydeps
  pydeps server.py --output-file=deps.json
  

  从生成的deps.json中提取本地模块的文件路径即可。
• PyInstaller依赖分析：虽主打打包可执行文件，但其内部依赖分析逻辑可复用。运行pyinstaller server.py后，查看build/server/server.spec文件，其中包含所有依赖的本地文件列表。

二、手动编程实现（基于AST解析） #

如果需要自定义依赖分析逻辑，可使用Python的ast模块递归解析导入语句，收集必要的本地文件。以下是实用脚本示例：

import ast
import os
from pathlib import Path

def get_local_dependencies(file_path, project_root, visited=None):
    if visited is None:
        visited = set()
    file_path = Path(file_path).resolve()
    if file_path in visited:
        return visited
    visited.add(file_path)
    
    # 解析文件的AST语法树
    with open(file_path, 'r', encoding='utf-8') as f:
        tree = ast.parse(f.read())
    
    # 遍历所有导入语句
    for node in ast.walk(tree):
        # 处理import ... 语句
        if isinstance(node, ast.Import):
            for alias in node.names:
                module_name = alias.name.split('.')[0]
                module_path = find_local_module(module_name, project_root)
                if module_path:
                    get_local_dependencies(module_path, project_root, visited)
        # 处理from ... import ... 语句
        elif isinstance(node, ast.ImportFrom):
            # 处理绝对导入
            if node.module and not node.level:
                module_name = node.module.split('.')[0]
                module_path = find_local_module(module_name, project_root)
                if module_path:
                    get_local_dependencies(module_path, project_root, visited)
            # 处理相对导入
            elif node.level > 0:
                parent_dir = file_path.parent
                # 向上遍历对应层级的目录
                for _ in range(node.level):
                    parent_dir = parent_dir.parent
                # 拼接模块路径
                rel_module_parts = node.module.split('.') if node.module else []
                rel_module_path = parent_dir.joinpath(*rel_module_parts)
                # 检查是包还是单个模块文件
                package_init = rel_module_path / '__init__.py'
                module_file = rel_module_path.with_suffix('.py')
                if package_init.exists():
                    get_local_dependencies(package_init, project_root, visited)
                elif module_file.exists():
                    get_local_dependencies(module_file, project_root, visited)
    return visited

def find_local_module(module_name, project_root):
    project_root = Path(project_root).resolve()
    # 检查是否为包（含__init__.py的目录）
    package_path = project_root / module_name
    if package_path.is_dir() and (package_path / '__init__.py').exists():
        return package_path / '__init__.py'
    # 检查是否为单个.py文件
    module_file = project_root / f"{module_name}.py"
    if module_file.is_file():
        return module_file
    return None

if __name__ == '__main__':
    # 配置参数
    entry_script = 'server.py'
    project_root_dir = '.'
    target_context_dir = './docker_context'

    # 获取所有依赖的本地文件
    required_files = get_local_dependencies(entry_script, project_root_dir)
    
    # 复制文件到Docker构建上下文目录
    os.makedirs(target_context_dir, exist_ok=True)
    for file in required_files:
        relative_path = file.relative_to(project_root_dir)
        target_file = Path(target_context_dir) / relative_path
        os.makedirs(target_file.parent, exist_ok=True)
        # 二进制复制保留文件权限
        with open(file, 'rb') as src_f, open(target_file, 'wb') as dst_f:
            dst_f.write(src_f.read())
    
    print(f"已将{len(required_files)}个必要文件复制到{target_context_dir}")

三、Docker镜像构建 #

基于生成的docker_context目录，编写轻量化Dockerfile：

# 使用最小的Python基础镜像
FROM python:3.11-slim

# 设置工作目录
WORKDIR /app

# 复制必要的代码文件
COPY docker_context/ .

# 安装仅运行所需的第三方库
RUN pip install --no-cache-dir flask

# 暴露服务端口
EXPOSE 5000

# 启动Flask服务
CMD ["python", "server.py"]

该构建方式仅包含server.py及其递归依赖的本地模块，完全剔除不需要的文件（如train_model.py），实现镜像最大化轻量化。

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