Don't encode byte responses

2019-10-06 03:10:43 +02:00 · 2019-10-06 03:10:43 +02:00 · 73e0aa17c4
commit 73e0aa17c4
parent 917a2e59ce
1 changed files with 8 additions and 2 deletions
--- a/src/util/helper.py
+++ b/src/util/helper.py
@ -340,8 +340,14 @@ def encodeResponse(func):
        back = func(*args, **kwargs)
        if "__next__" in dir(back):
            for part in back:
-                yield part.encode()
+                if type(part) == bytes:
+                    yield part
+                else:
+                    yield part.encode()
        else:
-            yield back.encode()
+            if type(back) == bytes:
+                yield back
+            else:
+                yield back.encode()

    return wrapper