程序员Python 运维寒哥管理的技术专题

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
上一篇下一篇

猜你喜欢

热点阅读