自动创建pid文件，并加锁

点击(此处)折叠或打开

1. #!/usr/bin/env python
   
2. #coding=utf-8
   
3. import sys
   
4. import os
   
5. import re
   
6. import traceback
   
7. import fcntl
   
8. 
   
9. class PidFile():
   
10.     def __init__(self, file = None):
    
11.         if file == None:
    
12.             argv0_list = sys.argv[0].split("\\")
    
13.             scname = argv0_list[len(argv0_list) - 1]
    
14.             if re.search(".py$", scname):
    
15.             scname = scname[0:-3]
    
16.             self.file = "/tmp/%s.pid" % (scname)
    
17.         else:
    
18.             self.file = file
    
19.         self.pidfile = None
    
20. 
    
21.     def _create(self, file):
    
22.         f = open(file, "w")
    
23.         pid = os.getpid()
    
24.         f.write("%s" % (pid))
    
25.         f.close()
    
26. 
    
27.     def __enter__(self):
    
28.         if not os.path.isfile(self.file):
    
29.             self._create(self.file)
    
30.         self.pidfile = open(self.file, 'a+')
    
31.         try:
    
32.             fcntl.flock(self.pidfile.fileno(), fcntl.LOCK_EX|fcntl.LOCK_NB)
    
33.             return True
    
34.         except:
    
35.             traceback.print_exc()
    
36.             fpid = open(self.file, "r").read()
    
37.             raise SystemExit("Process id %s Already running , pid to %s " % (fpid, self.file))
    
38.         return False    
    
39. 
    
40.     def __exit__(self, *args):
    
41.         try:
    
42.             self.pidfile.close()
    
43.         except :
    
44.             pass
    
45.         os.remove(self.file)
    
46. 
    
47. 
    
48. if __name__ == "__main__":
    
49.     # use
    
50.     with PidFile() as t:
    
51.         print "lock file %s" % (file)
    
52.         time.sleep(10)
    
53.     exit(0)

来自 “ ITPUB博客 ” ，链接：http://blog.itpub.net/26250550/viewspace-1816377/，如需转载，请注明出处，否则将追究法律责任。

转载于:http://blog.itpub.net/26250550/viewspace-1816377/