Find cgx files and imports from cgx files from project that includes collagraph (#96)

Author: berendkleinhaneveld

collagraph/__pyinstaller/__init__.py

@@ -0,0 +1,8 @@
+from os.path import dirname
+
+
+HERE = dirname(__file__)
+
+
+def get_hook_dirs():
+    return [HERE]

collagraph/__pyinstaller/hook-collagraph.py

@@ -0,0 +1,57 @@
+import ast
+from pathlib import Path
+
+from PyInstaller.utils.hooks import collect_data_files
+
+from collagraph.cgx.cgx import CGXParser, get_script_ast
+
+
+def hook(hook_api):
+    collagraph_uses = hook_api.analysis.graph.get_code_using("collagraph")
+
+    hidden_imports = set()
+    datas = []
+    for package, code in collagraph_uses.items():
+        filename = Path(code.co_filename)
+        hidden_imports |= collect_hidden_imports(filename.parent)
+        datas += collect_data_files(package, includes=["**/*.cgx"])
+
+    hook_api.add_imports(*hidden_imports)
+    hook_api.add_datas(datas)
+
+
+def collect_hidden_imports(folder):
+    folder = Path(folder)
+
+    hidden_imports = set()
+    for path in folder.glob("**/*.cgx"):
+        template = path.read_text()
+        # Parse the file component into a tree of Node instances
+        parser = CGXParser()
+        parser.feed(template)
+
+        # Get the AST from the script tag
+        script_tree = get_script_ast(parser, path)
+
+        # Find a list of imported names (or aliases, if any)
+        # Those names don't have to be wrapped by `_lookup`
+        imported_names = ImportsCollector()
+        imported_names.visit(script_tree)
+
+        hidden_imports |= imported_names.names
+
+    return hidden_imports
+
+
+class ImportsCollector(ast.NodeVisitor):
+    def __init__(self):
+        self.names = set()
+
+    def visit_ImportFrom(self, node):
+        for alias in node.names:
+            self.names.add(".".join([node.module, alias.name]))
+        self.names.add(node.module)
+
+    def visit_Import(self, node):
+        for alias in node.names:
+            self.names.add(alias.name)

pyproject.toml

@@ -39,6 +39,9 @@ optional = true
 pytest-qt = "*"
 pytest-xvfb = "*"
 
+[tool.poetry.plugins.pyinstaller40]
+hook-dirs = "collagraph.__pyinstaller:get_hook_dirs"
+
 [build-system]
 requires = ["poetry>=1.0.0"]
 build-backend = "poetry.masonry.api"