diff --git a/scripts-dev/definitions.py b/scripts-dev/definitions.py
index 1deb0fe2b7..9eddb6d515 100755
--- a/scripts-dev/definitions.py
+++ b/scripts-dev/definitions.py
@@ -19,10 +19,10 @@ class DefinitionVisitor(ast.NodeVisitor):
self.names = {}
self.attrs = set()
self.definitions = {
- 'def': self.functions,
- 'class': self.classes,
- 'names': self.names,
- 'attrs': self.attrs,
+ "def": self.functions,
+ "class": self.classes,
+ "names": self.names,
+ "attrs": self.attrs,
}
def visit_Name(self, node):
@@ -47,23 +47,23 @@ class DefinitionVisitor(ast.NodeVisitor):
def non_empty(defs):
- functions = {name: non_empty(f) for name, f in defs['def'].items()}
- classes = {name: non_empty(f) for name, f in defs['class'].items()}
+ functions = {name: non_empty(f) for name, f in defs["def"].items()}
+ classes = {name: non_empty(f) for name, f in defs["class"].items()}
result = {}
if functions:
- result['def'] = functions
+ result["def"] = functions
if classes:
- result['class'] = classes
- names = defs['names']
+ result["class"] = classes
+ names = defs["names"]
uses = []
- for name in names.get('Load', ()):
- if name not in names.get('Param', ()) and name not in names.get('Store', ()):
+ for name in names.get("Load", ()):
+ if name not in names.get("Param", ()) and name not in names.get("Store", ()):
uses.append(name)
- uses.extend(defs['attrs'])
+ uses.extend(defs["attrs"])
if uses:
- result['uses'] = uses
- result['names'] = names
- result['attrs'] = defs['attrs']
+ result["uses"] = uses
+ result["names"] = names
+ result["attrs"] = defs["attrs"]
return result
@@ -81,33 +81,33 @@ def definitions_in_file(filepath):
def defined_names(prefix, defs, names):
- for name, funcs in defs.get('def', {}).items():
- names.setdefault(name, {'defined': []})['defined'].append(prefix + name)
+ for name, funcs in defs.get("def", {}).items():
+ names.setdefault(name, {"defined": []})["defined"].append(prefix + name)
defined_names(prefix + name + ".", funcs, names)
- for name, funcs in defs.get('class', {}).items():
- names.setdefault(name, {'defined': []})['defined'].append(prefix + name)
+ for name, funcs in defs.get("class", {}).items():
+ names.setdefault(name, {"defined": []})["defined"].append(prefix + name)
defined_names(prefix + name + ".", funcs, names)
def used_names(prefix, item, defs, names):
- for name, funcs in defs.get('def', {}).items():
+ for name, funcs in defs.get("def", {}).items():
used_names(prefix + name + ".", name, funcs, names)
- for name, funcs in defs.get('class', {}).items():
+ for name, funcs in defs.get("class", {}).items():
used_names(prefix + name + ".", name, funcs, names)
- path = prefix.rstrip('.')
- for used in defs.get('uses', ()):
+ path = prefix.rstrip(".")
+ for used in defs.get("uses", ()):
if used in names:
if item:
- names[item].setdefault('uses', []).append(used)
- names[used].setdefault('used', {}).setdefault(item, []).append(path)
+ names[item].setdefault("uses", []).append(used)
+ names[used].setdefault("used", {}).setdefault(item, []).append(path)
-if __name__ == '__main__':
+if __name__ == "__main__":
- parser = argparse.ArgumentParser(description='Find definitions.')
+ parser = argparse.ArgumentParser(description="Find definitions.")
parser.add_argument(
"--unused", action="store_true", help="Only list unused definitions"
)
@@ -119,7 +119,7 @@ if __name__ == '__main__':
)
parser.add_argument(
"directories",
- nargs='+',
+ nargs="+",
metavar="DIR",
help="Directories to search for definitions",
)
@@ -164,7 +164,7 @@ if __name__ == '__main__':
continue
if ignore and any(pattern.match(name) for pattern in ignore):
continue
- if args.unused and definition.get('used'):
+ if args.unused and definition.get("used"):
continue
result[name] = definition
@@ -196,9 +196,9 @@ if __name__ == '__main__':
continue
result[name] = definition
- if args.format == 'yaml':
+ if args.format == "yaml":
yaml.dump(result, sys.stdout, default_flow_style=False)
- elif args.format == 'dot':
+ elif args.format == "dot":
print("digraph {")
for name, entry in result.items():
print(name)
|
