Python：以编程方式向.py文件添加代码行的最佳方案

可直接复用的工具包 #

不需要自己写脆弱的正则匹配，有成熟的Python代码修改工具可以用：

• libcst：是目前做Python源码增量修改最适配的工具，解析代码时会完整保留原文件的缩进、空行、注释、排版格式，修改后写回的代码不会打乱原有风格。你可以精准定位到导入区块末尾、match语句的case列表末尾插入对应内容，不会出现正则匹配错位的问题。
• ast：Python标准库自带的语法解析模块，不需要额外安装依赖，缺点是解析时会丢失注释、格式信息，修改后导出的代码需要重新格式化，适合对代码版式无要求的场景。

比正则更优的实现路径 #

非常不建议用正则处理Python代码修改：Python语法格式灵活度高，只要原文件出现额外注释、换行、缩进调整，正则就容易匹配失效，甚至插错位置破坏原有代码。按优先级从高到低可选方案如下：

• 方案1：自动注册扫描（长期维护成本最低）
  既然需要高频新增模型类，完全可以去掉手动维护导入、case分支的逻辑，改成类自动注册发现，之后新增类不需要修改工厂文件的任何代码：
  1. 给所有模型类加一个固定的类属性标记对应的template_id
  2. 工厂类初始化时自动扫描models包下的所有模块，收集带template_id标记的类，构建id到类的映射字典
  3. create方法直接查字典返回对应实例即可
     参考实现：

  import importlib
  import pkgutil
  import models
  
  # 服务启动时自动扫描加载所有模型类
  _TEMPLATE_CLS_MAP = {}
  for _, module_name, _ in pkgutil.iter_modules(models.__path__):
      module = importlib.import_module(f"models.{module_name}")
      for attr in dir(module):
          cls = getattr(module, attr)
          if hasattr(cls, "template_id"):
              _TEMPLATE_CLS_MAP[cls.template_id.lower()] = cls
  
  class SomeFactory:
      @staticmethod
      def create(template_id: str, **kwargs):
          target_cls = _TEMPLATE_CLS_MAP.get(template_id.lower())
          if not target_cls:
              raise ValueError(f"不支持的模板ID: {template_id}")
          return target_cls(**kwargs)
  

  后续新增模型时，只需要在models目录下新建对应类，按规范写好template_id属性即可，比如新建models/foo.py：

  class FooClass:
      template_id = "foo"
      def __init__(self, **kwargs):
          # 自定义初始化逻辑
          pass
  

• 方案2：基于CST做语法树级修改
  如果一定要保留原有手动维护导入和case的代码结构，直接用libcst解析源码为具体语法树，遍历定位到两个插入点：
  1. 最后一行模型导入语句的下方，插入新的from models.xxx import XxxClass导入行
  2. match语句中最后一个case分支的下方，插入对应新template_id的case块
     修改完成后直接导出为源码写回文件即可，全程不需要做字符串匹配，不会破坏原文件格式。
• 方案3：字符串/正则修改
  属于容错率最低的方案，仅适合临时快速凑合用，长期用大概率会因为代码格式变动出问题。

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