Sam Park 2020-01-18 11:29:05 -08:00
parent 31853d6c43
commit 513a24cde9
No known key found for this signature in database
GPG key ID: 5CA023C8A1C73D5F
5 changed files with 103 additions and 0 deletions

View file

@ -0,0 +1,25 @@
import os
from argparse import ArgumentParser
from typing import Optional
from typing import Sequence
def main(argv=None): # type: (Optional[Sequence[str]]) -> int
parser = ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to check')
args = parser.parse_args(argv)
directories = {os.path.dirname(f) for f in args.filenames}
missing_dirs = set()
for d in directories:
if not os.path.exists(os.path.join(d, '__init__.py')):
missing_dirs.add(d)
for d in sorted(missing_dirs):
print('No __init__.py file found in: {}'.format(d))
return 1 if len(missing_dirs) else 0
if __name__ == '__main__':
exit(main())