Refactored how entry points work.
diff --git a/pre_commit_hooks/trailing_whitespace_fixer.py b/pre_commit_hooks/trailing_whitespace_fixer.py
index 5038e6e..bf016af 100644
--- a/pre_commit_hooks/trailing_whitespace_fixer.py
+++ b/pre_commit_hooks/trailing_whitespace_fixer.py
@@ -3,7 +3,10 @@
import sys
from plumbum import local
+from pre_commit_hooks.util import entry
+
+@entry
def fix_trailing_whitespace(argv):
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*', help='Filenames to fix')
@@ -22,9 +25,5 @@
return 0
-def entry():
- fix_trailing_whitespace(sys.argv[1:])
-
-
if __name__ == '__main__':
- sys.exit(entry())
+ sys.exit(fix_trailing_whitespace())
