From 2db5707a2fc6170177e76666b8cf9cc59e3bafc6 Mon Sep 17 00:00:00 2001
From: Mostapha <sadeghipour@gmail.com>
Date: Thu, 10 Dec 2020 19:46:37 -0500
Subject: [PATCH] fix(schema): add missing models

---
 gen_schemas.py              | 11 ++++++-----
 queenbee/job/__init__.py    |  1 +
 queenbee/recipe/__init__.py |  2 +-
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/gen_schemas.py b/gen_schemas.py
index 86db7b79..f0c7a6b2 100644
--- 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
index 55605f0a..80b2c73d 100644
--- 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
index e8206b1d..acec7040 100644
--- a/queenbee/recipe/__init__.py
+++ b/queenbee/recipe/__init__.py
@@ -1 +1 @@
-from .recipe import Recipe, BakedRecipe
+from .recipe import Recipe, BakedRecipe, RecipeInterface