Fix removing python class fields on auto-completion (#62)

* Fix removing fields on non-subclasses of pydantic.BaseModel and
pydantic.dataclasses.dataclass

* fix history

* add testdata

resources/META-INF/plugin.xml

@@ -1,9 +1,14 @@
 <idea-plugin url="https://github.com/koxudaxi/pydantic-pycharm-plugin">
     <id>com.koxudaxi.pydantic</id>
     <name>Pydantic</name>
-    <version>0.0.16</version>
+    <version>0.0.17</version>
     <vendor email="koaxudai@gmail.com">Koudai Aono @koxudaxi</vendor>
     <change-notes><![CDATA[
+    <h2>version 0.0.17</h2>
+    <p>BugFixes</p>
+    <ul>
+        <li>Fix removing fields on non-subclasses of pydantic.BaseModel and pydantic.dataclasses.dataclass [#62] </li>
+    </ul>
     <h2>version 0.0.16</h2>
     <p>Features, BugFixes</p>
     <ul>

src/com/koxudaxi/pydantic/PydanticCompletionContributor.kt

@@ -142,6 +142,8 @@ class PydanticCompletionContributor : CompletionContributor() {
                                          pyClass: PyClass, typeEvalContext: TypeEvalContext,
                                          excludes: HashSet<String>? = null) {
 
+            if (!isPydanticModel(pyClass)) return
+
             val fieldElements: HashSet<String> = HashSet()
 
             pyClass.getAncestorClasses(typeEvalContext)

testData/completion/pythonClass.py

@@ -0,0 +1,13 @@
+from builtins import *
+from pydantic import BaseModel
+
+class B:
+    hij: str
+
+class A:
+    abc: str
+    cde: str
+    efg: str
+
+
+A.<caret>

testSrc/com/koxudaxi/pydantic/PydanticCompletionTest.kt

@@ -284,6 +284,16 @@ open class PydanticCompletionTest : PydanticTestCase() {
         )
     }
 
+    fun testPythonClass() {
+        doFieldTest(
+                listOf(
+                        Pair("abc", "A"),
+                        Pair("cde", "A"),
+                        Pair("efg", "A")
+                )
+        )
+    }
+
     fun testField() {
         doFieldTest(
                 listOf(