Добавление Django Channels к DRF
Всем привет!
Пытаюсь добавить в свой проект channel, чтобы слать состояние загрузки файла клиенту (процент загрузки).
У меня есть метод download_file(), который скачивает файл на сервер из некоторых эндпоинтов.
Получилось добавить channels в проект, но проект - это API без фронта.
asgi.py
import os
import django
from django.core.asgi import get_asgi_application
from src.apps import api
from channels.routing import ProtocolTypeRouter, URLRouter
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'src.settings')
django.setup()
application = ProtocolTypeRouter({
'http': get_asgi_application(),
'websocket': URLRouter(api.routing.websocket_urlpatterns),
})
settings.py
THIRD_PARTY_APPS = (
'...' ,
'channels',
)
# ...
# channels
ASGI_APPLICATION = 'src.asgi.application'
CHANNEL_LAYERS = {
"default": {
"BACKEND": "channels_rabbitmq.core.RabbitmqChannelLayer",
},
}
routing.py
from django.urls import path
from src.apps.api.consumers import DownloadStateConsumer
websocket_urlpatterns = [
path('download_state/', DownloadStateConsumer.as_asgi()),
]
consumers.py
from channels.generic.websocket import AsyncWebsocketConsumer
class DownloadStateConsumer(AsyncJsonWebsocketConsumer):
async def connect(self):
await self.accept()
async def receive(self, text_data=None, bytes_data=None, **kwargs):
await self.send_json({'message': text_data})
file_download.py
def download_file():
# ...
response = requests.get(link.link, stream=True)
total_length = response.headers.get('content-length')
size = 0
with response as r:
r.raise_for_status()
with open(filename, 'wb') as f:
state = -1
for chunk in r.iter_content(chunk_size=8192):
size += len(chunk)
f.write(chunk)
done = int(100 * size / int(total_length))
sys.stdout.write("\r%s%s" % (str(done), "%"))
if done != state:
async_to_sync(
get_channel_layer().group_send(
'download_state', {
'type': 'send.percent',
'message': done,
}
)
)
state = done
sys.stdout.flush()
# ...
я так понимаю, что на клиенте должен быть скрипт, типа
const webSocket = new WebSocket(
'ws://'
+ window.location.host
+ '/download_state/'
);
webSocket.onmessage = function(e) {
const data = JSON.parse(e.data);
document.querySelector('#download-state').value = data.message;
};
webSocket.onclose = function(e) {
console.error('Chat socket closed unexpectedly');
};
Вопрос: как правильно из метода download_file() отправлять на клиент состояние загрузки? То, как сейчас, не работает - клиент не принимает сообщения. Заранее спасибо!
Ответы (1 шт):
В итоге я поменяла channels_rabbitmq на channels_redis:
CHANNEL_LAYERS = {
'default': {
'BACKEND': 'channels_redis.core.RedisChannelLayer',
'CONFIG': {
"hosts": [('127.0.0.1', 6379)],
},
},
}
Запустила образ Redis через Docker на указанном порту, и теперь всё работает.