发个在linux下用python写守护进程的好东东（转）

最新推荐文章于 2024-08-07 17:42:24 发布

feneyChan

最新推荐文章于 2024-08-07 17:42:24 发布

阅读量801

点赞数

文章标签： python linux action file exception url

一个别人写的模块，可以直接用来把一个普通进程改为守护进程。
并且自动把标准输出重定向到日志文件，很实用啊。

view plain copy to clipboard print ?

'''''
This module is used to fork the current process into a daemon.
Almost none of this is necessary (or advisable) if your daemon
is being started by inetd. In that case, stdin, stdout and stderr are
all set up for you to refer to the network connection, and the fork()s
and session manipulation should not be done (to avoid confusing inetd).
Only the chdir() and umask() steps remain as useful.
References:
UNIX Programming FAQ
1.7 How do I get my program to act like a daemon?
[url]http://www.erlenstar.demon.co.uk/unix/faq_2.html#SEC16[/url]
Advanced Programming in the Unix Environment
W. Richard Stevens, 1992, Addison-Wesley, ISBN 0-201-56317-7.
History:
2001/07/10 by Juergen Hermann
2002/08/28 by Noah Spurrier
2003/02/24 by Clark Evans
2003/11/01 by Irmen de Jong --- adapted a bit for Snakelets
(raises exception at certain points)
[url]http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/66012[/url]
'''
import sys, os, time
from signal import SIGINT,SIGTERM,SIGKILL
def daemonize(stdout='/dev/null', stderr=None, stdin='/dev/null',
pidfile=None, startmsg = 'started with pid %s' ):
'''''
This forks the current process into a daemon.
The stdin, stdout, and stderr arguments are file names that
will be opened and be used to replace the standard file descriptors
in sys.stdin, sys.stdout, and sys.stderr.
These arguments are optional and default to /dev/null.
Note that stderr is opened unbuffered, so
if it shares a file with stdout then interleaved output
may not appear in the order that you expect.
'''
# flush io
sys.stdout.flush()
sys.stderr.flush()
# Do first fork.
try:
pid = os.fork()
if pid > 0: sys.exit(0) # Exit first parent.
except OSError, e:
sys.stderr.write("fork #1 failed: (%d) %s/n" % (e.errno, e.strerror))
sys.exit(1)
# Decouple from parent environment.
os.chdir("/")
os.umask(0)
os.setsid()
# Do second fork.
try:
pid = os.fork()
if pid > 0: sys.exit(0) # Exit second parent.
except OSError, e:
sys.stderr.write("fork #2 failed: (%d) %s/n" % (e.errno, e.strerror))
sys.exit(1)
# Open file descriptors and print start message
if not stderr: stderr = stdout
si = file(stdin, 'r')
so = file(stdout, 'a+')
se = file(stderr, 'a+', 0) #unbuffered
pid = str(os.getpid())
sys.stderr.write("/n%s/n" % startmsg % pid)
sys.stderr.flush()
if pidfile: file(pidfile,'w+').write("%s/n" % pid)
# Redirect standard file descriptors.
os.dup2(si.fileno(), sys.stdin.fileno())
os.dup2(so.fileno(), sys.stdout.fileno())
os.dup2(se.fileno(), sys.stderr.fileno())
class DaemonizeError(Exception): pass
def startstop(stdout='/dev/null', stderr=None, stdin='/dev/null',
pidfile='pid.txt', startmsg = 'started with pid %s', action=None ):
if not action and len(sys.argv) > 1:
action = sys.argv[1]
if action:
try:
pf = file(pidfile,'r')
pid = int(pf.read().strip())
pf.close()
except IOError:
pid = None
if 'stop' == action or 'restart' == action:
if not pid:
mess = "Could not stop, pid file '%s' missing./n"
raise DaemonizeError(mess % pidfile)
try:
while 1:
print "sending SIGINT to",pid
os.kill(pid,SIGINT)
time.sleep(2)
print "sending SIGTERM to",pid
os.kill(pid,SIGTERM)
time.sleep(2)
print "sending SIGKILL to",pid
os.kill(pid,SIGKILL)
time.sleep(1)
except OSError, err:
print "process has been terminated."
os.remove(pidfile)
if 'stop' == action:
return ## sys.exit(0)
action = 'start'
pid = None
if 'start' == action:
if pid:
mess = "Start aborted since pid file '%s' exists. Server still running?/n"
raise DaemonizeError(mess % pidfile)
daemonize(stdout,stderr,stdin,pidfile,startmsg)
return
print "usage: %s start|stop|restart" % sys.argv[0]
raise DaemonizeError("invalid command")
def test():
'''''
This is an example main function run by the daemon.
This prints a count and timestamp once per second.
'''
sys.stdout.write ('Message to stdout...')
sys.stderr.write ('Message to stderr...')
c = 0
while 1:
sys.stdout.write ('%d: %s/n' % (c, time.ctime(time.time())) )
sys.stdout.flush()
c = c + 1
time.sleep(1)
if __name__ == "__main__":
startstop(stdout='/tmp/daemonize.log',
pidfile='/tmp/daemonize.pid')
if sys.argv[1]in ('start', 'restart'):
test()

feneyChan

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
发个在linux下用python写守护进程的好东东（转）

一个别人写的模块，可以直接用来把一个普通进程改为守护进程。并且自动把标准输出重定向到日志文件，很实用啊。view plaincopy to clipboardprint? This module is used to fork the current process into a daemon. Almost none of
复制链接

扫一扫