web-dev-qa-db-fra.com

Exécutez Bash interactif avec popen et un ATS dédié Python

J'ai besoin d'exécuter une instance Bash interactive dans un processus séparé en Python avec son propre ATS dédié (je ne peux pas utiliser pexpect). J'ai utilisé cet extrait de code que je vois couramment utilisé dans des programmes similaires:

master, slave = pty.openpty()

p = subprocess.Popen(["/bin/bash", "-i"], stdin=slave, stdout=slave, stderr=slave)

os.close(slave)

x = os.read(master, 1026)

print x

subprocess.Popen.kill(p)
os.close(master)

Mais quand je l'exécute, j'obtiens la sortie suivante:

$ ./pty_try.py
bash: cannot set terminal process group (10790): Inappropriate ioctl for device
bash: no job control in this Shell

Strace de la course montre quelques erreurs:

...
readlink("/usr/bin/python2.7", 0x7ffc8db02510, 4096) = -1 EINVAL (Invalid argument)
...
ioctl(3, SNDCTL_TMR_TIMEBASE or SNDRV_TIMER_IOCTL_NEXT_DEVICE or TCGETS, 0x7ffc8db03590) = -1 ENOTTY (Inappropriate ioctl for device)
...
readlink("./pty_try.py", 0x7ffc8db00610, 4096) = -1 EINVAL (Invalid argument)

L'extrait de code semble assez simple, Bash n'obtient-il pas quelque chose dont il a besoin? Quel pourrait être le problème ici?

pythonlinuxbashpython-2.xtty
15
TKKS

C'est la solution qui a fonctionné pour moi à la fin (comme suggéré par qarma):

libc = ctypes.CDLL('libc.so.6')

master, slave = pty.openpty()
p = subprocess.Popen(["/bin/bash", "-i"], preexec_fn=libc.setsid, stdin=slave, stdout=slave, stderr=slave)
os.close(slave)

... do stuff here ...

x = os.read(master, 1026)
print x
2
TKKS

Il s'agit d'une solution pour exécuter une commande interactive dans un sous-processus. Il utilise un pseudo-terminal pour rendre stdout non bloquant (certaines commandes nécessitent également un périphérique tty, par exemple bash). il utilise select pour gérer l'entrée et la sortie du sous-processus.

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import sys
import select
import termios
import tty
import pty
from subprocess import Popen

command = 'bash'
# command = 'docker run -it --rm centos /bin/bash'.split()

# save original tty setting then set it to raw mode
old_tty = termios.tcgetattr(sys.stdin)
tty.setraw(sys.stdin.fileno())

# open pseudo-terminal to interact with subprocess
master_fd, slave_fd = pty.openpty()

# use os.setsid() make it run in a new process group, or bash job control will not be enabled
p = Popen(command,
          preexec_fn=os.setsid,
          stdin=slave_fd,
          stdout=slave_fd,
          stderr=slave_fd,
          universal_newlines=True)

while p.poll() is None:
    r, w, e = select.select([sys.stdin, master_fd], [], [])
    if sys.stdin in r:
        d = os.read(sys.stdin.fileno(), 10240)
        os.write(master_fd, d)
    Elif master_fd in r:
        o = os.read(master_fd, 10240)
        if o:
            os.write(sys.stdout.fileno(), o)

# restore tty settings back
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_tty)
14
Liao