From 0ee20f400c3aefeea2e2b2ebc6335c04d148d6f0 Mon Sep 17 00:00:00 2001 From: Georg Krause Date: Sat, 16 Jul 2022 09:21:33 +0200 Subject: [PATCH] Fix formatting --- api/config/schema.py | 6 ++++-- api/config/settings/local.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/api/config/schema.py b/api/config/schema.py index 2e7a28a9e..a78a96a89 100644 --- a/api/config/schema.py +++ b/api/config/schema.py @@ -1,6 +1,7 @@ from drf_spectacular.contrib.django_oauth_toolkit import OpenApiAuthenticationExtension import os + class CustomOAuthExt(OpenApiAuthenticationExtension): target_class = "funkwhale_api.common.authentication.OAuth2Authentication" name = "oauth2" @@ -26,11 +27,12 @@ class CustomOAuthExt(OpenApiAuthenticationExtension): return {"type": "oauth2", "flows": flows} + def custom_preprocessing_hook(endpoints): filtered = [] # your modifications to the list of operations that are exposed in the schema - api_type = os.environ['API_TYPE'] + api_type = os.environ["API_TYPE"] for (path, path_regex, method, callback) in endpoints: if path.startswith(f"/api/{api_type}"): - filtered.append((path, path_regex, method, callback)) + filtered.append((path, path_regex, method, callback)) return filtered diff --git a/api/config/settings/local.py b/api/config/settings/local.py index b720ce769..de4002782 100644 --- a/api/config/settings/local.py +++ b/api/config/settings/local.py @@ -131,7 +131,7 @@ SPECTACULAR_SETTINGS = { "OAUTH2_FLOWS": ["authorizationCode"], "OAUTH2_AUTHORIZATION_URL": "/authorize", "OAUTH2_TOKEN_URL": "/api/v1/oauth/token/", - 'PREPROCESSING_HOOKS': ["config.schema.custom_preprocessing_hook"], + "PREPROCESSING_HOOKS": ["config.schema.custom_preprocessing_hook"], } if env.bool("WEAK_PASSWORDS", default=False):