我有一个简单的Python脚本,它询问您的姓名,然后将其吐出:
def main():
print('Enter your name: ')
for line in sys.stdin:
print 'You entered: ' + line
很简单的东西!在OSX终端上运行时,效果很好:
$ python nameTest.py
Enter your name:
Craig^D
You entered: Craig
但是,当试图通过
NSTask
运行这个进程时,只有在Python脚本中添加了额外的flush()调用时,stdout才会出现。这就是我如何配置
NSTask
和管道的方法:NSTask *_currentTask = [[NSTask alloc] init];
_currentTask.launchPath = @"/usr/bin/python";
_currentTask.arguments = [NSArray arrayWithObject:@"nameTest.py"];
NSPipe *pipe = [[NSPipe alloc] init];
_currentTask.standardOutput = pipe;
_currentTask.standardError = pipe;
dispatch_queue_t stdout_queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0);
__block dispatch_block_t checkBlock;
checkBlock = ^{
NSData *readData = [[pipe fileHandleForReading] availableData];
NSString *consoleOutput = [[NSString alloc] initWithData:readData encoding:NSUTF8StringEncoding];
dispatch_sync(dispatch_get_main_queue(), ^{
[self.consoleView appendString:consoleOutput];
});
if ([_currentTask isRunning]) {
[NSThread sleepForTimeInterval:0.1];
checkBlock();
} else {
dispatch_sync(dispatch_get_main_queue(), ^{
NSData *readData = [[pipe fileHandleForReading] readDataToEndOfFile];
NSString *consoleOutput = [[NSString alloc] initWithData:readData encoding:NSUTF8StringEncoding];
[self.consoleView appendString:consoleOutput];
});
}
};
dispatch_async(stdout_queue, checkBlock);
[_currentTask launch];
但是当运行
NSTask
时,这就是它的显示方式(它最初是空白的,但是在输入我的名字并按CTRL+D之后,它会立即完成):Craig^DEnter your name:
You entered: Craig
所以,我的问题是:如何从
stdout
中读取NSTask
,而不需要Python脚本中的额外flush()语句?为什么在作为NSTask
运行时不立即显示Enter your name:提示符? 最佳答案
当Python看到它的标准输出是一个终端时,它安排在脚本从sys.stdout
读取时自动刷新sys.stdin
。使用NSTask
运行脚本时,脚本的标准输出是管道,而不是终端。
更新
有一个特定于Python的解决方案。您可以将-u
标志传递给Python解释器(例如_currentTask.arguments = @[ @"-u", @"nameTest.py"];
),它告诉Python不要缓冲标准输入、标准输出或标准错误。您也可以在流程的环境中设置PYTHONUNBUFFERED=1
,以达到相同的效果。
原件
适用于任何程序的更通用的解决方案是使用所谓的“伪终端”(或历史上的“伪电传打字机”),我们将其缩短为“pty”。(事实上,这就是终端应用程序本身的功能。这是一种罕见的Mac,它有一个连接到串行端口的物理终端或电传机!)
每个pty实际上是一对虚拟设备:从设备和主设备。你写给主站的字节,你可以从从站读取,反之亦然。所以这些设备更像是插座(双向的)而不是管道(单向的)。此外,pty还允许您设置终端I/O标志(或“termios”),这些标志控制从机是否回显其输入,一次传递一行还是一次传递一个字符,等等。
无论如何,您可以使用openpty
函数轻松打开主/从对。这里有一个小类别,可以用来让NSTask
对象使用从端作为任务的标准输入和输出。
NSTask+PTY.h公司
@interface NSTask (PTY)
- (NSFileHandle *)masterSideOfPTYOrError:(NSError **)error;
@end
NSTask+PTY.m公司
#import "NSTask+PTY.h"
#import <util.h>
@implementation NSTask (PTY)
- (NSFileHandle *)masterSideOfPTYOrError:(NSError *__autoreleasing *)error {
int fdMaster, fdSlave;
int rc = openpty(&fdMaster, &fdSlave, NULL, NULL, NULL);
if (rc != 0) {
if (error) {
*error = [NSError errorWithDomain:NSPOSIXErrorDomain code:errno userInfo:nil];
}
return NULL;
}
fcntl(fdMaster, F_SETFD, FD_CLOEXEC);
fcntl(fdSlave, F_SETFD, FD_CLOEXEC);
NSFileHandle *masterHandle = [[NSFileHandle alloc] initWithFileDescriptor:fdMaster closeOnDealloc:YES];
NSFileHandle *slaveHandle = [[NSFileHandle alloc] initWithFileDescriptor:fdSlave closeOnDealloc:YES];
self.standardInput = slaveHandle;
self.standardOutput = slaveHandle;
return masterHandle;
}
@end
你可以这样使用它:
NSTask *_currentTask = [[NSTask alloc] init];
_currentTask.launchPath = @"/usr/bin/python";
_currentTask.arguments = @[[[NSBundle mainBundle] pathForResource:@"nameTest" ofType:@"py"]];
NSError *error;
NSFileHandle *masterHandle = [_currentTask masterSideOfPTYOrError:&error];
if (!masterHandle) {
NSLog(@"error: could not set up PTY for task: %@", error);
return;
}
然后,您可以使用
masterHandle
从任务读取并写入任务。关于python - 从任务的标准输出中读取时,NSTask需要刷新,而终端则不需要,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/13355363/