1. 程式人生 > >python中進程間通訊——文件鎖之fcntl模塊的使用

python中進程間通訊——文件鎖之fcntl模塊的使用

too clas tool python except sleep bject ati tle

python 中給文件加鎖——fcntl模塊
import fcntl

打開一個文件
##當前目錄下test文件要先存在,如果不存在會報錯。或者以寫的方式打開
f = open(‘./test‘)
對該文件加密:
fcntl.flock(f,fcntl.LOCK_EX)
這樣就對文件test加鎖了,如果有其他進程對test文件加鎖,則不能成功,會被阻塞,但不會退出程序。
解鎖:fcntl.flock(f,fcntl.LOCK_UN)

fcntl模塊:
flock() : flock(f, operation)
operation : 包括:
fcntl.LOCK_UN 解鎖
fcntl.LOCK_EX 排他鎖
fcntl.LOCK_SH 共享鎖
fcntl.LOCK_NB 非阻塞鎖
LOCK_SH 共享鎖:所有進程沒有寫訪問權限,即使是加鎖進程也沒有。所有進程有讀訪問權限。
LOCK_EX 排他鎖:除加鎖進程外其他進程沒有對已加鎖文件讀寫訪問權限。
LOCK_NB 非阻塞鎖:
如果指定此參數,函數不能獲得文件鎖就立即返回,否則,函數會等待獲得文件鎖。
LOCK_NB可以同LOCK_SH或LOCK_NB進行按位或(|)運算操作。 fcnt.flock(f,fcntl.LOCK_EX|fcntl.LOCK_NB)

請看事例:
技術分享圖片
 1 import sys
 2 import time
 3 import fcntl
 4 
 5 class FLOCK(object):
 6 
 7     def __init__(self, name):
 8         self.fobj = open(name, ‘w‘)
 9         self.fd = self.fobj.fileno()
10 
11     def lock(self):
12         try:
13             fcntl.lockf(self.fd, fcntl.LOCK_EX | fcntl.LOCK_NB)  # 給文件加鎖,使用了fcntl.LOCK_NB
14             print ‘給文件加鎖,稍等 ... ...‘
15             time.sleep(20)
16             return True
17         except:
18             print ‘文件加鎖,無法執行,請稍後運行。‘
19             return False
20 
21 
22 def unlock(self):
23     self.fobj.close()
24     print ‘已解鎖‘
25 
26 if __name__ == "__main__":
27     print sys.argv[1]
28     locker = FLOCK(sys.argv[1])
29     a = locker.lock()
30     if a:
31         print ‘文件已加鎖‘
32     else:
33         print ‘無法執行,程序已鎖定,請稍等‘
技術分享圖片

先運行一個終端會打印:

  python lockfile.py test

  test

  給文件加鎖,稍等 ... ...

  文件已加鎖

運行另外一個終端:

  test

文件加鎖,無法執行,請稍後運行。

無法執行,程序已鎖定,請稍等

python中進程間通訊——文件鎖之fcntl模塊的使用