如何将CSV表格转换为Python中的简单if-else代码块？

核心思路 #

通过Python读取CSV中的属性与条件映射关系，动态拼接生成包含多分支if/elif判断的函数代码，并写入.py文件，无需手动编写重复的判断逻辑。

步骤说明 #

1. CSV格式要求
   确保CSV列标题固定：第一列为Attribute（属性），后续为Condition1至Condition5（最多5个条件列），每行对应一个属性的各条件取值（空值会自动跳过）。示例CSV内容：

   Attribute,Condition1,Condition2,Condition3
   UserTypeA,Admin,Editor,Viewer
   UserTypeB,Guest,Limited,
   

2. 生成脚本代码
   以下脚本会读取CSV并自动生成目标函数文件：

   import csv
   
   def generate_condition_function(csv_path, output_py_path):
       # 读取CSV数据
       with open(csv_path, 'r', encoding='utf-8') as f:
           reader = csv.DictReader(f)
           # 提取所有条件列（Condition1~Condition5）
           condition_cols = [col for col in reader.fieldnames if col.startswith('Condition')]
           
           # 初始化函数代码行
           code = [
               "def myFunction(Attribute, Condition):",
               "    \"\"\"根据属性与条件组合执行判断的函数\"\"\"",
               "    # 多分支条件判断",
           ]
           
           is_first_branch = True
           # 遍历每行数据生成判断分支
           for row in reader:
               attr = row['Attribute'].replace("'", "\\'")  # 转义单引号避免语法错误
               for cond_col in condition_cols:
                   cond_val = row[cond_col].replace("'", "\\'")
                   if not cond_val:
                       continue
                   
                   # 第一个分支用if，其余用elif
                   if is_first_branch:
                       code.append(f"    if Attribute == '{attr}' and Condition == '{cond_val}':")
                       is_first_branch = False
                   else:
                       code.append(f"    elif Attribute == '{attr}' and Condition == '{cond_val}':")
                   # 添加业务逻辑占位（用户需替换为实际代码）
                   code.append("        # 此处编写对应逻辑，如返回权限等级、执行操作")
                   code.append("        pass")
           
           # 添加默认分支处理未匹配情况
           code.append("    else:")
           code.append("        raise ValueError(f'未找到属性[{Attribute}]与条件[{Condition}]的匹配组合')")
           
           # 将代码写入目标.py文件
           with open(output_py_path, 'w', encoding='utf-8') as py_file:
               py_file.write('\n'.join(code))
   
   # 执行生成（替换为你的CSV路径和输出文件路径）
   if __name__ == "__main__":
       generate_condition_function("conditions.csv", "permission_check.py")
   

3. 使用生成的函数
   运行脚本后，会生成如permission_check.py的文件，直接导入即可使用：

   from permission_check import myFunction
   
   myFunction("UserTypeA", "Admin")
   

关键细节 #

• 自动处理单引号转义，避免生成的代码出现语法错误。
• 跳过CSV中空的条件值，不会生成无效分支。
• 默认分支抛出异常，可根据需求改为返回默认值或其他逻辑。

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