pythonnative/src/pythonnative/refresh.py at v0.43.1 · pythonnative/pythonnative · GitHub

Repository navigation

History

55 lines (50 loc) · 2.13 KB

Raw

"""Conservative component compatibility for Fast Refresh."""

from __future__ import annotations

import ast

import inspect

import textwrap

from typing import Any, Callable

def hook_signature(function: Callable[..., Any], _seen: frozenset[int] = frozenset()) -> tuple[str, ...] | None:

"""Record hook order and binding names before the source file changes.

Binding names distinguish inserting or moving hooks of the same kind.

Uninspectable functions remount instead of risking slot corruption.

"""

if id(function) in _seen:

return None

_seen = _seen | {id(function)}

try:

tree = ast.parse(textwrap.dedent(inspect.getsource(function)))

except (OSError, TypeError, SyntaxError):

return None

result = []

parents = {id(child): node for node in ast.walk(tree) for child in ast.iter_child_nodes(node)}

for node in ast.walk(tree):

if not isinstance(node, ast.Call):

continue

name = (

node.func.id

if isinstance(node.func, ast.Name)

else node.func.attr if isinstance(node.func, ast.Attribute) else ""

)

if not name.startswith("use_"):

continue

owner = parents.get(id(node))

binding = (

ast.dump(owner.targets[0])

if isinstance(owner, ast.Assign)

else ast.dump(owner.target) if isinstance(owner, ast.AnnAssign) else ""

)

signature = name + ":" + binding

target = None

if isinstance(node.func, ast.Name):

target = function.__globals__.get(node.func.id)

elif isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name):

owner = function.__globals__.get(node.func.value.id)

target = getattr(owner, node.func.attr, None)

if inspect.isfunction(target) and not target.__module__.startswith("pythonnative."):

nested = hook_signature(target, _seen)

if nested is None:

return None

signature += repr(nested)

result.append((node.lineno, node.col_offset, signature))

return tuple(value for _, _, value in sorted(result))

pythonnative/src/pythonnative/refresh.py at v0.43.1 · pythonnative/pythonnative · GitHub

Repository navigation

Expand file tree

Breadcrumbs

refresh.py

Latest commit

History

Breadcrumbs

refresh.py

File metadata and controls