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

转载自：http://zhou123.blog.51cto.com/4355617/1650185

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)

以下为实例：

#!/usr/bin/env python

#coding=utf8

import os

import sys

import time

import fcntl

class FLOCK():

    def __init__(self,name):

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

        self.fd = self.fobj.fileno()

    def lock(self):

        try:

            fcntl.flock(self.fd,fcntl.LOCK_EX|fcntl.LOCK_NB)  ####文件加锁

            print('file  is   lock.........')

            data = self.fobj.read()

            print(data)

            time.sleep(20) 

            return(True)

        except:            

            print('file is locked  please waiting!!!!!!!!!!!!!!')

            return(False)

    def unlock(self):

        fcntl.flock(self.fd,fcntl.LOCK_UN)

        self.fobj.close()

        frint('is  unlock!!!!!!!!!')

if __name__ == '__main__':

    print sys.argv[1]

    locker = FLOCK(sys.argv[1])

    a = locker.lock()

    if a:

        print('lock success')

    else:

        print('no')

执行结果：

[root@cdn ~]# python fc.py test     #######此脚本锁定文件

test

file  is   lock............

@!sdagdggggggggggggg            ########test文件内容

lock success

[root@cdn ~]# python fc.py  test   ########开启另一个终端，运行脚本，文件被锁定

test

file is locked  please waiting!!!!!!!!!!!!!!

no