为csh引用字符串

时间:2013-03-07 23:59:49

标签: python csh tcsh quoting

就本问题而言,“csh”是指tcsh。

我知道避免编程的csh的标准建议。但是,有时需要与现有的csh代码进行交互,然后可能需要引用csh的字符串。换句话说,问题是如何用csh语法表示任意字节串。

以下csh_escape_arg函数是否正确?也就是说,是否存在一个字符串,如果它被添加到测试中的字符串列表中,会导致该测试失败吗?如果有这样的字符串,我如何修复我的函数,以便所有字符串都通过测试?

import string
import subprocess
import unittest

# Safe unquoted
_safechars = frozenset(string.ascii_letters + string.digits + '@%_-+:,./')

def csh_escape_arg(str_):
    """Return a representation of str_ in csh.

    Based on the standard library's pipes.quote
    """
    for c in str_:
        if c not in _safechars:
            break
    else:
        if not str_:
            return "''"
        return str_
    str_ = str_.replace("\\", "\\\\")
    str_ = str_.replace("\n", "\\\n")
    str_ = str_.replace("!", "\\!")
    # use single quotes, and put single quotes into double quotes
    # the string $'b is then quoted as '$'"'"'b'
    return "'" + str_.replace("'", "'\"'\"'") + "'"

def csh_escape(args):
    return " ".join(csh_escape_arg(arg) for arg in args)

def get_cmd_stdout(args, **kwargs):
    child = subprocess.Popen(args, stdout=subprocess.PIPE, **kwargs)
    stdout, stderr = child.communicate()
    rc = child.returncode
    if rc != 0:
        raise Exception("Command failed with return code %d: %s:\n%s" % (rc, args, stderr))
    else:
        return stdout

class TestCsh(unittest.TestCase):

    def test_hard_cases(self):
        for angry_string in [
            "\\!\n\"'`",
            "\\\\!\n\"'`",
            "=0",
            ]:
            out = get_cmd_stdout(["tcsh", "-c", csh_escape(["echo", "-n", angry_string])])
            self.assertEqual(out, angry_string)

unittest.main()

1 个答案:

答案 0 :(得分:2)

1)对于tcsh,您还需要引用“=”以防止目录堆栈替换。 2)我认为你的算法也会遇到带有不成对双引号的字符串的问题。 3)另一种方法是编写目标脚本,使字符串不受替换。例如,通过将字符串写入文件,然后让脚本将文件中的字符串读入变量,例如

set a = `cat file`

然后根据需要使用变量。