Improved django example

Author: syrusakbary

examples/starwars_django/schema.py

@@ -12,7 +12,6 @@
 
 
 class Ship(DjangoNode):
-
     class Meta:
         model = ShipModel
 
@@ -21,15 +20,12 @@ def get_node(cls, id):
         return Ship(get_ship(id))
 
 
-@schema.register
 class Character(DjangoObjectType):
-
     class Meta:
         model = CharacterModel
 
 
 class Faction(DjangoNode):
-
     class Meta:
         model = FactionModel
 
@@ -39,7 +35,6 @@ def get_node(cls, id):
 
 
 class IntroduceShip(relay.ClientIDMutation):
-
     class Input:
         ship_name = graphene.StringField(required=True)
         faction_id = graphene.StringField(required=True)
@@ -79,5 +74,9 @@ class Mutation(graphene.ObjectType):
     introduce_ship = graphene.Field(IntroduceShip)
 
 
+# We register the Character Model because if not would be
+# inaccessible for the schema
+schema.register(Character)
+
 schema.query = Query
 schema.mutation = Mutation