Jak uniknąć wywołań os.system ()?

Podczas korzystania z os.system () często konieczne jest uniknięcie nazw plików i innych argumentów przekazywanych jako parametry do poleceń. W jaki sposób mogę to zrobić? Najlepiej coś, co działałoby na wielu systemach operacyjnych / powłokach, ale w szczególności na bash.

Obecnie wykonuję następujące czynności, ale jestem pewien, że musi to być funkcja biblioteczna lub przynajmniej bardziej elegancka / solidna / wydajna opcja:

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)))

Edycja: zaakceptowałem prostą odpowiedź, używając cytatów, nie wiem, dlaczego o tym nie pomyślałem; Chyba dlatego, że pochodzę z systemu Windows, gdzie „i” zachowują się trochę inaczej.

Jeśli chodzi o bezpieczeństwo, rozumiem obawy, ale w tym przypadku interesuje mnie szybkie i łatwe rozwiązanie, które zapewnia os.system (), a źródło ciągów nie jest generowane przez użytkownika lub przynajmniej wprowadzane przez zaufany użytkownik (ja).

Oto, czego używam:

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

Powłoka zawsze akceptuje nazwę pliku w cudzysłowie i usuwa otaczające ją cudzysłowy przed przekazaniem jej do danego programu. Warto zauważyć, że pozwala to uniknąć problemów z nazwami plików, które zawierają spacje lub inne nieprzyjemne metaznaky powłoki.

Aktualizacja : Jeśli korzystasz z Pythona 3.3 lub nowszego, użyj shlex.quote zamiast tworzyć własne.

shlex.quote() robi, co chcesz, od Pythona 3.

(Służy pipes.quotedo obsługi języka Python 2 i Python 3)

Być może masz konkretny powód, dla którego używasz os.system(). Ale jeśli nie, prawdopodobnie powinieneś używać subprocessmodułu . Możesz określić rury bezpośrednio i uniknąć używania powłoki.

Poniższy tekst pochodzi z 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]

Może subprocess.list2cmdlinejest lepszy strzał?

Zauważ, że pipe.quote jest faktycznie uszkodzony w Pythonie 2.5 i Pythonie 3.1 i nie jest bezpieczny w użyciu - nie obsługuje argumentów o zerowej długości.

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

Zobacz wydanie Pythona 7476 ; został naprawiony w Pythonie 2.6 i 3.2 i nowszych.

Uwaga : to jest odpowiedź dla Pythona 2.7.x.

Według źródła , pipes.quote()jest to sposób na " Rzetelne cytowanie łańcucha jako pojedynczego argumentu dla / bin / sh ". (Chociaż jest przestarzały od wersji 2.7 i ostatecznie ujawniony publicznie w Pythonie 3.3 jakoshlex.quote() funkcja).

Na drugiej strony , subprocess.list2cmdline()jest to sposób na „ Translate sekwencję argumentów na ciąg wiersza poleceń, przy użyciu tych samych zasad jak w czasie wykonywania MS C ”.

Oto jesteśmy, niezależny od platformy sposób cytowania ciągów znaków w wierszach poleceń.

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)

Stosowanie:

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

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

Uważam, że os.system po prostu wywołuje dowolną powłokę poleceń skonfigurowaną dla użytkownika, więc nie sądzę, aby można było to zrobić w sposób niezależny od platformy. Moją powłoką poleceń może być wszystko, od bash, emacs, ruby, a nawet quake3. Niektóre z tych programów nie oczekują argumentów, które im przekazujesz, a nawet jeśli to zrobiły, nie ma gwarancji, że uciekną w ten sam sposób.

Funkcja, której używam, to:

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

to znaczy: zawsze umieszczam argument w cudzysłowach, a jedyne znaki specjalne w cudzysłowach umieszczam w odwrotnym ukośniku.

Jeśli użyjesz polecenia systemowego, spróbuję umieścić na białej liście to, co wchodzi do wywołania os.system () .. Na przykład ..

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

Moduł podprocesu jest lepszą opcją i zalecałbym unikanie używania czegoś takiego jak os.system / subprocess, gdy tylko jest to możliwe.

Prawdziwa odpowiedź brzmi: nie używaj os.system()w pierwszej kolejności. Użyj subprocess.callzamiast tego i podaj argumenty bez zmiany znaczenia.