@@ -14,33 +14,18 @@ class Meta:
1414 interfaces = (relay .Node , )
1515
1616
17- class DepartmentConnection (relay .Connection ):
18- class Meta :
19- node = Department
20-
21-
2217class Employee (SQLAlchemyObjectType ):
2318 class Meta :
2419 model = EmployeeModel
2520 interfaces = (relay .Node , )
2621
2722
28- class EmployeeConnection (relay .Connection ):
29- class Meta :
30- node = Employee
31-
32-
3323class Role (SQLAlchemyObjectType ):
3424 class Meta :
3525 model = RoleModel
3626 interfaces = (relay .Node , )
3727
3828
39- class RoleConnection (relay .Connection ):
40- class Meta :
41- node = Role
42-
43-
4429SortEnumEmployee = utils .sort_enum_for_model (EmployeeModel , 'SortEnumEmployee' ,
4530 lambda c , d : c .upper () + ('_ASC' if d else '_DESC' ))
4631
@@ -49,14 +34,14 @@ class Query(graphene.ObjectType):
4934 node = relay .Node .Field ()
5035 # Allow only single column sorting
5136 all_employees = SQLAlchemyConnectionField (
52- EmployeeConnection ,
37+ Employee ,
5338 sort = graphene .Argument (
5439 SortEnumEmployee ,
5540 default_value = utils .EnumValue ('id_asc' , EmployeeModel .id .asc ())))
5641 # Allows sorting over multiple columns, by default over the primary key
57- all_roles = SQLAlchemyConnectionField (RoleConnection )
42+ all_roles = SQLAlchemyConnectionField (Role )
5843 # Disable sorting over this field
59- all_departments = SQLAlchemyConnectionField (DepartmentConnection , sort = None )
44+ all_departments = SQLAlchemyConnectionField (Department , sort = None )
6045
6146
6247schema = graphene .Schema (query = Query , types = [Department , Employee , Role ])
0 commit comments