38 lines
906 B
Plaintext
38 lines
906 B
Plaintext
from stdint import *
|
|
from win32.win32base import *
|
|
from win32.win32sync import CRITICAL_SECTION, InitializeCriticalSection, EnterCriticalSection, LeaveCriticalSection, DeleteCriticalSection, TryEnterCriticalSection
|
|
|
|
|
|
class Lock:
|
|
"""Mutex built on CRITICAL_SECTION, supports recursive locking.
|
|
|
|
Usage:
|
|
lock: Lock = Lock()
|
|
with lock:
|
|
# Critical section
|
|
pass
|
|
"""
|
|
cs: CRITICAL_SECTION
|
|
|
|
def __init__(self):
|
|
InitializeCriticalSection(&self.cs)
|
|
|
|
def acquire(self):
|
|
EnterCriticalSection(&self.cs)
|
|
|
|
def release(self):
|
|
LeaveCriticalSection(&self.cs)
|
|
|
|
def try_acquire(self) -> BOOL:
|
|
return TryEnterCriticalSection(&self.cs)
|
|
|
|
def __enter__(self) -> Lock*:
|
|
self.acquire()
|
|
return self
|
|
|
|
def __exit__(self):
|
|
self.release()
|
|
|
|
def delete(self):
|
|
DeleteCriticalSection(&self.cs)
|