From 96dcbc1aab152df332c732fff462f235feae7a79 Mon Sep 17 00:00:00 2001 From: darko-poljak Date: Mon, 27 Jan 2014 09:17:29 +0100 Subject: [PATCH] Changed main function to support setuptools entry_points console_scripts generation. --- sweeper/sweeper.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/sweeper/sweeper.py b/sweeper/sweeper.py index 2101453..84440c6 100644 --- a/sweeper/sweeper.py +++ b/sweeper/sweeper.py @@ -109,11 +109,14 @@ def iter_file_dups(topdirs=['./'], hashalg='md5', block_size=4096): yield fpaths -def main(args): +def main(): """Main when used as script. See usage (--help). """ import json + from docopt import docopt + arguments = docopt(__doc__) + topdirs = args[''] if not topdirs: topdirs = ['./'] @@ -140,6 +143,4 @@ def main(args): # if used as script call main function if __name__ == '__main__': - from docopt import docopt - arguments = docopt(__doc__) - main(arguments) + main()