Skip to content

Commit cc598e7

Browse files
committed
refactor
1 parent 5d36dd8 commit cc598e7

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/stac_auth_proxy/utils/middleware.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -101,11 +101,14 @@ async def transform_response(message: Message) -> None:
101101
return await self.app(scope, receive, transform_response)
102102

103103

104-
def required_conformance(*conformances: str):
104+
def required_conformance(
105+
*conformances: str,
106+
attr_name: str = "__required_conformances__",
107+
):
105108
"""Register required conformance classes with a middleware class."""
106109

107-
def decorator(func):
108-
func.__required_conformances__ = list(conformances)
109-
return func
110+
def decorator(middleware):
111+
setattr(middleware, attr_name, list(conformances))
112+
return middleware
110113

111114
return decorator

0 commit comments

Comments
 (0)