From 9eff7a9b1e32649c2addba488cef2323dcc92b9a Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Thu, 7 Sep 2017 21:38:38 +0200 Subject: o Adding a 'ee' binary entry point. --- setup.py | 3 +++ src/ee/__main__.py | 6 ++++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index e1a428f..1580612 100644 --- a/setup.py +++ b/setup.py @@ -10,4 +10,7 @@ setup(name='ee', version='1.0', packages=find_packages('src'), package_dir={'': 'src'}, + entry_points = { + 'console_scripts': ['ee=ee.__main__:main'], + }, ) diff --git a/src/ee/__main__.py b/src/ee/__main__.py index 75688c0..6bb40b0 100644 --- a/src/ee/__main__.py +++ b/src/ee/__main__.py @@ -29,8 +29,7 @@ def find_tools(): tools.append(Tool(module_name, name)) return sorted(tools) -if __name__ == "__main__": - +def main(): logging.basicConfig() # you need to initialize logging, otherwise you will not see anything from requests logging.getLogger().setLevel(logging.DEBUG) requests_log = logging.getLogger("requests.packages.urllib3") @@ -51,3 +50,6 @@ if __name__ == "__main__": eprint("No such tool: {}".format(name)) exit(1) + +if __name__ == "__main__": + main() -- cgit v1.2.3