From ae78b1d1662435d6e36848e90af8576525f64a0d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mat=C3=BA=C5=A1=20Kobolka?= Date: Wed, 22 Jun 2022 12:28:20 +0200 Subject: [PATCH] makes versioning work also for websocket routes --- fastapi_versioning/versioning.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/fastapi_versioning/versioning.py b/fastapi_versioning/versioning.py index 0f282d2..9f7ed95 100644 --- a/fastapi_versioning/versioning.py +++ b/fastapi_versioning/versioning.py @@ -59,8 +59,12 @@ def VersionedFastAPI( version=semver, ) for route in version_route_mapping[version]: - for method in route.methods: - unique_routes[route.path + "|" + method] = route + try: + for method in route.methods: + unique_routes[route.path + "|" + method] = route + except AttributeError: + unique_routes[route.path] = route + for route in unique_routes.values(): versioned_app.router.routes.append(route) parent_app.mount(prefix, versioned_app)