hack: try out some gql annotation ideas

gifter/graphql.py

@@ -0,0 +1,67 @@
+from . import models
+
+import inspect, sys
+from pprint import pprint
+
+schema = {}
+
+
+class String:
+ pass
+
+
+class Int:
+ pass
+
+
+class UUID:
+ pass
+
+
+class JSON:
+ pass
+
+
+class DateTime:
+ pass
+
+
+class Boolean:
+ pass
+
+
+for name, obj in inspect.getmembers(sys.modules[models.__name__]):
+ if inspect.isclass(obj):
+ if hasattr(obj, "_meta") and not obj._meta.abstract:
+ if not hasattr(obj, 'GQLMeta'):
+ continue
+
+ obj_gql_name = f"{obj.__name__}Type"
+ schema[obj_gql_name] = {}
+ fields = obj._meta.fields
+ for field in fields:
+
+ if not hasattr(field, "__graphql_enabled"):
+ continue
+
+ match field.get_internal_type():
+ case "CharField" | "TextField":
+ schema[obj_gql_name][field.name] = String
+ case "IntegerField":
+ schema[obj_gql_name][field.name] = Int
+ case "UUIDField":
+ schema[obj_gql_name][field.name] = UUID
+ case "JSONField":
+ schema[obj_gql_name][field.name] = JSON
+ case "DateTimeField":
+ schema[obj_gql_name][field.name] = DateTime
+ case "BooleanField":
+ schema[obj_gql_name][field.name] = Boolean
+ case "ForeignKey" | "AutoField":
+ pprint(field)
+ case _:
+ raise Exception(f"Unknown field type: {field.get_internal_type()}")
+
+
+def foo():
+ pprint(schema)

gifter/models.py

@@ -1,4 +1,6 @@
 import hashlib
+from dataclasses import dataclass
+from typing import Any
 import uuid
 import secrets
 
@@ -84,8 +86,19 @@ def __str__(self):
  return self.display_name
 
 
+def with_graphql(field):
+ setattr(field, "__graphql_enabled", True)
+ return field
+
+
 class Wishlist(CommonBaseClass):
- title = models.CharField(max_length=120, blank=False, null=False)
+
+ @dataclass
+ class GQLMeta:
+ enabled = True
+ auth_check_via = "groups"
+
+ title = with_graphql(models.CharField(max_length=120, blank=False, null=False))
  owner = models.ForeignKey(User, on_delete=models.CASCADE)
  group = models.ForeignKey(Group, on_delete=models.CASCADE)
 

gifter/test_graphql.py

@@ -0,0 +1,7 @@
+from django.test import TestCase
+from . import graphql
+
+class TestGraphql(TestCase):
+
+ def test_graphql(self):
+ graphql.foo()