pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-08 20:44:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove @entry decorator (and misc cleanup)

Browse source
This commit is contained in:
Anthony Sottile 2015-01-04 11:08:53 -08:00
parent 9bfa01da81
commit 2f1d2bbe5b
13 changed files with 25 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pre_commit_hooks/requirements_txt_fixer.py
View file

@ -2,8 +2,6 @@ from __future__ import print_function
import argparse
from pre_commit_hooks.util import entry
class Requirement(object):
@ -67,8 +65,7 @@ def fix_requirements(f):
return 1
@entry
def fix_requirements_txt(argv):
def fix_requirements_txt(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to fix')
args = parser.parse_args(argv)
@ -76,7 +73,7 @@ def fix_requirements_txt(argv):
retv = 0
for arg in args.filenames:
with open(arg, 'rb+') as f:
retv |= fix_requirements(f)
with open(arg, 'rb+') as file_obj:
retv |= fix_requirements(file_obj)
return retv