1#!/usr/local/bin/python3.8
2
3import os
4import subprocess
5import sys
6
7base = os.path.dirname(os.path.abspath(__file__))
8files = [os.path.relpath(x, base) for x in sys.argv[1:] if not x.startswith('-')]
9if not files:
10    raise SystemExit(subprocess.Popen(['mypy'] + sys.argv[1:]).wait())
11
12output = subprocess.run('dmypy run -- --follow-imports=error --show-column-numbers --no-color-output'.split(), stdout=subprocess.PIPE).stdout
13q = files[0] + ':'
14rc = 0
15for line in output.decode('utf-8').splitlines():
16    if line.startswith(q):
17        print(line)
18        rc = 1
19raise SystemExit(rc)
20