[REFACTOR] memindahkan fungsi get detil ke model

sizakat/mustahik/models.py

@@ -51,6 +51,14 @@ class DataSource(models.Model):
 
     category = models.CharField(max_length=32, choices=Category.choices)
 
+    def get_source_detail(self):
+        if self.category == DataSource.Category.INSTITUSI:
+            return DataSourceInstitusi.objects.get(data_source=self)
+        if self.category == DataSource.Category.PEKERJA:
+            return DataSourcePekerja.objects.get(data_source=self)
+        if self.category == DataSource.Category.WARGA:
+            return DataSourceWarga.objects.get(data_source=self)
+
 
 class DataSourceDetail(models.Model):
     class Meta:

sizakat/mustahik/types.py

@@ -45,12 +45,6 @@ class DataSourceType(DjangoObjectType):
             'datasourceinstitusi', 'datasourcepekerja', 'datasourcewarga'
         )
 
-    data_source_detail = graphene.Field(DataSourceDetailType)
-
-    def resolve_data_source_detail(self, info):
-        if self.category == DataSource.Category.INSTITUSI:
-            return DataSourceInstitusi.objects.get(data_source__pk=self.pk)
-        if self.category == DataSource.Category.PEKERJA:
-            return DataSourcePekerja.objects.get(data_source__pk=self.pk)
-        if self.category == DataSource.Category.WARGA:
-            return DataSourceWarga.objects.get(data_source__pk=self.pk)
+    data_source_detail = graphene.Field(
+        DataSourceDetailType, source='get_source_detail'
+    )

sizakat/schema.py

@@ -2,7 +2,11 @@ import graphene
 
 from graphene_django import DjangoObjectType
 
-from .mustahik.mutations import MustahikMutation, DeleteMustahik, DataSourceMutation, DataSourceWargaMutation, DataSourceInstitusiMutation, DataSourcePekerjaMutation
+from .mustahik.mutations import (
+    MustahikMutation, DeleteMustahik, DataSourceMutation,
+    DataSourceWargaMutation, DataSourceInstitusiMutation,
+    DataSourcePekerjaMutation
+)
 from .mustahik.query import MustahikQuery
 
 ABOUT = ('Si Zakat merupakan sistem informasi untuk membantu masjid dalam '