use threading.lock as context_manager

1 files changed, 8 insertions, 10 deletions

diff --git a/at.py b/at.py
index c50ddd8..0c5fb63 100644
--- a/at.py
+++ b/at.py
@@ -117,10 +117,9 @@ class Updater(threading.Thread):
                 del self.active[addr]
 
     def get_active_devices(self):
-        self.lock.acquire()
-        self.purge_stale()
-        r = dict(self.active)
-        self.lock.release()
+        with self.lock:
+            self.purge_stale()
+            r = dict(self.active)
         return r
 
     def get_device(self, ip):
@@ -135,12 +134,11 @@ class Updater(threading.Thread):
             atime -= self.lease_offset
         else:
             atime = time()
-        self.lock.acquire()
-        if hwaddr not in self.active or self.active[hwaddr][0] < atime:
-            self.active[hwaddr] = (atime, ip, name)
-            app.logger.info('updated %s with atime %s and ip %s',
-                            hwaddr, strfts(atime), ip)
-        self.lock.release()
+        with self.lock:
+            if hwaddr not in self.active or self.active[hwaddr][0] < atime:
+                self.active[hwaddr] = (atime, ip, name)
+                app.logger.info('updated %s with atime %s and ip %s',
+                    hwaddr, strfts(atime), ip)
 
 
 class CapUpdater(Updater):