@@ -12,33 +12,18 @@ class Meta:
1212 interfaces = (relay .Node , )
1313
1414
15- class DepartmentConnection (relay .Connection ):
16- class Meta :
17- node = Department
18-
19-
2015class Employee (SQLAlchemyObjectType ):
2116 class Meta :
2217 model = EmployeeModel
2318 interfaces = (relay .Node , )
2419
2520
26- class EmployeeConnection (relay .Connection ):
27- class Meta :
28- node = Employee
29-
30-
3121class Role (SQLAlchemyObjectType ):
3222 class Meta :
3323 model = RoleModel
3424 interfaces = (relay .Node , )
3525
3626
37- class RoleConnection (relay .Connection ):
38- class Meta :
39- node = Role
40-
41-
4227SortEnumEmployee = utils .sort_enum_for_model (EmployeeModel , 'SortEnumEmployee' ,
4328 lambda c , d : c .upper () + ('_ASC' if d else '_DESC' ))
4429
@@ -47,14 +32,14 @@ class Query(graphene.ObjectType):
4732 node = relay .Node .Field ()
4833 # Allow only single column sorting
4934 all_employees = SQLAlchemyConnectionField (
50- EmployeeConnection ,
35+ Employee ,
5136 sort = graphene .Argument (
5237 SortEnumEmployee ,
5338 default_value = utils .EnumValue ('id_asc' , EmployeeModel .id .asc ())))
5439 # Allows sorting over multiple columns, by default over the primary key
55- all_roles = SQLAlchemyConnectionField (RoleConnection )
40+ all_roles = SQLAlchemyConnectionField (Role )
5641 # Disable sorting over this field
57- all_departments = SQLAlchemyConnectionField (DepartmentConnection , sort = None )
42+ all_departments = SQLAlchemyConnectionField (Department , sort = None )
5843
5944
6045schema = graphene .Schema (query = Query , types = [Department , Employee , Role ])
0 commit comments