RevitDevTool/samples/PythonDemo/commands/host_python_overlay_spike_script.py at develop · trgiangv/RevitDevTool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
"""Spike: inject AppData overlay into Plant attach. Do not touch PLNT3D.

Requires a pre-staged overlay (stdlib .pyd + python3.dll + packaging):
  %APPDATA%\\RevitDevTool\\overlays\\cp313-win_amd64

Re-stage from a CPython 3.13 Windows install (example uv layout)::

    $uv = ...\\cpython-3.13-windows-x86_64-none
    $o  = $env:APPDATA\\RevitDevTool\\overlays\\cp313-win_amd64
    Copy python3.dll, DLLs\\_ssl.pyd, _ctypes.pyd, _hashlib.pyd,
         libssl-3-x64.dll, libcrypto-3-x64.dll, libffi-8.dll
    uv pip install --target $o --python-version 3.13 --only-binary=:all: packaging
"""

from __future__ import annotations

import importlib
import os
import site
import sys

OVERLAY = os.path.join(
    os.environ.get("APPDATA", ""),
    "RevitDevTool",
    "overlays",
    "cp313-win_amd64",
)


def _try_import(name: str) -> str:
    try:
        mod = importlib.import_module(name)
        origin = getattr(mod, "__file__", None) or "(builtin/unknown)"
        return f"ok  {origin}"
    except Exception as ex:
        return f"FAIL {type(ex).__name__}: {ex}"


def _load_native(path: str) -> str:
    try:
        from System.Runtime.InteropServices import NativeLibrary
        NativeLibrary.Load(path)
        return "ok"
    except Exception as ex:
        return f"FAIL {type(ex).__name__}: {ex}"


def _mapped() -> None:
    try:
        from System.Diagnostics import Process
    except Exception as ex:
        print("mapped FAIL", type(ex).__name__, ex)
        return
    for module in Process.GetCurrentProcess().Modules:
        path = module.FileName or ""
        lower = path.lower()
        if "python" in lower or lower.endswith(".pyd") or "libssl" in lower or "libcrypto" in lower or "libffi" in lower:
            print(" ", path)


def main() -> None:
    print("overlay", OVERLAY, "exists=", os.path.isdir(OVERLAY))
    if not os.path.isdir(OVERLAY):
        print("stage the overlay first; abort")
        return

    python3 = os.path.join(OVERLAY, "python3.dll")
    print("LoadLibrary python3.dll", _load_native(python3))

    if hasattr(os, "add_dll_directory"):
        cookie = os.add_dll_directory(OVERLAY)
        print("add_dll_directory", OVERLAY, "cookie", cookie)
    else:
        print("add_dll_directory missing")

    if OVERLAY not in sys.path:
        site.addsitedir(OVERLAY)
    importlib.invalidate_caches()

    print()
    print("sys.path tail")
    for i, p in enumerate(sys.path):
        if i < 4:
            continue
        print(f"  [{i}] {p!r}")

    print()
    print("mapped after inject")
    _mapped()
    print()
    for name in (
        "packaging",
        "packaging.version",
        "_ctypes",
        "ctypes",
        "select",
        "_socket",
        "socket",
        "_ssl",
        "ssl",
        "_hashlib",
        "clr",
    ):
        print(f"import {name:<20} {_try_import(name)}")


if __name__ == "__main__":
    main()