Python通过文件锁实现同一时刻只能有一个进程在运行某项任务（Windows平台）

最新推荐文章于 2023-11-23 17:50:02 发布

Y32579

最新推荐文章于 2023-11-23 17:50:02 发布

阅读量666

点赞数

文章标签： python windows 开发语言

本文链接：https://blog.csdn.net/y32579/article/details/130907503

版权

该代码实现了一个Python类`SingleInstanceGuardian`，用于确保同一时间只有一个实例运行。它通过创建并锁定一个文件来实现进程间的同步，如果检测到锁已被其他实例持有，则认为已有应用在运行并退出。`run_task`方法执行任务，在任务完成后释放文件锁。依赖`pywin32`库。

摘要由CSDN通过智能技术生成

# coding: utf-8
import win32con
import win32file
import pywintypes
import winerror
import sys


class SingleInstanceGuardian:
    def __init__(self, lock_file_path):
        self._lock_file_path = lock_file_path
        self._lock_file = None

    def _acquire_lock(self):
        try:
            # 创建一个文件用于锁定
            self._lock_file = win32file.CreateFile(
                self._lock_file_path,
                win32file.GENERIC_WRITE,
                0,
                None,
                win32file.CREATE_ALWAYS,
                win32file.FILE_ATTRIBUTE_NORMAL,
                None
            )

            # 尝试获取文件锁
            win32file.LockFileEx(self._lock_file, win32con.LOCKFILE_EXCLUSIVE_LOCK | win32con.LOCKFILE_FAIL_IMMEDIATELY, 0, -0x10000, pywintypes.OVERLAPPED())

            return True
        except pywintypes.error as e:
            if e.winerror == winerror.ERROR_LOCK_VIOLATION:
                return False
            else:
                raise

    def _release_lock(self):
        if self._lock_file:
            win32file.UnlockFileEx(self._lock_file, 0, -0x10000, pywintypes.OVERLAPPED())
            win32file.CloseHandle(self._lock_file)

    def is_already_running(self):
        return not self._acquire_lock()

    def run_task(self):
        if self.is_already_running():
            print("Another instance is already running.")
            sys.exit(1)

        # 执行任务代码

        # 任务完成后释放锁
        self._release_lock()


# 使用示例
lock_file_path = r"D:\lock_file.lock"
guardian = SingleInstanceGuardian(lock_file_path)
guardian.run_task()