diff options
author | Trygve Laugstøl <trygvis@inamo.no> | 2017-09-07 21:44:16 +0200 |
---|---|---|
committer | Trygve Laugstøl <trygvis@inamo.no> | 2017-09-07 21:44:16 +0200 |
commit | 6ea743740ff4524287e5b63a18a45bd19d2451bb (patch) | |
tree | d14ec9c5b580450b8e2db2b115be85aea48afc42 /src/ee/__main__.py | |
parent | a3da99e22540c026e4acf7bdc735aa221a4dd752 (diff) | |
download | ee-python-6ea743740ff4524287e5b63a18a45bd19d2451bb.tar.gz ee-python-6ea743740ff4524287e5b63a18a45bd19d2451bb.tar.bz2 ee-python-6ea743740ff4524287e5b63a18a45bd19d2451bb.tar.xz ee-python-6ea743740ff4524287e5b63a18a45bd19d2451bb.zip |
o Reformat all.
Diffstat (limited to 'src/ee/__main__.py')
-rw-r--r-- | src/ee/__main__.py | 69 |
1 files changed, 37 insertions, 32 deletions
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() |