Apply typing to all of pre-commit-hooks

This commit is contained in:
Anthony Sottile
2019-01-31 19:19:10 -08:00
parent 63cc3414e9
commit 030bfac7e4
54 changed files with 401 additions and 264 deletions

View File

@ -5,10 +5,12 @@ from __future__ import unicode_literals
import argparse
import io
import sys
import xml.sax
import xml.sax.handler
from typing import Optional
from typing import Sequence
def check_xml(argv=None):
def main(argv=None): # type: (Optional[Sequence[str]]) -> int
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='XML filenames to check.')
args = parser.parse_args(argv)
@ -17,7 +19,7 @@ def check_xml(argv=None):
for filename in args.filenames:
try:
with io.open(filename, 'rb') as xml_file:
xml.sax.parse(xml_file, xml.sax.ContentHandler())
xml.sax.parse(xml_file, xml.sax.handler.ContentHandler())
except xml.sax.SAXException as exc:
print('{}: Failed to xml parse ({})'.format(filename, exc))
retval = 1
@ -25,4 +27,4 @@ def check_xml(argv=None):
if __name__ == '__main__':
sys.exit(check_xml())
sys.exit(main())