python笔记 os.dup2
2016-01-05 本文已影响3236人
Stansosleepy
最近看框架的源码,总是遇到os.dup2这个函数,不是很能理解它的用法,于是去看了python的源码,Dup2.c:
/*
* Public domain dup2() lookalike
* by Curtis Jackson @ AT&T Technologies, Burlington, NC
* electronic address: burl!rcj
*
* dup2 performs the following functions:
*
* Check to make sure that fd1 is a valid open file descriptor.
* Check to see if fd2 is already open; if so, close it.
* Duplicate fd1 onto fd2; checking to make sure fd2 is a valid fd.
* Return fd2 if all went well; return BADEXIT otherwise.
*/
#include <fcntl.h>
#include <unistd.h>
#define BADEXIT -1
int
dup2(int fd1, int fd2)
{
if (fd1 != fd2) {
if (fcntl(fd1, F_GETFL) < 0)
return BADEXIT;
if (fcntl(fd2, F_GETFL) >= 0)
close(fd2);
if (fcntl(fd1, F_DUPFD, fd2) < 0)
return BADEXIT;
}
return fd2;
}
从源码上看,dup2传入两个文件描述符,fd1和fd2(fd1是必须存在的),如果fd2存在,就关闭fd2,然后将fd1代表的那个文件(可以想象成是P_fd1指针)强行复制给fd2,fd2这个文件描述符不会发生变化,但是fd2指向的文件就变成了fd1指向的文件。
这个函数最大的作用是重定向,参考下面的python代码:
import os
#打开一个文件
f=open('txt','a')
#将这个文件描述符代表的文件,传递给1描述符指向的文件(也就是stdout)
os.dup2(f.fileno(),1)
f.close()
#print输出到标准输出流,就是文件描述符1
print 'line1'
print 'line2'
#脚本执行结果:
#生成一个txt文件,内容是:
#line1
#line2