mirror of
https://github.com/graphql-python/graphene.git
synced 2024-09-22 11:58:54 +03:00
28 lines
641 B
Python
28 lines
641 B
Python
import graphene
|
|
from graphene import relay
|
|
from graphene.contrib.sqlalchemy import SQLAlchemyNode, SQLAlchemyConnectionField
|
|
from models import Department as DepartmentModel, Employee as EmployeeModel
|
|
|
|
from database import db_session
|
|
|
|
schema = graphene.Schema(session=db_session)
|
|
|
|
|
|
@schema.register
|
|
class Department(SQLAlchemyNode):
|
|
class Meta:
|
|
model = DepartmentModel
|
|
|
|
|
|
@schema.register
|
|
class Employee(SQLAlchemyNode):
|
|
class Meta:
|
|
model = EmployeeModel
|
|
|
|
|
|
class Query(graphene.ObjectType):
|
|
node = relay.NodeField(Department, Employee)
|
|
all_employees = SQLAlchemyConnectionField(Employee)
|
|
|
|
schema.query = Query
|