Add types to pre-commit

pre_commit/commands/install_uninstall.py

@@ -3,12 +3,16 @@ import logging
 import os.path
 import shutil
 import sys
+from typing import Optional
+from typing import Sequence
+from typing import Tuple
 
 from pre_commit import git
 from pre_commit import output
 from pre_commit.clientlib import load_config
 from pre_commit.repository import all_hooks
 from pre_commit.repository import install_hook_envs
+from pre_commit.store import Store
 from pre_commit.util import make_executable
 from pre_commit.util import mkdirp
 from pre_commit.util import resource_text
@@ -29,13 +33,16 @@ TEMPLATE_START = '# start templated\n'
 TEMPLATE_END = '# end templated\n'
 
 
-def _hook_paths(hook_type, git_dir=None):
+def _hook_paths(
+        hook_type: str,
+        git_dir: Optional[str] = None,
+) -> Tuple[str, str]:
     git_dir = git_dir if git_dir is not None else git.get_git_dir()
     pth = os.path.join(git_dir, 'hooks', hook_type)
     return pth, f'{pth}.legacy'
 
 
-def is_our_script(filename):
+def is_our_script(filename: str) -> bool:
     if not os.path.exists(filename):  # pragma: windows no cover (symlink)
         return False
     with open(filename) as f:
@@ -43,7 +50,7 @@ def is_our_script(filename):
     return any(h in contents for h in (CURRENT_HASH,) + PRIOR_HASHES)
 
 
-def shebang():
+def shebang() -> str:
     if sys.platform == 'win32':
         py = 'python'
     else:
@@ -63,9 +70,12 @@ def shebang():
 
 
 def _install_hook_script(
-        config_file, hook_type,
-        overwrite=False, skip_on_missing_config=False, git_dir=None,
-):
+        config_file: str,
+        hook_type: str,
+        overwrite: bool = False,
+        skip_on_missing_config: bool = False,
+        git_dir: Optional[str] = None,
+) -> None:
     hook_path, legacy_path = _hook_paths(hook_type, git_dir=git_dir)
 
     mkdirp(os.path.dirname(hook_path))
@@ -108,10 +118,14 @@ def _install_hook_script(
 
 
 def install(
-        config_file, store, hook_types,
-        overwrite=False, hooks=False,
-        skip_on_missing_config=False, git_dir=None,
-):
+        config_file: str,
+        store: Store,
+        hook_types: Sequence[str],
+        overwrite: bool = False,
+        hooks: bool = False,
+        skip_on_missing_config: bool = False,
+        git_dir: Optional[str] = None,
+) -> int:
     if git.has_core_hookpaths_set():
         logger.error(
             'Cowardly refusing to install hooks with `core.hooksPath` set.\n'
@@ -133,11 +147,12 @@ def install(
     return 0
 
 
-def install_hooks(config_file, store):
+def install_hooks(config_file: str, store: Store) -> int:
     install_hook_envs(all_hooks(load_config(config_file), store), store)
+    return 0
 
 
-def _uninstall_hook_script(hook_type):  # type: (str) -> None
+def _uninstall_hook_script(hook_type: str) -> None:
     hook_path, legacy_path = _hook_paths(hook_type)
 
     # If our file doesn't exist or it isn't ours, gtfo.
@@ -152,7 +167,7 @@ def _uninstall_hook_script(hook_type):  # type: (str) -> None
         output.write_line(f'Restored previous hooks to {hook_path}')
 
 
-def uninstall(hook_types):
+def uninstall(hook_types: Sequence[str]) -> int:
     for hook_type in hook_types:
         _uninstall_hook_script(hook_type)
     return 0