Взаимодействовать с python REPL из скрипта python - PullRequest
Новая
       37

Взаимодействовать с python REPL из скрипта python

0 голосов
/ 23 сентября 2018

Я хочу найти (или сделать) скрипт Python, который построчно читает другой скрипт Python и печатает выполненные команды и вывод сразу после него.

Предположим, у вас есть скрипт Python, testfile.py как таковой: 

print("Hello world")

for i in range(3):
    print(f"i is: {i}")

Теперь я хочу другой скрипт на python, который анализирует testfile.py и выводит следующее: 

print("Hello world")
## Hello world
for i in range(3):
    print(f"i is: {i}")
## i is: 0
## i is: 1
## i is: 2

Любые предложения по существующему программному обеспечению или новому коду накак этого добиться, очень ценится!

Попытки / код концепции:

Запуск ipython из python:

Одной из первых мыслей было запустить ipythonиз python с использованием subprocess: 

import subprocess
import re
try:
    proc = subprocess.Popen(args=["ipython", "-i"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, universal_newlines=True)
    # Delimiter to know when to stop reading
    OUTPUT_DELIMITER = ":::EOL:::"
    # Variable to contain the entire interaction:
    output_string = ""
    # Open testfile.py
    with open("testfile.py") as file_:
        for line in file_:
            # Read command
            cmd = line.rstrip()
            # Add the command to the output string
            output_string += cmd + "\n"
            proc.stdin.write(f"{cmd}\n")
            # Print the delimiter so we know when to end:
            proc.stdin.write('print("{}")\n'.format(OUTPUT_DELIMITER))
            proc.stdin.flush()
            # Start reading output from ipython
            while True:
                thisoutput = proc.stdout.readline()
                thisoutput = thisoutput.rstrip()
                # Now check if it's the delimiter
                if thisoutput.find(OUTPUT_DELIMITER) >= 0:
                    break
                output_string += thisoutput + "\n"

except Exception as e:
    proc.stdout.close()
    proc.stdin.close()
    raise
proc.stdout.close()
proc.stdin.close()
print("-" * 4 + "START OUTPUT" + "-" * 4)
print(output_string)
print("-" * 4 + "END OUTPUT" + "-" * 4)

. При таком подходе проблема превращается в блоки с отступом, как в цикле for.В идеале что-то вроде этого будет работать, используя просто python (а не ipython).

...