fix-rate-limit-serializer
Georg Krause 2022-07-16 09:21:33 +02:00
rodzic 1fa648ecea
commit 0ee20f400c
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 2970D504B2183D22
2 zmienionych plików z 5 dodań i 3 usunięć

Wyświetl plik

@ -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

Wyświetl plik

@ -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):