22
33import argparse
44import re
5+ from typing import FrozenSet
56from typing import Optional
67from typing import Sequence
7- from typing import Set
88
99from pre_commit_hooks .util import CalledProcessError
1010from pre_commit_hooks .util import cmd_output
1111
1212
13- def is_on_branch (protected , patterns = set ()):
14- # type: (Set [str], Set [str]) -> bool
13+ def is_on_branch (protected , patterns = frozenset ()):
14+ # type: (FrozenSet [str], FrozenSet [str]) -> bool
1515 try :
1616 ref_name = cmd_output ('git' , 'symbolic-ref' , 'HEAD' )
1717 except CalledProcessError :
@@ -33,13 +33,13 @@ def main(argv=None): # type: (Optional[Sequence[str]]) -> int
3333 '-p' , '--pattern' , action = 'append' ,
3434 help = (
3535 'regex pattern for branch name to disallow commits to, '
36- 'May be specified multiple times'
36+ 'may be specified multiple times'
3737 ),
3838 )
3939 args = parser .parse_args (argv )
4040
41- protected = set (args .branch or ('master' ,))
42- patterns = set (args .pattern or ())
41+ protected = frozenset (args .branch or ('master' ,))
42+ patterns = frozenset (args .pattern or ())
4343 return int (is_on_branch (protected , patterns ))
4444
4545
0 commit comments