From cfc8d927bc3c2ac818802c3958da99e9b826480f Mon Sep 17 00:00:00 2001
From: Andrey Vlasovskikh <andrey.vlasovskikh@jetbrains.com>
Date: Thu, 18 Apr 2013 15:49:52 +0400
Subject: [PATCH] Added PyClass.getAncestorClasses(TypeEvalContext)

---
 python/psi-api/src/com/jetbrains/python/psi/PyClass.java  | 3 +++
 python/src/com/jetbrains/python/psi/impl/PyClassImpl.java | 8 +++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/python/psi-api/src/com/jetbrains/python/psi/PyClass.java b/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
index 26553ee50951..c5054bee8cee 100644
--- a/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
+++ b/python/psi-api/src/com/jetbrains/python/psi/PyClass.java
@@ -113,6 +113,9 @@ public interface PyClass extends PsiNameIdentifierOwner, PyStatement, NameDefine
   @NotNull
   List<PyClass> getAncestorClasses();
 
+  @NotNull
+  List<PyClass> getAncestorClasses(@NotNull TypeEvalContext context);
+
   /**
    * Get a list of all ancestor types.
    */
diff --git a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
index a9881353f251..1f0243284ebe 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyClassImpl.java
@@ -177,8 +177,14 @@ public class PyClassImpl extends PyPresentableElementImpl<PyClassStub> implement
   @NotNull
   @Override
   public List<PyClass> getAncestorClasses() {
+    return getAncestorClasses(TypeEvalContext.fastStubOnly(null));
+  }
+
+  @NotNull
+  @Override
+  public List<PyClass> getAncestorClasses(@NotNull TypeEvalContext context) {
     final List<PyClass> results = new ArrayList<PyClass>();
-    for (PyClassLikeType type : getAncestorTypes(TypeEvalContext.fastStubOnly(null))) {
+    for (PyClassLikeType type : getAncestorTypes(context)) {
       if (type instanceof PyClassType) {
         results.add(((PyClassType)type).getPyClass());
       }