Add types to pre-commit

pre_commit/commands/autoupdate.py

@@ -1,8 +1,12 @@
-import collections
 import os.path
 import re
+from typing import Any
+from typing import Dict
 from typing import List
+from typing import NamedTuple
 from typing import Optional
+from typing import Sequence
+from typing import Tuple
 
 from aspy.yaml import ordered_dump
 from aspy.yaml import ordered_load
@@ -16,20 +20,23 @@ from pre_commit.clientlib import load_manifest
 from pre_commit.clientlib import LOCAL
 from pre_commit.clientlib import META
 from pre_commit.commands.migrate_config import migrate_config
+from pre_commit.store import Store
 from pre_commit.util import CalledProcessError
 from pre_commit.util import cmd_output
 from pre_commit.util import cmd_output_b
 from pre_commit.util import tmpdir
 
 
-class RevInfo(collections.namedtuple('RevInfo', ('repo', 'rev', 'frozen'))):
-    __slots__ = ()
+class RevInfo(NamedTuple):
+    repo: str
+    rev: str
+    frozen: Optional[str]
 
     @classmethod
-    def from_config(cls, config):
+    def from_config(cls, config: Dict[str, Any]) -> 'RevInfo':
         return cls(config['repo'], config['rev'], None)
 
-    def update(self, tags_only, freeze):
+    def update(self, tags_only: bool, freeze: bool) -> 'RevInfo':
         if tags_only:
             tag_cmd = ('git', 'describe', 'FETCH_HEAD', '--tags', '--abbrev=0')
         else:
@@ -57,7 +64,11 @@ class RepositoryCannotBeUpdatedError(RuntimeError):
     pass
 
 
-def _check_hooks_still_exist_at_rev(repo_config, info, store):
+def _check_hooks_still_exist_at_rev(
+        repo_config: Dict[str, Any],
+        info: RevInfo,
+        store: Store,
+) -> None:
     try:
         path = store.clone(repo_config['repo'], info.rev)
         manifest = load_manifest(os.path.join(path, C.MANIFEST_FILE))
@@ -78,7 +89,11 @@ REV_LINE_RE = re.compile(r'^(\s+)rev:(\s*)([^\s#]+)(.*)(\r?\n)$', re.DOTALL)
 REV_LINE_FMT = '{}rev:{}{}{}{}'
 
 
-def _original_lines(path, rev_infos, retry=False):
+def _original_lines(
+        path: str,
+        rev_infos: List[Optional[RevInfo]],
+        retry: bool = False,
+) -> Tuple[List[str], List[int]]:
     """detect `rev:` lines or reformat the file"""
     with open(path) as f:
         original = f.read()
@@ -95,7 +110,7 @@ def _original_lines(path, rev_infos, retry=False):
         return _original_lines(path, rev_infos, retry=True)
 
 
-def _write_new_config(path, rev_infos):
+def _write_new_config(path: str, rev_infos: List[Optional[RevInfo]]) -> None:
     lines, idxs = _original_lines(path, rev_infos)
 
     for idx, rev_info in zip(idxs, rev_infos):
@@ -119,7 +134,13 @@ def _write_new_config(path, rev_infos):
         f.write(''.join(lines))
 
 
-def autoupdate(config_file, store, tags_only, freeze, repos=()):
+def autoupdate(
+        config_file: str,
+        store: Store,
+        tags_only: bool,
+        freeze: bool,
+        repos: Sequence[str] = (),
+) -> int:
     """Auto-update the pre-commit config to the latest versions of repos."""
     migrate_config(config_file, quiet=True)
     retv = 0