python 进程单例，起停功能

最新推荐文章于 2023-04-11 18:45:22 发布

nehc

最新推荐文章于 2023-04-11 18:45:22 发布

阅读量2.2k

点赞数

分类专栏：杂谈 Python 文章标签： python

本文链接：https://blog.csdn.net/cenziboy/article/details/17370193

版权

杂谈同时被 2 个专栏收录

32 篇文章 0 订阅

订阅专栏

Python

7 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Python的文件锁实现进程单例，同时展示了如何通过参数来控制进程的启动和停止，类似于HTTP服务器的命令行控制。

摘要由CSDN通过智能技术生成

python文件锁实现进程单例

通过参数控制进程起停，类似 httpd -k <start | stop>

#!/usr/bin/python
import os
import getopt
import fcntl
import time
import sys

def main():
    while(True):
        print os.getpid()
        time.sleep(3)

def print_usage():
    print 'usage'

if __name__ == "__main__":
    pid_file = '/opt/nsfocus/mp-wafsl/tmp/daemon.pid'
    optarg = 'null'
    try:
        opts, args = getopt.getopt(sys.argv[1:], "hk:", ['help', 'kk='])
        for name,value in opts:
            if name in ('-h'):
                print_usage()
                sys.exit(0)
            if name in ('-k'):
                optarg = value
    except getopt.GetoptError:
        print_usage()
        sys.exit(2)
    # Must use a+ here
    fp = open(pid_file, 'a+')
    is_running = False
    try:
        fcntl.lockf(fp, fcntl.LOCK_EX | fcntl.LOCK_NB)
    except IOError:
        is_running = True
    
    if is_running and optarg == 'stop':
        # stop running process
        content = fp.read()
        fp.close()
        pid = int(content)
        os.kill(pid, 9)
        sys.exit(0)
    elif optarg == 'start' and is_running is False:
        fp.truncate()
        fp.write(str(os.getpid()))
        fp.flush()
        main()
    else:
        sys.exit(0)