diff --git a/falcon_apispec/falcon_plugin.py b/falcon_apispec/falcon_plugin.py index 9583a93..427fb54 100644 --- a/falcon_apispec/falcon_plugin.py +++ b/falcon_apispec/falcon_plugin.py @@ -2,6 +2,7 @@ import re from apispec import BasePlugin, yaml_utils from apispec.exceptions import APISpecError +from apispec.core import VALID_METHODS class FalconPlugin(BasePlugin): @@ -11,8 +12,14 @@ def __init__(self, app): super(FalconPlugin, self).__init__() self._app = app - @staticmethod - def _generate_resource_uri_mapping(app): + def init_spec(self, spec): + self._spec = spec + + def _get_valid_methods(self): + return set(VALID_METHODS[self._spec.openapi_version.major]) + + def _generate_resource_uri_mapping(self, app): + valid_methods = self._get_valid_methods() routes_to_check = copy.copy(app._router._roots) mapping = {} @@ -26,7 +33,11 @@ def _generate_resource_uri_mapping(app): if route.method_map: for method_name, method_handler in route.method_map.items(): - if method_handler.__dict__.get("__module__") == "falcon.responders": + if method_handler.__module__ == "falcon.responders": + continue + if method_name.lower() not in valid_methods: + continue + if method_name.lower() not in valid_methods: continue mapping[resource]["methods"][method_name.lower()] = method_handler diff --git a/falcon_apispec/version.py b/falcon_apispec/version.py index 6a9beea..3d26edf 100644 --- a/falcon_apispec/version.py +++ b/falcon_apispec/version.py @@ -1 +1 @@ -__version__ = "0.4.0" +__version__ = "0.4.1"