【发布时间】:2017-03-23 19:04:45
【问题描述】:
我知道这个问题已经被问过很多次了,但是我的问题是不同的,因为我使用的是第三方代码并且不能修改太多。 我有一个在代码中多次调用的函数来创建一个子进程,将数据写入标准输入然后读取。它只是挂在这一行
line = self.classifier.stderr.readline()
使用 popen.communicate 确实解决了这个问题,但是由于 func2(classifier, vectors) 需要多次调用,所以会引发异常
subprocess I/O operation on closed file
有没有办法执行非阻塞读取操作?
def func1 (extcmd):
cmd=extcmd
classifier = subprocess.Popen(self.classifier_cmd, shell = True, stdin = subprocess.PIPE, stderr = subprocess.PIPE)
if self.classifier.poll():
raise OSError('Could not create classifier subprocess')
return classifier
def func2(classifier, vectors):
classifier.stdin.write('\n'.join(vectors) + "\n\n")
lines = []
line = self.classifier.stderr.readline()
print("not reaching")
while (line.strip() != ''):
# print line
lines.append(line)
line = self.classifier.stderr.readline()
if __name__ == '__main__':
extcmd="some external shell script"
vectors="some results"
classifier=func1(extcmd)
func2(classifier, vectors)
修改代码以添加更多细节
import subprocess
import paths
import os.path
class CRFClassifier:
def __init__(self, name, model_type, model_path, model_file, verbose):
self.verbose = verbose
self.name = name
self.type = model_type
self.model_fname = model_file
self.model_path = model_path
if not os.path.exists(os.path.join(self.model_path, self.model_fname)):
print 'The model path %s for CRF classifier %s does not exist.' % (os.path.join(self.model_path, self.model_fname), name)
raise OSError('Could not create classifier subprocess')
self.classifier_cmd = '%s/crfsuite-stdin tag -pi -m %s -' % (paths.CRFSUITE_PATH,
os.path.join(self.model_path, self.model_fname))
# print self.classifier_cmd
self.classifier = subprocess.Popen(self.classifier_cmd, shell = True, stdin = subprocess.PIPE, stderr = subprocess.PIPE)
if self.classifier.poll():
raise OSError('Could not create classifier subprocess, with error info:\n%s' % self.classifier.stderr.readline())
#self.cnt = 0
def classify(self, vectors):
# print '\n'.join(vectors) + "\n\n"
self.classifier.stdin.write('\n'.join(vectors) + "\n\n")
lines = []
line = self.classifier.stderr.readline()
while (line.strip() != ''):
# print line
lines.append(line)
line = self.classifier.stderr.readline()
if self.classifier.poll():
raise OSError('crf_classifier subprocess died')
predictions = []
for line in lines[1 : ]:
line = line.strip()
# print line
if line != '':
fields = line.split(':')
# print fields
label = fields[0]
prob = float(fields[1])
predictions.append((label, prob))
seq_prob = float(lines[0].split('\t')[1])
return seq_prob, predictions
def poll(self):
"""
Checks that the classifier processes are still alive
"""
if self.classifier is None:
return True
else:
return self.classifier.poll() != None
为输入文件创建分类器对象,该文件是包含句子列表的文档,并且在创建时它还使用此句子列表执行外部命令。然后在一个单独的函数中处理每个句子,为每个句子提供一个单独的向量。这个新向量被传递给分类函数。
def func2():
classifier=create a classifier object for an input file, this executes the external command
for sentence in sentences:
vectors=process(sentence)# some external function
classifier.classify(features)
【问题讨论】:
-
一个不要脸的插件:你可能会发现github.com/TeamHG-Memex/sklearn-crfsuite很有用;它应该比包装 crfsuite 命令行更容易。
标签: python subprocess deadlock popen