From 6ea743740ff4524287e5b63a18a45bd19d2451bb Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Thu, 7 Sep 2017 21:44:16 +0200 Subject: o Reformat all. --- src/ee/__main__.py | 69 +++++++++++++++++++++++++++++------------------------- 1 file changed, 37 insertions(+), 32 deletions(-) (limited to 'src/ee/__main__.py') diff --git a/src/ee/__main__.py b/src/ee/__main__.py index 6bb40b0..ecc0a98 100644 --- a/src/ee/__main__.py +++ b/src/ee/__main__.py @@ -5,51 +5,56 @@ import logging import pkgutil import sys + def eprint(*args, **kwargs): - print(*args, file=sys.stderr, **kwargs) + print(*args, file=sys.stderr, **kwargs) + @total_ordering class Tool(object): - def __init__(self, module_name, name): - self.module_name = module_name - self.name = name + def __init__(self, module_name, name): + self.module_name = module_name + self.name = name + + def __eq__(self, other): + return self.name == other.name - def __eq__(self, other): - return self.name == other.name + def __lt__(self, other): + return self.name < other.name - def __lt__(self, other): - return self.name < other.name def find_tools(): - prefix = ee.tools.__name__ + '.' - ps = pkgutil.walk_packages(ee.tools.__path__, prefix) - tools = [] - for (module_loader, module_name, ispkg) in ps: - name = module_name.replace(prefix, '').replace('_', '-') - tools.append(Tool(module_name, name)) - return sorted(tools) + prefix = ee.tools.__name__ + '.' + ps = pkgutil.walk_packages(ee.tools.__path__, prefix) + tools = [] + for (module_loader, module_name, ispkg) in ps: + name = module_name.replace(prefix, '').replace('_', '-') + tools.append(Tool(module_name, name)) + return sorted(tools) + 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") - requests_log.setLevel(logging.DEBUG) - requests_log.propagate = True + 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") + requests_log.setLevel(logging.DEBUG) + requests_log.propagate = True + + tools = find_tools() - tools = find_tools() + name = sys.argv[1] + del sys.argv[1] - name = sys.argv[1] - del sys.argv[1] + for t in tools: + if t.name != name: + continue + sys.argv[0] = t.name + importlib.import_module(t.module_name) + exit(0) - for t in tools: - if t.name != name: - continue - sys.argv[0] = t.name - importlib.import_module(t.module_name) - exit(0) + eprint("No such tool: {}".format(name)) + exit(1) - eprint("No such tool: {}".format(name)) - exit(1) if __name__ == "__main__": - main() + main() -- cgit v1.2.3