@@ -21,8 +21,8 @@ class SalaryModel(pydantic.BaseModel):
21
21
22
22
23
23
class EmployeeModel (PersonModel ):
24
- hired_on : datetime .datetime = None
25
- salary : T .Optional [SalaryModel ]
24
+ hired_on : T . Optional [ datetime .datetime ] = None
25
+ salary : T .Optional [SalaryModel ] = None
26
26
27
27
28
28
class ManagerModel (EmployeeModel ):
@@ -96,13 +96,13 @@ def resolve_list_departments(self, info):
96
96
salary = SalaryModel (rating = "GS-9" , amount = 75000.23 ),
97
97
hired_on = datetime .datetime (2019 , 1 , 1 , 15 , 26 ),
98
98
),
99
- EmployeeModel (id = uuid .uuid4 (), name = "Derek" ),
99
+ EmployeeModel (id = uuid .uuid4 (), name = "Derek" , salary = None ),
100
100
],
101
101
)
102
102
]
103
103
104
104
105
- if __name__ == "__main__" :
105
+ def main () :
106
106
schema = graphene .Schema (query = Query )
107
107
query = """
108
108
query {
@@ -128,7 +128,10 @@ def resolve_list_departments(self, info):
128
128
}
129
129
}
130
130
"""
131
- result = schema .execute (query )
131
+ return schema .execute (query )
132
+
132
133
134
+ if __name__ == "__main__" :
135
+ result = main ()
133
136
print (result .errors )
134
137
print (json .dumps (result .data , indent = 2 ))
0 commit comments