Understand API 调试实录：深入追踪 Python @overload 调用链

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 示例
@overload
def func(a: int) -> int: ...
@overload
def func(a: str) -> str: ...

def func(a: int | str) -> int | str:
  # 实现
  if isinstance(a, int):
    return a + 1
  else:
    return a + "a"

caller_func():
  func(1) # 调用点 1
  func("b") # 调用点 2

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
# (在 build_chains_recursive 内部)
            if not direct_refs:
                # --- 最终变通逻辑：全局查找模糊实体，查询其 Callby/Useby ---
                print(f"[INFO] No direct callers for implementation {ent.longname()} (ID: {ent.id()}). Attempting Ambiguous Attribute workaround...")
                potential_ambiguous_ent = None
                implementation_ent = ent
                parent = implementation_ent.parent()

                # 步骤 1: 全局查找同名、同父级、类型为 Ambiguous 的实体
                if parent:
                    target_name = implementation_ent.name()
                    try:
                        all_named_ents = self.find_entities_by_name(target_name, "Function, Method, Attribute, Unknown")
                        for sib in all_named_ents:
                            sib_parent = sib.parent()
                            if sib_parent and sib_parent.id() == parent.id() and sib.id() != implementation_ent.id():
                                 if "Ambiguous" in sib.kindname():
                                      potential_ambiguous_ent = sib
                                      print(f"[INFO]   Found Ambiguous Attribute sibling via global lookup: {potential_ambiguous_ent.longname()} (ID: {potential_ambiguous_ent.id()})")
                                      break
                        if not potential_ambiguous_ent: print(f"[INFO]   Global lookup did not find an 'Ambiguous Attribute' sibling.")
                    except Exception as e_lookup: print(f"[WARN] Error during global lookup for ambiguous sibling: {e_lookup}")
                else: print("[WARN]   Cannot find ambiguous sibling because implementation parent is None.")

                # 步骤 2: 如果找到模糊实体，查询它的 Callby, Useby 引用
                if potential_ambiguous_ent:
                    caller_ref_kinds_on_ambiguous = "Callby,Useby" # <-- 使用这个过滤器！
                    print(f"[INFO]   Querying Ambiguous Attribute {potential_ambiguous_ent.id()} for incoming '{caller_ref_kinds_on_ambiguous}' references...")
                    try:
                        ambiguous_incoming_refs = potential_ambiguous_ent.refs(caller_ref_kinds_on_ambiguous)
                        valid_caller_refs = []
                        if ambiguous_incoming_refs:
                            # ... (过滤 valid_caller_refs 的逻辑不变) ...
                            for r in ambiguous_incoming_refs:
                                source = r.ent()
                                if source and ("Function" in source.kindname() or "Method" in source.kindname()):
                                    valid_caller_refs.append(r)
                        print(f"[INFO]   Found {len(valid_caller_refs)} valid incoming refs pointing TO Ambiguous Attribute (using filter '{caller_ref_kinds_on_ambiguous}').")

                        if valid_caller_refs:
                            refs_to_process = valid_caller_refs
                            processed_via_stub = True # 标记为间接
                            node_info["note"] = f"Callers inferred via Ambiguous Attribute {potential_ambiguous_ent.id()} (queried '{caller_ref_kinds_on_stub}')"
                        else:
                            print(f"[INFO]   No valid incoming refs found for Ambiguous Attribute with filter '{caller_ref_kinds_on_stub}'.")
                            # refs_to_process 保持为空
                    except understand.UnderstandError as e_amb_call:
                         print(f"[WARN] Error querying Ambiguous Attribute for '{caller_ref_kinds_on_stub}' refs: {e_amb_call}")
                else:
                    print("[INFO] Workaround failed: Could not find associated Ambiguous Attribute entity.")
                    # refs_to_process 保持为空
                # --- 变通逻辑结束 ---
            # ... (else: direct_refs 存在 的逻辑不变) ...
            # ... (处理 refs_to_process 的循环不变) ...