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]# 

 

上一篇:Linux进程的创建与撤销


下一篇:最新kali之tsk_gettimes