]> git.0d.be Git - django-panik-nonstop.git/commitdiff
stamina: handle invalid incoming messages
authorFrédéric Péters <fpeters@0d.be>
Thu, 9 Jul 2020 20:36:02 +0000 (22:36 +0200)
committerFrédéric Péters <fpeters@0d.be>
Thu, 9 Jul 2020 20:36:02 +0000 (22:36 +0200)
nonstop/management/commands/stamina.py

index ddb1ac99b6149d45e5b4e8f7807ff2da02527393..3bcd19dfecb443222abe246334d0774e7d4671a6 100644 (file)
@@ -324,7 +324,11 @@ class Command(BaseCommand):
         end = False
         while not end:
             data = await reader.read(100)
-            message = data.decode().strip()
+            try:
+                message = data.decode().strip()
+            except UnicodeDecodeError:
+                print('got invalid message %r' % message)
+                continue
             print('got message: %r' % message)
             if message == 'status':
                 response = {'slot': str(self.slot)}