feat: return m2m fields

Author: freemindcore

easy/domain/orm.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Tuple, Type
+from typing import Any, Dict, List, Tuple, Type
 
 from django.db import models
 from django.db.models.query import QuerySet
@@ -14,8 +14,13 @@
 class CrudModel(object):
     def __init__(self, model: Type[models.Model]):
         self.model = model
+        self.m2m_fields = [
+            f
+            for f in self.model._meta.get_fields(include_hidden=True)
+            if isinstance(f, models.ManyToManyField)
+        ]
 
-    def __get_fields(self, payload: dict) -> Tuple[dict, dict]:
+    def __get_fields(self, payload: Dict) -> Tuple[Dict, Dict]:
         m2m_fields = {}
         local_fields = {}
         for field in payload.keys():
@@ -27,12 +32,12 @@ def __get_fields(self, payload: dict) -> Tuple[dict, dict]:
         return local_fields, m2m_fields
 
     # Define BASE CRUD
-    def _crud_add_obj(self, **payload: dict) -> Any:
+    def _crud_add_obj(self, **payload: Dict) -> Any:
         local_fields, m2m_fields = self.__get_fields(payload)
         obj = self.model.objects.create(**local_fields)
         if m2m_fields:
             for field, value in m2m_fields.items():
-                if value and isinstance(value, list):
+                if value and isinstance(value, List):
                     m2m_f = getattr(obj, field)
                     m2m_f.set(value)
         return obj
@@ -47,7 +52,7 @@ def _crud_del_obj(self, id: int) -> "BaseApiResponse":
                 {"Detail": "Not found."}, message="Not found."
             )  # pragma: no cover
 
-    def _crud_update_obj(self, id: int, payload: dict) -> "BaseApiResponse":
+    def _crud_update_obj(self, id: int, payload: Dict) -> "BaseApiResponse":
         local_fields, m2m_fields = self.__get_fields(payload)
         try:
             obj, created = self.model.objects.update_or_create(
@@ -64,7 +69,17 @@ def _crud_update_obj(self, id: int, payload: dict) -> "BaseApiResponse":
         return BaseApiResponse({"id": obj.id, "created": created})
 
     def _crud_get_obj(self, id: int) -> Any:
-        return get_object_or_none(self.model, id=id)
+        if self.m2m_fields:
+            qs = self.model.objects.filter(id=id).prefetch_related(
+                self.m2m_fields[0].name
+            )
+            for f in self.m2m_fields[1:]:
+                qs = qs.prefetch_related(f.name)
+        else:
+            qs = self.model.objects.filter(id=id)
+        if qs:
+            return qs.first()
+        return BaseApiResponse()
 
     def _crud_get_objs_all(self, maximum: int = None, **filters: Any) -> Any:
         """

easy/domain/serializers.py

@@ -46,7 +46,6 @@ def serialize_model_instance(
     """Serializes Django model instance to dictionary"""
     out = {}
     for field in obj._meta.get_fields():
-        # TODO: Add OneToOne relationship??
         if is_one_relationship(field):
             out.update(serialize_foreign_key(obj, field, referrers + (obj,)))
 

tests/demo_app/models.py

@@ -35,6 +35,10 @@ class Event(TestBaseModel):
 
     owner = models.ManyToManyField(to=Client, related_name="events", null=True)
 
+    lead_owner = models.ManyToManyField(to=Client, related_name="lead_owner", null=True)
+
+    boss_owner = models.ManyToManyField(to=Client, related_name="boss_owner", null=True)
+
     type = models.ForeignKey(Type, on_delete=models.CASCADE, null=True)
 
     def __str__(self):