blocco pitone con-dichiarazione e il timeout

Question

Sto usando una sequenza Python 3 in questo modo:

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

io preferirei usare un con-dichiarazione al posto di esplicita di "acquisizione" e "release", ma io don so come ottenere l'effetto di timeout.

Answer 1

È possibile farlo abbastanza facilmente con un contesto manager:

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 ... 

* Nota: Questo non funziona in Python 2.x dal momento che non v'è un argomento timeout per Lock.acquire

Answer 2

versione leggermente più bello:

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 ... 

sembra non si può sottoclasse threading.Lock, così Ho dovuto invece fare una lezione di wrapper.