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)

---

下面是一個簡單的例子：

#!/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[1]
    locker = FLOCK(sys.argv[1])
    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 因此程序會直接退出，沒有等待。