Python Schloss mit-Anweisung und Timeout

Question

Ich bin mit einem Python 3-Sequenz wie folgt aus:

lock = threading.Lock() 
res = lock.acquire(timeout=10) 
if res: 
    # do something .... 
    lock.release() 
else: 
    # do something else ... 

Ich würde es vorziehen, eine mit-Anweisung anstelle von expliziten „acquire“ und „Release“ zu verwenden, aber ich don Ich weiß nicht, wie man den Timeout-Effekt bekommt.

Answer 1

Sie dies mit einem Kontext-Manager ziemlich leicht tun können:

import threading 
from contextlib import contextmanager 

@contextmanager 
def acquire_timeout(lock, timeout): 
    result = lock.acquire(timeout=timeout) 
    yield result 
    if result: 
     lock.release() 


# Usage: 
lock = threading.Lock() 

with acquire_timeout(lock, 2) as acquired: 
    if acquired: 
     print('got the lock') 
     # do something .... 
    else: 
     print('timeout: lock not available') 
     # do something else ... 

* Hinweis: Dies wird nicht in Python 2.x funktionieren, da es zu Lock.acquire

kein timeout Argument ist

Answer 2

Etwas schönere Version:

import threading 
from contextlib import contextmanager 


class TimeoutLock(object): 
    def __init__(self): 
     self._lock = threading.Lock() 

    def acquire(self, blocking=True, timeout=-1): 
     return self._lock.acquire(blocking, timeout) 

    @contextmanager 
    def acquire_timeout(self, timeout): 
     result = self._lock.acquire(timeout=timeout) 
     yield result 
     if result: 
      self._lock.release() 

    def release(self): 
     self._lock.release() 

# Usage: 
lock = TimeoutLock() 

with lock.acquire_timeout(3) as result: 
    if result: 
     print('got the lock') 
     # do something .... 
    else: 
     print('timeout: lock not available') 
     # do something else ... 

Es scheint Sie nicht threading.Lock Unterklasse kann, so Ich musste stattdessen eine Wrapper-Klasse erstellen.