pycodestyle

This commit is contained in:
jonaswinkler 2021-01-26 15:43:07 +01:00
parent 313f2886d7
commit 3b5d55c428
2 changed files with 7 additions and 5 deletions

View File

@ -8,10 +8,10 @@ from django.core.asgi import get_asgi_application
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "paperless.settings") os.environ.setdefault("DJANGO_SETTINGS_MODULE", "paperless.settings")
django_asgi_app = get_asgi_application() django_asgi_app = get_asgi_application()
from channels.auth import AuthMiddlewareStack from channels.auth import AuthMiddlewareStack # NOQA: E402
from channels.routing import ProtocolTypeRouter, URLRouter from channels.routing import ProtocolTypeRouter, URLRouter # NOQA: E402
from paperless.urls import websocket_urlpatterns from paperless.urls import websocket_urlpatterns # NOQA: E402
application = ProtocolTypeRouter({ application = ProtocolTypeRouter({
"http": get_asgi_application(), "http": get_asgi_application(),

View File

@ -7,10 +7,12 @@ from channels.generic.websocket import WebsocketConsumer
class StatusConsumer(WebsocketConsumer): class StatusConsumer(WebsocketConsumer):
def connect(self): def connect(self):
self.accept() self.accept()
async_to_sync(self.channel_layer.group_add)('status_updates', self.channel_name) async_to_sync(self.channel_layer.group_add)(
'status_updates', self.channel_name)
def disconnect(self, close_code): def disconnect(self, close_code):
async_to_sync(self.channel_layer.group_discard)('status_updates', self.channel_name) async_to_sync(self.channel_layer.group_discard)(
'status_updates', self.channel_name)
def status_update(self, event): def status_update(self, event):
self.send(json.dumps(event['data'])) self.send(json.dumps(event['data']))