Only print that the lock is being acquired when waiting

asottile · asottile · commit 7cb3e00731d5 · 2017-09-04T11:53:17.000-07:00
diff --git a/pre_commit/file_lock.py b/pre_commit/file_lock.py
@@ -12,18 +12,22 @@
     _region = 0xffff
 
     @contextlib.contextmanager
-    def _locked(fileno):
-        while True:
-            try:
-                msvcrt.locking(fileno, msvcrt.LK_LOCK, _region)
-            except OSError as e:
-                # Locking violation. Returned when the _LK_LOCK or _LK_RLCK
-                # flag is specified and the file cannot be locked after 10
-                # attempts.
-                if e.errno != errno.EDEADLOCK:
-                    raise
-            else:
-                break
+    def _locked(fileno, blocked_cb):
+        try:
+            msvcrt.locking(fileno, msvcrt.LK_NBLCK, _region)
+        except IOError:
+            blocked_cb()
+            while True:
+                try:
+                    msvcrt.locking(fileno, msvcrt.LK_LOCK, _region)
+                except IOError as e:
+                    # Locking violation. Returned when the _LK_LOCK or _LK_RLCK
+                    # flag is specified and the file cannot be locked after 10
+                    # attempts.
+                    if e.errno != errno.EDEADLOCK:
+                        raise
+                else:
+                    break
 
         try:
             yield
@@ -38,16 +42,20 @@ def _locked(fileno):
     import fcntl
 
     @contextlib.contextmanager
-    def _locked(fileno):
-        fcntl.flock(fileno, fcntl.LOCK_EX)
+    def _locked(fileno, blocked_cb):
+        try:
+            fcntl.flock(fileno, fcntl.LOCK_EX | fcntl.LOCK_NB)
+        except IOError:
+            blocked_cb()
+            fcntl.flock(fileno, fcntl.LOCK_EX)
         try:
             yield
         finally:
             fcntl.flock(fileno, fcntl.LOCK_UN)
 
 
 @contextlib.contextmanager
-def lock(path):
+def lock(path, blocked_cb):
     with open(path, 'a+') as f:
-        with _locked(f.fileno()):
+        with _locked(f.fileno(), blocked_cb):
             yield
diff --git a/pre_commit/store.py b/pre_commit/store.py
@@ -47,10 +47,11 @@ def __init__(self, directory=None):
         self.directory = directory
 
     @contextlib.contextmanager
-    def exclusive_lock(self, quiet=False):
-        if not quiet:
+    def exclusive_lock(self):
+        def blocked_cb():  # pragma: no cover (tests are single-process)
             logger.info('Locking pre-commit directory')
-        with file_lock.lock(os.path.join(self.directory, '.lock')):
+
+        with file_lock.lock(os.path.join(self.directory, '.lock'), blocked_cb):
             yield
 
     def _write_readme(self):
@@ -89,7 +90,7 @@ def _create(self):
 
         if os.path.exists(self.db_path):
             return
-        with self.exclusive_lock(quiet=True):
+        with self.exclusive_lock():
             # Another process may have already completed this work
             if os.path.exists(self.db_path):  # pragma: no cover (race)
                 return
diff --git a/tests/commands/install_uninstall_test.py b/tests/commands/install_uninstall_test.py
@@ -142,8 +142,7 @@ def _get_commit_output(tempdir_factory, touch_file='foo', **kwargs):
 
 
 NORMAL_PRE_COMMIT_RUN = re.compile(
-    r'^\[INFO\] Locking pre-commit directory\r?\n'
-    r'\[INFO\] Initializing environment for .+\.\r?\n'
+    r'^\[INFO\] Initializing environment for .+\.\r?\n'
     r'Bash hook\.+Passed\r?\n'
     r'\[master [a-f0-9]{7}\] Commit!\r?\n' +
     FILES_CHANGED +
@@ -255,8 +254,7 @@ def test_environment_not_sourced(tempdir_factory):
 
 
 FAILING_PRE_COMMIT_RUN = re.compile(
-    r'^\[INFO\] Locking pre-commit directory\r?\n'
-    r'\[INFO\] Initializing environment for .+\.\r?\n'
+    r'^\[INFO\] Initializing environment for .+\.\r?\n'
     r'Failing hook\.+Failed\r?\n'
     r'hookid: failing_hook\r?\n'
     r'\r?\n'
@@ -334,7 +332,6 @@ def test_install_existing_hook_no_overwrite_idempotent(tempdir_factory):
 
 FAIL_OLD_HOOK = re.compile(
     r'fail!\r?\n'
-    r'\[INFO\] Locking pre-commit directory\r?\n'
     r'\[INFO\] Initializing environment for .+\.\r?\n'
     r'Bash hook\.+Passed\r?\n',
 )
diff --git a/tests/repository_test.py b/tests/repository_test.py
@@ -559,8 +559,8 @@ def test_reinstall(tempdir_factory, store, log_info_mock):
     config = make_config_from_repo(path)
     repo = Repository.create(config, store)
     repo.require_installed()
-    # We print some logging during clone (2) + install (4)
-    assert log_info_mock.call_count == 6
+    # We print some logging during clone (1) + install (3)
+    assert log_info_mock.call_count == 4
     log_info_mock.reset_mock()
     # Reinstall with same repo should not trigger another install
     repo.require_installed()
diff --git a/tests/store_test.py b/tests/store_test.py
@@ -88,7 +88,7 @@ def test_clone(store, tempdir_factory, log_info_mock):
 
     ret = store.clone(path, sha)
     # Should have printed some stuff
-    assert log_info_mock.call_args_list[1][0][0].startswith(
+    assert log_info_mock.call_args_list[0][0][0].startswith(
         'Initializing environment for ',
     )
 

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def test_clone(store, tempdir_factory, log_info_mock):`
`88`	`88`
`89`	`89`	`ret = store.clone(path, sha)`
`90`	`90`	`# Should have printed some stuff`
`91`		`- assert log_info_mock.call_args_list[1][0][0].startswith(`
	`91`	`+ assert log_info_mock.call_args_list[0][0][0].startswith(`
`92`	`92`	`'Initializing environment for ',`
`93`	`93`	`)`
`94`	`94`