erqe211 发表于 2015-5-12 08:47:23

python中给程序加锁之fcntl模块的使用

python 中给文件加锁——fcntl模块
import fcntl

打开一个文件
f = open('./test') ##当前目录下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
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/usr/bin/python
#coding:utf8

import os
import sys
import time
import fcntl #导入模块

class FLOCK(ojbect):
    def __init__(self,name):
      """
      :param name: 文件名
      """

      self.fobj = open(name,'w')
      self.fd = self.fobj.fileno()

    def lock(self):
      try:
            fcntl.lockf(sefl.fd,fcntl.LOCK_EX|fcntl.LOCK_NB) #给文件加锁,使用了fcntl.LOCK_NB
            print '给文件加锁,稍等 ... ...'
            time.sleep(20)
            return True
      except:
            print '文件加锁,无法执行,请稍后运行。'
            retrun False

    def unlock(self):
      self.fobj.close()
      print '已解锁'

if __name__ == "__main__":
    print sys.argv
    locker = FLOCK(sys.argv)
    a = locker.lock()
    if a:
      print '文件已加锁'
    else:
      print '无法执行,程序已锁定,请稍等'




执行结果:
# python suo.py test
test
file lock please waiting.....(在等待期间,在另一个终端,运行此程序会有下面的结果)
文件已加锁

# python suo.py test
test
file no lock please check file...
无法执行,程序已锁定,请稍等

##程序中使用了fcntl.LOCK_NB 所以程序会直接退出,没有等待。


页: [1]
查看完整版本: python中给程序加锁之fcntl模块的使用