docs(Example): Update example.

Author: abawchen

examples/flask_mongoengine/app.py

@@ -17,10 +17,27 @@
           id,
           name
         },
-        role {
+        roles {
+          edges {
+            node {
+              id,
+              name
+            }
+          }
+        },
+        leader {
           id,
           name
         }
+        tasks {
+          edges {
+            node {
+              id,
+              name,
+              deadline
+            }
+          }
+        }
       }
     }
   }

examples/flask_mongoengine/database.py

@@ -1,6 +1,6 @@
 from mongoengine import connect
 
-from models import Department, Employee, Role
+from models import Department, Employee, Role, Task
 
 connect('graphene-mongo-example', host='mongomock://localhost', alias='default')
 
@@ -19,12 +19,31 @@ def init_db():
     engineer = Role(name='engineer')
     engineer.save()
 
-    peter = Employee(name='Peter', department=engineering, role=engineer)
-    peter.save()
-
-    roy = Employee(name='Roy', department=engineering, role=engineer)
-    roy.save()
+    debug = Task(name='Debug')
+    test = Task(name='Test')
 
-    tracy = Employee(name='Tracy', department=hr, role=manager)
+    tracy = Employee(
+        name='Tracy',
+        department=hr,
+        roles=[engineer, manager],
+        tasks=[]
+    )
     tracy.save()
 
+    peter = Employee(
+        name='Peter',
+        department=engineering,
+        leader=tracy,
+        roles=[engineer],
+        tasks=[debug, test]
+    )
+    peter.save()
+
+    roy = Employee(
+        name='Roy',
+        department=engineering,
+        leader=tracy,
+        roles=[engineer],
+        tasks=[debug]
+    )
+    roy.save()

examples/flask_mongoengine/models.py

@@ -1,24 +1,35 @@
 from datetime import datetime
-from mongoengine import Document
+from mongoengine import Document, EmbeddedDocument
 from mongoengine.fields import (
-    DateTimeField, ReferenceField, StringField,
+    DateTimeField, EmbeddedDocumentField,
+    ListField, ReferenceField, StringField,
 )
 
 
 class Department(Document):
+
     meta = {'collection': 'department'}
     name = StringField()
 
 
 class Role(Document):
+
     meta = {'collection': 'role'}
     name = StringField()
 
 
+class Task(EmbeddedDocument):
+
+    name = StringField()
+    deadline = DateTimeField(default=datetime.now)
+
+
 class Employee(Document):
+
     meta = {'collection': 'employee'}
     name = StringField()
     hired_on = DateTimeField(default=datetime.now)
     department = ReferenceField(Department)
-    role = ReferenceField(Role)
-
+    roles = ListField(ReferenceField(Role))
+    leader = ReferenceField('Employee')
+    tasks = ListField(EmbeddedDocumentField(Task))

examples/flask_mongoengine/schema.py

@@ -4,6 +4,7 @@
 from models import Department as DepartmentModel
 from models import Employee as EmployeeModel
 from models import Role as RoleModel
+from models import Task as TaskModel
 
 class Department(MongoengineObjectType):
 
@@ -19,6 +20,12 @@ class Meta:
         interfaces = (Node,)
 
 
+class Task(MongoengineObjectType):
+
+    class Meta:
+        model = TaskModel
+        interfaces = (Node,)
+
 class Employee(MongoengineObjectType):
 
     class Meta: