python之給文件加鎖（fcntl模塊）

對該文件加密：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 python2
# -*- coding: utf-8 -*-


import os
import sys
import time
import fcntl  # 導入模塊


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

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

    def lock(self):

        try:
            fcntl.lockf(self.fd, fcntl.LOCK_EX | fcntl.LOCK_NB)  # 給文件加鎖，使用了fcntl.LOCK_NB
            print '給文件加鎖，稍等 ... ...'
            time.sleep(20)
            return True
        except:
            print '文件加鎖，無法執行，請稍后運行。'
            return 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 '無法執行，程序已鎖定，請稍等'

運行1： python  text.py  1.txt

[root@localhost text]# python text.py 1.txt 
1.txt
給文件加鎖，稍等 ... ...
文件已加鎖

 

運行2：python text.py  1.txt

[root@localhost text]# python text.py 1.txt 
1.txt
文件加鎖，無法執行，請稍后運行。
無法執行，程序已鎖定，請稍等
[root@localhost text]# python text.py 1.txt 
1.txt
給文件加鎖，稍等 ... ...
^C文件加鎖，無法執行，請稍后運行。
無法執行，程序已鎖定，請稍等
[root@localhost text]#