Extend schema decorator for Channels
Project description
Extend schema decorator for Channels
Provides extend_ws_schema decorator to documentation Cunsumer methods from channels just like it does drf-spectacular
Requirements
- django>=4.2.11
- djangorestframework>=3.14.0
- drf-spectacular>=0.27.2
- drf-spectacular-sidecar>=2024.4.1
- channels>=4.1.0
Install
pip install drf-spectacular-websocket
python manage.py collectstatic
for sidecar static- Add app name to
INSTALLED_APPS
INSTALLED_APPS = [
# drf_spectacular_websocket must be higher than drf_spectacular
'drf_spectacular_websocket',
'drf_spectacular',
'drf_spectacular_sidecar',
]
Configure settings
ASGI_APPLICATION = 'path.to.your.application'
# default settings (there is no need to define in your application)
SPECTACULAR_SETTINGS = {
'DEFAULT_GENERATOR_CLASS': 'drf_spectacular_websocket.schemas.WsSchemaGenerator',
'SWAGGER_UI_DIST': 'SIDECAR',
'SWAGGER_UI_FAVICON_HREF': 'SIDECAR',
'REDOC_DIST': 'SIDECAR',
'SWAGGER_UI_SETTINGS': {
'connectSocket': True, # Automatically establish a WS connection when opening swagger
'socketMaxMessages': 8, # Max number of messages displayed in the log window in swagger
'socketMessagesInitialOpened': False, # Automatically open the log window when opening swagger
},
}
# you can override this default settings in your application:
# connectSocket
# socketMaxMessages
# socketMessagesInitialOpened
About decorator
drf_spectacular_websocket extend_ws_schema
accepts one new type
parameter relative to drf_spectacular extend_schema
.
type
send
- Type of interaction, [request -> response]receive
- Type of interaction, [response without request]
Usage example
drf_spectacular_websocket automatically finds websocket urls and related consumer using
ASGI_APPLICATION
setting.
Direct usage
from channels.generic.websocket import AsyncJsonWebsocketConsumer
from rest_framework.serializers import Serializer, CharField
from drf_spectacular_websocket.decorators import extend_ws_schema
class SomeMethodInputSerializer(Serializer):
some_field = CharField()
class SomeMethodOutputSerializer(Serializer):
some_field = CharField()
class SendMessageOutputSerializer(Serializer):
some_field = CharField()
class SomeConsumer(AsyncJsonWebsocketConsumer):
async def receive_json(self, content, **kwargs):
some_value = content.get('some_key')
if some_value:
await self.some_method(some_value)
@extend_ws_schema(
type='send',
summary='some_method_summary',
description='some_method_description',
request=SomeMethodInputSerializer,
responses=SomeMethodOutputSerializer,
)
async def some_method(self, some_value):
input_serializer = SomeMethodInputSerializer(data=some_value)
input_serializer.is_valid(raise_exception=True)
return_value = await some_business_logic(input_serializer.validated_data)
output_serializer = SomeMethodOutputSerializer(data=return_value)
output_serializer.is_valid(raise_exception=True)
await self.send_json(output_serializer.validated_data)
@extend_ws_schema(
type='receive',
summary='send_message_summary',
description='send_message_description',
request=None,
responses=SendMessageOutputSerializer,
)
async def send_message(self, message):
await self.send_json(message)
Send
Receive
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for drf_spectacular_websocket-1.2.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | bc913b7b6a08fcbdb2b836e85f71c7f8da51906bf437e17370a85236d0af70aa |
|
MD5 | 352df116dfeac0a5a81b3cb23b90d437 |
|
BLAKE2b-256 | 3883a08c2df3f24a85b71693d4d73d484b60bab639c326d0c19244ed1e7bd289 |
Close
Hashes for drf_spectacular_websocket-1.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b48a75894ef704ce722a58f83f917756bdcda2871fb0ab492dd57fd571bae494 |
|
MD5 | c07d7a06153af5c93b3609418a8d0119 |
|
BLAKE2b-256 | 4cd449a03a10ad0c6299bb893edca5af69818716d65a6766f9ccd25f662f41ea |