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 所以程序會直接退出，沒有等待。