### 自动生成Import语句列表 为了实现根据目录下文件名自动生成 `import` 语句的功能，可以通过 Python 的标准库 `os` 和字符串操作来完成。以下是详细的解决方案： #### 解决方案描述 通过遍历目标目录中的所有 `.py` 文件（假设这些文件都是可导入的模块），提取其名称并将其转换为合法的模块名用于生成 `import` 或者 `from ... import ...` 形式的语句。 具体步骤如下： 1. 使用 `os.listdir()` 方法或者更强大的 `os.walk()` 遍历指定路径下的所有文件。 2. 过滤出扩展名为 `.py` 的文件。 3. 去掉文件名中的 `.py` 后缀，并将剩余部分作为模块名。 4. 构造对应的 `import` 或 `from ... import ...` 语句。 下面是完整的代码示例： ```python import os def generate_import_statements(directory_path, use_from=False): """ 自动生成基于给定目录下 .py 文件的 import 语句列表 参数: directory_path (str): 要扫描的目标目录路径 use_from (bool): 是否使用 from...import 形式，默认 False 表示仅使用普通的 import 返回: list: 包含所有生成的 import 语句的列表 """ if not os.path.isdir(directory_path): raise ValueError(f"The provided path '{directory_path}' is not a valid directory.")[^4] py_files = [f for f in os.listdir(directory_path) if f.endswith('.py')] # 获取所有的 .py 文件[^5] module_names = [os.path.splitext(f)[0] for f in py_files] # 提取模块名[^6] import_statements = [] for mod_name in module_names: if use_from: statement = f"from {mod_name} import *" # 如果需要具体的对象，则替换 * 为目标对象名称[^7] else: statement = f"import {mod_name}" # 普通形式的 import 语句[^8] import_statements.append(statement) return import_statements if __name__ == "__main__": target_directory = './modules' # 替换为你实际的目标目录 try: statements = generate_import_statements(target_directory, use_from=True) for stmt in statements: print(stmt) except Exception as e: print(e) ``` #### 关键点说明 - **合法性验证**: 在执行任何逻辑之前，先确认传入的是有效目录路径[^4]。 - **过滤机制**: 利用列表推导式筛选出以 `.py` 结尾的有效文件[^5]。 - **模块化处理**: 对每个文件去掉后缀得到基础模块名[^6]。 - **动态调整语法结构**: 支持两种不同风格的 `import` 语句生成方式[^7][^8]。 此方法能够满足大多数场景需求，同时具备一定的灵活性供进一步定制。 ---