1. 程式人生 > >python 的fcntl模組

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...

無法執行,程式已鎖定,請稍等