如何编写Python程序识别字符串中的Bash或Python代码？

核心思路 #

检测字符串中的代码，本质是判断片段是否符合对应语言的语法规则，或识别特征性语法元素。通常有两种方向：基于语法解析（精准验证）和基于特征匹配（快速筛选）。

一、检测Python代码 #

1. 用Python内置ast模块（精准验证） #

Python的ast模块可直接解析代码字符串，若解析无报错，说明是合法的Python代码片段，这是最可靠的方法。

示例代码：

import ast

def is_valid_python(code_str):
    try:
        ast.parse(code_str)
        return True
    except SyntaxError:
        return False

# 测试用例
test_python = "def add(a, b): return a + b"
test_non_python = "echo 'hello' # 这是Bash"
print(is_valid_python(test_python))  # 输出True
print(is_valid_python(test_non_python))  # 输出False

注意：ast.parse要求代码是完整语法单元（如单个语句、完整函数），不完整片段（如x = 5 +）会被判定无效，符合“有效代码”的定义。

2. 基于特征匹配（快速筛选） #

若无需严格验证语法，可通过检查Python特有关键词（def、class、import）、语法符号（:、=、()）快速判断。但这种方法存在误判可能，仅适合初步过滤。

二、检测Bash代码 #

Bash语法灵活性高，无官方Python解析库，可通过以下两种方式实现：

1. 调用Bash语法检查（精准验证） #

利用Bash的-n参数（不执行脚本，仅验证语法），通过subprocess调用系统Bash完成检查。

示例代码：

import subprocess
import tempfile
import os

def is_valid_bash(code_str):
    # 创建临时脚本文件
    with tempfile.NamedTemporaryFile(mode='w', suffix='.sh', delete=False) as f:
        f.write(code_str)
        temp_path = f.name
    
    try:
        # 执行语法检查
        result = subprocess.run(
            ['bash', '-n', temp_path],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True
        )
        # 退出码为0表示语法有效
        return result.returncode == 0
    finally:
        # 清理临时文件
        os.unlink(temp_path)

# 测试用例
test_bash = "if [ -f test.txt ]; then echo 'exists'; fi"
test_non_bash = "print('hello') # Python代码"
print(is_valid_bash(test_bash))  # 输出True
print(is_valid_bash(test_non_bash))  # 输出False

注意：该方法依赖系统安装的Bash，对不完整的Bash片段（如if [ -f file ]缺fi）会返回False，符合语法有效性要求。

2. 基于特征匹配 #

检查Bash特有的语法元素：$变量、echo、ls、管道|、重定向>/<、if []; then结构等。适合快速初步判断，但误判率较高。

三、综合检测：判断字符串是否包含代码片段 #

若需检测字符串中包含代码（而非整个字符串都是代码），可先提取可能的代码片段（如按换行、代码块标记分割），再逐一验证。

示例代码：

def contains_valid_code(s):
    segments = [seg.strip() for seg in s.split('\n') if seg.strip()]
    for seg in segments:
        if is_valid_python(seg) or is_valid_bash(seg):
            return True
    return False

# 测试用例
mixed_text = "这是一段普通文本，里面有Python代码：print('hi')，还有Bash代码：ls -l"
print(contains_valid_code(mixed_text))  # 输出True

注意事项 #

• 语法解析精准但速度稍慢，适合需严格验证的场景；特征匹配快速但易误判，适合初步筛选。
• 混合文本场景下，需先提取可能的代码片段再验证，否则整段文本会因非代码内容导致解析失败。
• Bash语法检查依赖系统环境，需确保目标机器已安装Bash。

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