new semanal wip 1

mypy_django_plugin_newsemanal/transformers/init_create.py

@@ -0,0 +1,35 @@
+from typing import cast
+
+from mypy.checker import TypeChecker
+from mypy.nodes import Argument, Var, ARG_NAMED
+from mypy.plugin import FunctionContext
+from mypy.types import Type as MypyType, Instance
+
+from mypy_django_plugin_newsemanal.context import DjangoContext
+from mypy_django_plugin_newsemanal.lib import helpers
+
+
+def redefine_and_typecheck_model_init(ctx: FunctionContext, django_context: DjangoContext) -> MypyType:
+    assert isinstance(ctx.default_return_type, Instance)
+
+    api = cast(TypeChecker, ctx.api)
+
+    model_info = ctx.default_return_type.type
+    model_cls = django_context.get_model_class_by_fullname(model_info.fullname())
+
+    # expected_types = {}
+    # for field in model_cls._meta.get_fields():
+    #     field_fullname = helpers.get_class_fullname(field.__class__)
+    #     field_info = api.lookup_typeinfo(field_fullname)
+    #     field_set_type = helpers.get_private_descriptor_type(field_info, '_pyi_private_set_type',
+    #                                                          is_nullable=False)
+    # field_kwarg = Argument(variable=Var(field.attname, field_set_type),
+    #                        type_annotation=field_set_type,
+    #                        initializer=None,
+    #                        kind=ARG_NAMED)
+    # expected_types[field.attname] = field_set_type
+    # for field_name, field in model_cls._meta.fields_map.items():
+    #     print()
+
+    # print()
+    return ctx.default_return_type