fix(python): preserve inheritance fact compatibility

Fixing regression in the golden test.

Author: jnthnhrrr

internal/extractors/pythonextractor/python_ast.go

@@ -477,6 +477,7 @@ func (w *pyWalker) handleFromImport(node *sitter.Node) {
 	// flag them as orphans. Kept only here to avoid bloating every from-import.
 	isInit := w.relFile == "__init__.py" || strings.HasSuffix(w.relFile, "/__init__.py")
 	var reexported []string
+	bindings := make(map[string]string)
 
 	// Map each imported name to a resolvable target or "" (external).
 	for i := uint(0); i < uint(node.ChildCount()); i++ {
@@ -501,6 +502,9 @@ func (w *pyWalker) handleFromImport(node *sitter.Node) {
 			importedName = pyText(c, w.src)
 			localName = importedName
 		}
+		if localName != "" && importedName != "" && importedName != "*" {
+			bindings[localName] = importedName
+		}
 
 		if isInit && importedName != "" && importedName != "*" {
 			reexported = append(reexported, importedName)
@@ -522,6 +526,9 @@ func (w *pyWalker) handleFromImport(node *sitter.Node) {
 		}
 	}
 
+	if len(bindings) > 0 {
+		depProps["bindings"] = bindings
+	}
 	if len(reexported) > 0 {
 		depProps["reexports"] = reexported
 	}
@@ -795,9 +802,6 @@ func (w *pyWalker) handleClass(node *sitter.Node, decorators []string) {
 			switch c.Kind() {
 			case "identifier":
 				base := pyText(c, w.src)
-				if imported := w.importMap[base]; imported != "" {
-					base = imported
-				}
 				bases = append(bases, base)
 				rels = append(rels, facts.Relation{Kind: facts.RelImplements, Target: base})
 			case "attribute":

internal/extractors/pythonextractor/resolve.go

@@ -119,7 +119,30 @@ func resolveImplementsTargets(allFacts []facts.Fact, fileModules map[string]bool
 	reexports := buildReexportIndex(allFacts, pkgDirs)
 	symbols := make(map[string]bool)
 	byShortName := make(map[string][]string)
+	importedByFile := make(map[string]map[string]string)
 	for i := range allFacts {
+		if allFacts[i].Kind == facts.KindDependency && allFacts[i].Props["from"] == true {
+			var module string
+			for _, rel := range allFacts[i].Relations {
+				if rel.Kind == facts.RelImports {
+					module = rel.Target
+					break
+				}
+			}
+			if module != "" {
+				if importedByFile[allFacts[i].File] == nil {
+					importedByFile[allFacts[i].File] = make(map[string]string)
+				}
+				for local, imported := range stringMapProp(allFacts[i].Props, "bindings") {
+					target := module + "." + imported
+					if previous, exists := importedByFile[allFacts[i].File][local]; exists && previous != target {
+						importedByFile[allFacts[i].File][local] = ""
+					} else {
+						importedByFile[allFacts[i].File][local] = target
+					}
+				}
+			}
+		}
 		if allFacts[i].Kind != facts.KindSymbol {
 			continue
 		}
@@ -131,6 +154,11 @@ func resolveImplementsTargets(allFacts []facts.Fact, fileModules map[string]bool
 		}
 		byShortName[short] = append(byShortName[short], name)
 	}
+	for i := range allFacts {
+		if allFacts[i].Kind == facts.KindDependency && allFacts[i].Props != nil {
+			delete(allFacts[i].Props, "bindings")
+		}
+	}
 
 	for i := range allFacts {
 		f := &allFacts[i]
@@ -151,6 +179,12 @@ func resolveImplementsTargets(allFacts []facts.Fact, fileModules map[string]bool
 				}
 				continue
 			}
+			if imported := importedByFile[f.File][rel.Target]; imported != "" {
+				if resolved, keep := resolveDottedTarget(imported, fileIdx, topPkgs, fileDir(f.File), reexports, symbols); keep && symbols[resolved] {
+					rel.Target = resolved
+				}
+				continue
+			}
 			candidates := byShortName[rel.Target]
 			if len(candidates) == 1 {
 				rel.Target = candidates[0]
@@ -288,6 +322,23 @@ func stringSliceProp(props map[string]any, key string) []string {
 	return nil
 }
 
+func stringMapProp(props map[string]any, key string) map[string]string {
+	out := make(map[string]string)
+	switch v := props[key].(type) {
+	case map[string]string:
+		for k, value := range v {
+			out[k] = value
+		}
+	case map[string]any:
+		for k, value := range v {
+			if s, ok := value.(string); ok {
+				out[k] = s
+			}
+		}
+	}
+	return out
+}
+
 // resolveDottedTarget maps a dotted call target ("a.b.c.sym") to a canonical slash
 // symbol name when its module prefix resolves to an internal file or to a package
 // that re-exports the symbol, keeps it dotted when the prefix is internal but