Make init_spec more convenient

Fatal1ty · Jan 30, 2024 · debd7c6 · debd7c6
1 parent f24f3d4
commit debd7c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/openapify/core/base_plugins.py b/openapify/core/base_plugins.py
@@ -37,11 +37,6 @@ def media_type_helper(
 
 
 class BaseSchemaPlugin(BasePlugin):
- spec: APISpec
-
- def init_spec(self, spec: APISpec) -> None:
- self.spec = spec
-
  def schema_helper(
  self,
  obj: Union[Body, Cookie, Header, QueryParam],

diff --git a/openapify/plugin.py b/openapify/plugin.py
@@ -6,8 +6,10 @@
 
 
 class BasePlugin:
+ spec: APISpec
+
  def init_spec(self, spec: APISpec) -> None:
- pass
+ self.spec = spec
 
  def schema_helper(
  self,