在上一篇文章中我们讲了如何在linux上用python写一个守护进程。主要原理是利用linux的fork函数来创建一个进程然后退出父进程运行内页排名然后退出父进程运行生成的子进程就会成为一个守护进程。细心观察的可能会发现这个守护进程的运行身份是执行这个程序的用户如果把这个守护程序加入到系统的服务项那么这个守护程序的执行身份应该是root。
一个情况出现了root的权限比较大如果通过这个root身份的守护程序来进行操作危险性是比较大的。一种好的办法是生成一个身份为root的master进程用来接受请求生成若干个woker进程用来处理请求这样就不会出现权限过大问题。事实上现在很多软件nginxmysqlapachevsftpd等几乎都是这样做的。
#!/usr/bin/env pythonimport time,os,pwd,sys,signallogfile="/tmp/d2.log"#step one, get the username you want to running withtry: user=sys.argv[1]except: user=raw_input('Please input a username in this machine you want to run this program: ')if user=="":sys.exit(-1)try: uid=pwd.getpwnam(user) uid=uid.pw_uidexcept: print "Uer not exists!" sys.exit(-1)#step two:Generation of daemonpid=os.fork()if(pid):sys.exit(0)os.setsid()os.chdir("http://www.qhdseo.net/")os.umask(0)#step three :fork againpid=os.fork()if(pid==0): os.setuid(uid) os.setsid() os.chdir("http://www.qhdseo.net/") os.umask(0)log=open(logfile,'a')log.write('Daemon start up at %s\n'%(time.strftime('%Y:%m:%d',time.localtime(time.time()))))log.close()def reload(a,b): log=open(logfile,'a') log.write('Daemon reload at %s\n'%(time.strftime('%Y:%m:%d',time.localtime(time.time())))) log.close()while True: signal.signal(signal.SIGHUP,reload) time.sleep(2)
运行这个程序输入nginx（nginx为系统中已经添加的用户）然后用ps aux|grep python查看系统当中运行的python程序可以看到有一个woker进程的身份已经改变了成为nginx了：
