Fix __enter__

sebastinas · sebastinas · commit 4e611563e7bd · 2016-12-21T00:17:33.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/filelock.py b/bpython/filelock.py
@@ -53,6 +53,7 @@ def release(self):
 
     def __enter__(self):
         self.acquire()
+        return self
 
     def __exit__(self, *args):
         if self.locked:
@@ -81,7 +82,6 @@ def acquire(self):
         except IOError as e:
             if e.errno != errno.ENOLCK:
                 raise e
-        return self
 
     def release(self):
         fcntl.flock(self.fd, fcntl.LOCK_UN)
@@ -98,7 +98,6 @@ def __init__(self, fd, mode=None):
     def acquire(self):
         msvcrt.locking(self.fd, msvcrt.LK_NBLCK, 1)
         self.locked = True
-        return self
 
     def release(self):
         msvcrt.locking(self.fd, msvcrt.LK_UNLCK, 1)
