自動建立pid檔案，並加鎖

G8bao7發表於2015-10-26


點選(此處)摺疊或開啟

  1. #!/usr/bin/env python
  2. #coding=utf-8
  3. import sys
  4. import os
  5. import re
  6. import traceback
  7. import fcntl

  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

  21.     def _create(self, file):
  22.         f = open(file, "w")
  23.         pid = os.getpid()
  24.         f.write("%s" % (pid))
  25.         f.close()

  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    

  40.     def __exit__(self, *args):
  41.         try:
  42.             self.pidfile.close()
  43.         except :
  44.             pass
  45.         os.remove(self.file)


  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/，如需轉載，請註明出處，否則將追究法律責任。

相關文章