VimPlug/django-stubs
1
0
Fork 0
mirror of https://github.com/davidhalter/django-stubs.git synced 2026-08-13 19:31:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add custom Field processing for mixins used in the Model subclasses (#167)

Browse Source
This commit is contained in:
Maxim Kurnikov
2019-09-18 01:40:41 +03:00
committed by GitHub
parent 64720f492c
commit 23ad65033b
4 changed files with 42 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mypy_django_plugin/transformers/models.py
+14 -1
View File
@@ -212,6 +212,18 @@ class AddMetaOptionsAttribute(ModelClassInitializer):
]))
class RecordAllModelMixins(ModelClassInitializer):
def run(self) -> None:
basemodel_info = self.lookup_typeinfo_or_incomplete_defn_error(fullnames.MODEL_CLASS_FULLNAME)
basemodel_metadata = helpers.get_django_metadata(basemodel_info)
if 'model_mixins' not in basemodel_metadata:
basemodel_metadata['model_mixins'] = {}
for base_info in self.model_classdef.info.mro[1:]:
if base_info.fullname() != 'builtins.object':
basemodel_metadata['model_mixins'][base_info.fullname()] = 1
def process_model_class(ctx: ClassDefContext,
django_context: DjangoContext) -> None:
initializers = [
@@ -220,7 +232,8 @@ def process_model_class(ctx: ClassDefContext,
AddRelatedModelsId,
AddManagers,
AddExtraFieldMethods,
AddMetaOptionsAttribute
AddMetaOptionsAttribute,
RecordAllModelMixins,
]
for initializer_cls in initializers:
try: