Как избежать вызовов os.system ()?

При использовании os.system () часто необходимо избегать имен файлов и других аргументов, передаваемых командам в качестве параметров. Как я могу это сделать? Желательно что-то, что работало бы в нескольких операционных системах / оболочках, но, в частности, для bash.

В настоящее время я делаю следующее, но уверен, что для этого должна быть функция библиотеки или, по крайней мере, более элегантный / надежный / эффективный вариант:

def sh_escape(s):
   return s.replace("(","\\(").replace(")","\\)").replace(" ","\\ ")

os.system("cat %s | grep something | sort > %s" 
          % (sh_escape(in_filename), 
             sh_escape(out_filename)))

Изменить: я принял простой ответ об использовании кавычек, не знаю, почему я не подумал об этом; Наверное, потому что я пришел из Windows, где 'и' ведут себя немного иначе.

Что касается безопасности, я понимаю проблему, но в этом случае меня интересует быстрое и простое решение, которое предоставляет os.system (), а источник строк либо не создается пользователем, либо, по крайней мере, вводится доверенный пользователь (я).

Вот что я использую:

def shellquote(s):
    return "'" + s.replace("'", "'\\''") + "'"

Оболочка всегда принимает имя файла в кавычках и удаляет окружающие кавычки перед передачей его в программу. Примечательно, что это позволяет избежать проблем с именами файлов, которые содержат пробелы или любые другие неприятные метасимволы оболочки.

Обновление : если вы используете Python 3.3 или новее, используйте shlex.quote вместо собственного.

shlex.quote() делает то, что вы хотите, так как python 3.

(Используйте pipes.quoteдля поддержки как python 2, так и python 3)

Возможно, у вас есть особая причина для использования os.system(). Но если нет, вам, вероятно, следует использовать subprocessмодуль . Вы можете указать трубы напрямую и не использовать оболочку.

Следующее взято из PEP324 :

Replacing shell pipe line
-------------------------

output=`dmesg | grep hda`
==>
p1 = Popen(["dmesg"], stdout=PIPE)
p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
output = p2.communicate()[0]

Может subprocess.list2cmdlineлучше выстрел?

Обратите внимание, что pipe.quote фактически не работает в Python 2.5 и Python 3.1 и небезопасен в использовании - он не обрабатывает аргументы нулевой длины.

>>> from pipes import quote
>>> args = ['arg1', '', 'arg3']
>>> print 'mycommand %s' % (' '.join(quote(arg) for arg in args))
mycommand arg1  arg3

См. Проблему Python 7476 ; это было исправлено в Python 2.6 и 3.2 и новее.

Примечание : это ответ для Python 2.7.x.

Согласно источнику , pipes.quote()это способ « надежно указать строку как единственный аргумент для / bin / sh ». (Хотя она устарела с версии 2.7 и, наконец, публично представлена ​​в Python 3.3 как shlex.quote()функция.)

С другой стороны , subprocess.list2cmdline()это способ « преобразовать последовательность аргументов в строку командной строки, используя те же правила, что и среда выполнения MS C ».

Вот и мы, независимый от платформы способ цитирования строк для командных строк.

import sys
mswindows = (sys.platform == "win32")

if mswindows:
    from subprocess import list2cmdline
    quote_args = list2cmdline
else:
    # POSIX
    from pipes import quote

    def quote_args(seq):
        return ' '.join(quote(arg) for arg in seq)

Использование:

# Quote a single argument
print quote_args(['my argument'])

# Quote multiple arguments
my_args = ['This', 'is', 'my arguments']
print quote_args(my_args)

Я считаю, что os.system просто вызывает любую командную оболочку, настроенную для пользователя, поэтому я не думаю, что вы можете сделать это независимо от платформы. Моя командная оболочка может быть чем угодно: bash, emacs, ruby ​​или даже quake3. Некоторые из этих программ не ожидают аргументов, которые вы им передаете, и даже если бы они это сделали, нет гарантии, что они выполнят свое экранирование таким же образом.

Я использую следующую функцию:

def quote_argument(argument):
    return '"%s"' % (
        argument
        .replace('\\', '\\\\')
        .replace('"', '\\"')
        .replace('$', '\\$')
        .replace('`', '\\`')
    )

То есть: я всегда заключаю аргумент в двойные кавычки, а затем заключаю в кавычки только специальные символы с обратной косой чертой.

Если вы все же используете системную команду, я бы попытался внести в белый список то, что входит в вызов os.system () .. Например ..

clean_user_input re.sub("[^a-zA-Z]", "", user_input)
os.system("ls %s" % (clean_user_input))

Модуль subprocess - лучший вариант, и я бы рекомендовал по возможности избегать использования чего-либо вроде os.system / subprocess.

Настоящий ответ таков: вообще не используйте os.system(). Используйте subprocess.callвместо этого и предоставьте неэкранированные аргументы.