add clean django-stubs folder, add some tests

django-stubs/contrib/postgres/fields/__init__.pyi

@@ -0,0 +1 @@
+from .array import ArrayField as ArrayField

django-stubs/contrib/postgres/fields/array.pyi

@@ -0,0 +1,11 @@
+from typing import List, Any
+
+from django.contrib.postgres.fields.mixins import CheckFieldDefaultMixin
+from django.db.models import Field
+
+
+class ArrayField(CheckFieldDefaultMixin, Field):
+    def __init__(self,
+                 base_field: Field,
+                 **kwargs): ...
+    def __get__(self, instance, owner) -> List[Any]: ...

django-stubs/contrib/postgres/fields/mixins.pyi

@@ -0,0 +1,2 @@
+class CheckFieldDefaultMixin:
+    pass

django-stubs/db/models/__init__.pyi

@@ -0,0 +1,6 @@
+from .base import Model as Model
+
+from .fields import (AutoField as AutoField,
+                     IntegerField as IntegerField,
+                     CharField as CharField,
+                     Field as Field)

django-stubs/db/models/base.pyi

@@ -0,0 +1,6 @@
+class ModelBase(type):
+    pass
+
+
+class Model(metaclass=ModelBase):
+    pass

django-stubs/db/models/fields/__init__.pyi

@@ -0,0 +1,26 @@
+from typing import Any
+
+from django.db.models.query_utils import RegisterLookupMixin
+
+
+class Field(RegisterLookupMixin):
+    def __init__(self,
+                 primary_key: bool = False,
+                 **kwargs): ...
+    def __get__(self, instance, owner) -> Any: ...
+
+
+class IntegerField(Field):
+    def __get__(self, instance, owner) -> int: ...
+
+
+class AutoField(Field):
+    def __get__(self, instance, owner) -> int: ...
+
+
+class CharField(Field):
+    def __init__(self,
+                 max_length: int,
+                 **kwargs): ...
+    def __get__(self, instance, owner) -> str: ...
+

django-stubs/db/models/query_utils.pyi

@@ -0,0 +1,2 @@
+class RegisterLookupMixin:
+    pass

mypy_django_plugin/plugins/postgres_fields.py

@@ -0,0 +1,11 @@
+from mypy.plugin import Plugin, ClassDefContext
+
+
+def determine_type_of_array_field(context: ClassDefContext) -> None:
+    pass
+
+
+class PostgresFieldsPlugin(Plugin):
+    def get_base_class_hook(self, fullname: str
+                            ):
+        return determine_type_of_array_field

test/test-data/basic-check.test

@@ -1,9 +0,0 @@
-[case testBasicCheck]
-from typing import Any
-
-class A:
-    pass
-
-class B(A):
-    pass
-[out]

test/test-data/check-model-fields.test

@@ -0,0 +1,12 @@
+[case testBasicModelFields]
+from django.db import models
+
+
+class User(models.Model):
+    id = models.AutoField(primary_key=True)
+    name = models.CharField(max_length=255)
+
+user = User()
+reveal_type(user.id)  # E: Revealed type is 'builtins.int'
+reveal_type(user.name)  # E: Revealed type is 'builtins.str'
+[out]

test/test-data/check-postgres-fields.test

@@ -0,0 +1,13 @@
+[case testArrayField]
+import typing
+
+from django.db import models
+from django.contrib.postgres.fields import ArrayField
+
+
+class User(models.Model):
+    members = ArrayField(base_field=models.IntegerField())
+
+user = User()
+reveal_type(user.members)  # E: Revealed type is 'typing.List[int]'
+[out]

test/testdjango.py

@@ -13,6 +13,7 @@ TEST_DATA_DIR = ROOT_DIR / 'test' / 'test-data'
 
 class DjangoTestSuite(DataSuite):
     files = [
+        'check-model-fields.test',
         'check-postgres-fields.test'
     ]
     data_prefix = str(TEST_DATA_DIR)