python 中给文件加锁——fcntl模块
如果没有fcntl模块则用
sudo pip install fcntl安装
模块简单说明:
打开文件,不存在则创建之
f = open('./test','w')
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 #!/usr/bin/python 2 #coding:utf8 3 4 import os 5 import sys 6 import time 7 import fcntl 8 9 class FLOCK(ojbect): 10 def __init__(self,name): 11 12 self.fobj = open(name,'w') 13 self.fd = self.fobj.fileno() 14 15 def lock(self): 16 try: 17 #给文件加锁,使用了fcntl.LOCK_NB 18 fcntl.lockf(sefl.fd,fcntl.LOCK_EX|fcntl.LOCK_NB) 19 print '给文件加锁,稍等 ... ...' 20 time.sleep(20) 21 return True 22 except: 23 print '文件加锁,无法执行,请稍后运行。' 24 retrun False 25 26 def unlock(self): 27 self.fobj.close() 28 print '已解锁' 29 30 if __name__ == "__main__": 31 print sys.argv[1] 32 locker = FLOCK(sys.argv[1]) 33 a = locker.lock() 34 if a: 35 print '文件已加锁' 36 else: 37 print '无法执行,程序已锁定,请稍等'
执行结果:
# python lock.py test
test
file lock please waiting..... (在等待期间,在另一个终端,运行此程序会有下面的结果)
文件已加锁
# python lock.py test
test
file no lock please check file...
无法执行,程序已锁定,请稍等