45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
import os
|
||
import sys
|
||
import tempfile
|
||
from pathlib import Path
|
||
|
||
class SingleInstance:
|
||
def __init__(self, app_name="PCM_Report"):
|
||
self.lockfile = Path(tempfile.gettempdir()) / f"{app_name}.lock"
|
||
self.fp = None
|
||
|
||
def __enter__(self):
|
||
try:
|
||
if self.lockfile.exists():
|
||
# 尝试读取PID，检查进程是否还在运行
|
||
try:
|
||
pid = int(self.lockfile.read_text().strip())
|
||
# 检查进程是否存在
|
||
if sys.platform == "win32":
|
||
import ctypes
|
||
kernel32 = ctypes.windll.kernel32
|
||
PROCESS_QUERY_INFORMATION = 0x0400
|
||
handle = kernel32.OpenProcess(PROCESS_QUERY_INFORMATION, 0, pid)
|
||
if handle:
|
||
kernel32.CloseHandle(handle)
|
||
return False # 进程存在，不允许启动
|
||
else:
|
||
os.kill(pid, 0) # Unix系统检查
|
||
return False
|
||
except (ValueError, ProcessLookupError, OSError):
|
||
# 进程不存在，删除旧锁文件
|
||
self.lockfile.unlink(missing_ok=True)
|
||
|
||
# 创建锁文件
|
||
self.lockfile.write_text(str(os.getpid()))
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def __exit__(self, *args):
|
||
try:
|
||
if self.lockfile.exists():
|
||
self.lockfile.unlink()
|
||
except Exception:
|
||
pass
|