mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 19:46:54 +00:00
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
import os.path
|
|
from argparse import ArgumentParser
|
|
from typing import Optional
|
|
from typing import Sequence
|
|
from typing import Set
|
|
|
|
OK = 0
|
|
ERR = 1
|
|
|
|
|
|
class ModuleInitChecker:
|
|
def __init__(self):
|
|
# type: () -> None
|
|
self.seen_dirnames = set() # type: Set[str]
|
|
|
|
def check(self, filename):
|
|
# type: (str) -> int
|
|
dirname = os.path.dirname(filename)
|
|
if dirname in self.seen_dirnames:
|
|
return OK
|
|
|
|
init_file = os.path.join(dirname, '__init__.py')
|
|
if dirname and not os.path.exists(init_file):
|
|
self.seen_dirnames.add(dirname)
|
|
with open(init_file, 'w'):
|
|
print(f'Created {init_file}')
|
|
return ERR
|
|
|
|
return OK
|
|
|
|
|
|
def main(argv=None): # type: (Optional[Sequence[str]]) -> int
|
|
parser = ArgumentParser()
|
|
parser.add_argument('filenames', nargs='*')
|
|
|
|
args = parser.parse_args(argv)
|
|
|
|
status = OK
|
|
checker = ModuleInitChecker()
|
|
for filename in args.filenames:
|
|
status |= checker.check(filename)
|
|
|
|
return status
|
|
|
|
|
|
if __name__ == '__main__':
|
|
exit(main())
|