mirror of
https://github.com/YunoHost-Apps/zwave-js-ui_ynh.git
synced 2024-09-03 18:06:00 +02:00
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
"""
|
|
Utility for locating a module (or package's __main__.py) with a given name
|
|
and verifying it contains the PYTHON_ARGCOMPLETE_OK marker.
|
|
|
|
The module name should be specified in a form usable with `python -m`.
|
|
|
|
Intended to be invoked by argcomplete's global completion function.
|
|
"""
|
|
import os
|
|
import sys
|
|
import tokenize
|
|
|
|
try:
|
|
from importlib.util import find_spec # type:ignore
|
|
except ImportError:
|
|
import typing as t
|
|
from collections import namedtuple
|
|
from imp import find_module
|
|
|
|
ModuleSpec = namedtuple("ModuleSpec", ["origin", "has_location", "submodule_search_locations"])
|
|
|
|
def find_spec( # type:ignore
|
|
name: str,
|
|
package: t.Optional[str] = None,
|
|
) -> t.Optional[ModuleSpec]:
|
|
"""Minimal implementation as required by `find`."""
|
|
try:
|
|
f, path, _ = find_module(name)
|
|
except ImportError:
|
|
return None
|
|
has_location = path is not None
|
|
if f is None:
|
|
return ModuleSpec(None, has_location, [path])
|
|
f.close()
|
|
return ModuleSpec(path, has_location, None)
|
|
|
|
|
|
class ArgcompleteMarkerNotFound(RuntimeError):
|
|
pass
|
|
|
|
|
|
def find(name, return_package=False):
|
|
names = name.split(".")
|
|
spec = find_spec(names[0])
|
|
if spec is None:
|
|
raise ArgcompleteMarkerNotFound('no module named "{}"'.format(names[0]))
|
|
if not spec.has_location:
|
|
raise ArgcompleteMarkerNotFound("cannot locate file")
|
|
if spec.submodule_search_locations is None:
|
|
if len(names) != 1:
|
|
raise ArgcompleteMarkerNotFound("{} is not a package".format(names[0]))
|
|
return spec.origin
|
|
if len(spec.submodule_search_locations) != 1:
|
|
raise ArgcompleteMarkerNotFound("expecting one search location")
|
|
path = os.path.join(spec.submodule_search_locations[0], *names[1:])
|
|
if os.path.isdir(path):
|
|
filename = "__main__.py"
|
|
if return_package:
|
|
filename = "__init__.py"
|
|
return os.path.join(path, filename)
|
|
else:
|
|
return path + ".py"
|
|
|
|
|
|
def main():
|
|
try:
|
|
name = sys.argv[1]
|
|
except IndexError:
|
|
raise ArgcompleteMarkerNotFound("missing argument on the command line")
|
|
|
|
filename = find(name)
|
|
if hasattr(tokenize, "open"):
|
|
open_func = tokenize.open
|
|
else:
|
|
open_func = open
|
|
|
|
try:
|
|
fp = open_func(filename)
|
|
except OSError:
|
|
raise ArgcompleteMarkerNotFound("cannot open file")
|
|
|
|
with fp:
|
|
head = fp.read(1024)
|
|
|
|
if "PYTHON_ARGCOMPLETE_OK" not in head:
|
|
raise ArgcompleteMarkerNotFound("marker not found")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except ArgcompleteMarkerNotFound as e:
|
|
sys.exit(str(e))
|