Collect object_types

djlsp/scripts/django-collector.py

@@ -14,6 +14,7 @@
 from django.template.engine import Engine
 from django.template.library import InvalidTemplateLibrary
 from django.template.utils import get_app_template_dirs
+from django.test import RequestFactory
 from django.urls import URLPattern, URLResolver
 
 # Some tags are added with a Node, like end*, elif else.
@@ -112,6 +113,19 @@ def get_file_watcher_globs():
     return patterns
 
 
+def get_model_data():
+    pass
+
+
+def get_object_types():
+    models = apps.get_models()
+    object_types = {}
+    for model in models:
+        model_path = model.__module__ + "." + model.__name__
+        object_types[model_path] = {field.name: None for field in model._meta.fields}
+    return object_types
+
+
 def get_static_files():
     # TODO: Add option to ignore some static folders
     # (like static that is generated with a JS bundler)
@@ -282,6 +296,7 @@ def collect_project_data():
         "urls": get_urls(),
         "libraries": get_libraries(),
         "templates": get_templates(),
+        "object_types": get_object_types(),
     }