fix(schema): add missing models

pollination · Dec 11, 2020 · d9dd695 · d9dd695
1 parent b50912c
commit d9dd695
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/gen_schemas.py b/gen_schemas.py
@@ -9,8 +9,8 @@
 from pydantic_openapi_helper.inheritance import class_mapper
 
 from queenbee.repository import RepositoryIndex
-from queenbee.job import Job
-from queenbee.recipe import Recipe
+from queenbee.job import Job, JobStatus
+from queenbee.recipe import Recipe, RecipeInterface
 from queenbee.plugin import Plugin
 
 folder = os.path.join(os.path.dirname(__file__), 'docs/_static/schemas')
@@ -101,8 +101,9 @@
     "url": "./queenbee_inheritance.json"
 }
 
+models = [Recipe, Plugin, Job, RepositoryIndex, RecipeInterface, JobStatus]
 openapi = get_openapi(
-    [Recipe, Plugin, Job],
+    models,
     title='Queenbee Schema',
     description='Documentation for Queenbee schema.',
     version=VERSION, info=info,
@@ -113,7 +114,7 @@
 
 # with inheritance
 openapi = get_openapi(
-    [Recipe, Plugin, Job],
+    models,
     title='Queenbee Schema with Inheritance',
     description='Documentation for Queenbee schema.',
     version=VERSION, info=info,
@@ -127,7 +128,7 @@
 with open(os.path.join(folder, 'queenbee_mapper.json'), 'w') as out_file:
     json.dump(
         class_mapper(
-            [Recipe, Plugin, Job],
+            models,
             ['queenbee', 'queenbee.interface']
         ),
         out_file, indent=2

diff --git a/queenbee/job/__init__.py b/queenbee/job/__init__.py
@@ -1 +1,2 @@
 from .job import Job, JobArguments
+from .status import JobStatus
diff --git a/queenbee/recipe/__init__.py b/queenbee/recipe/__init__.py
@@ -1 +1 @@
-from .recipe import Recipe, BakedRecipe
+from .recipe import Recipe, BakedRecipe, RecipeInterface