将参数从一个python脚本传递到另一个

时间:2016-01-25 01:55:41

标签: python python-2.7 loops subprocess

我共有3个文件:clean.txt,origin.py和pump.py

Clean.txt中有一些行(实际上是网站链接。例如:

www.link1.com
www.link2.com
www.link3.com
www.link4.com

origin.py是逐个读取行的脚本,我希望这个脚本将链接(一次一个)发送到pump.py

pump.py是一个脚本,它要求我输入一个链接。

现在,我想这样做,以便我从clean.txt读取行(origin.py正在执行此任务)并一次将它们发送到pump.py。像一个循环。 Origin.py的代码是:

fo = open("Clean.txt", "r")
nremoval = str(fo.readlines()).replace('\\n','')
      for lines in fo :
          if __name__ =="__main__":
             response = nremoval
             p = subprocess.Popen("pump.py", stdin = subprocess.PIPE)
             time.sleep(2) 
             p.stdin.write(bytes(response, 'ascii'))
             print("injected string :", response)

1 个答案:

答案 0 :(得分:2)

origin.py

import subprocess
import time

# open file 
with open("Clean.txt", "r") as fo:

     # read line by line
     for line in fo:

         # remove ENTER and SPACES from single line 
         line = line.strip() 

         # call "python pump.py" 
         p = subprocess.Popen(["python","pump.py"], stdin=subprocess.PIPE)

         # or "python3 pump.py"
         #p = subprocess.Popen(["python3","pump.py"], stdin=subprocess.PIPE)

         #time.sleep(2) 

         # send line to standard input
         #p.stdin.write(bytes(line, 'ascii')) # python 3
         p.stdin.write(line) # python 2

         print("injected string:", line)

pump.py

#line = input() # python 3
line = raw_input() # python 2

print("received:", line)