Add include_in_schema

PythonistaGuild · May 5, 2024 · d191db7 · d191db7
1 parent 8ba004b
commit d191db7
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/starlette_plus/core.py b/starlette_plus/core.py
@@ -215,6 +215,7 @@ def __new__(cls, *args: Any, **kwargs: Any) -> Self:
                     endpoint=member,
                     methods=member._methods,
                     name=f"{name}.{member._coro.__name__}",
+                    include_in_schema=member._include_in_schema,
                 )
 
             new.limits = getattr(member, "_limits", [])  # type: ignore
@@ -244,7 +245,13 @@ def add_view(self, view: View | Self) -> None:
                 new = WebSocketRoute(path, endpoint=route_.endpoint, name=route_.name)
             else:
                 methods: list[str] | None = list(route_.methods) if route_.methods else None
-                new = Route(path, endpoint=route_.endpoint, methods=methods, name=route_.name)
+                new = Route(
+                    path,
+                    endpoint=route_.endpoint,
+                    methods=methods,
+                    name=route_.name,
+                    include_in_schema=route_.include_in_schema,
+                )
 
             new.limits = route_.limits  # type: ignore
             self.routes.append(new)
@@ -290,6 +297,7 @@ def __new__(cls, *args: Any, **kwargs: Any) -> Self:
                     endpoint=member,
                     methods=member._methods,
                     name=f"{name}.{member._coro.__name__}",
+                    include_in_schema=member._include_in_schema,
                 )
 
             new.limits = getattr(member, "_limits", [])  # type: ignore