From 11616428141cc33c87793dafd8899658f034edc3 Mon Sep 17 00:00:00 2001
From: Carlos Camacho <ccamacho@redhat.com>
Date: Wed, 26 Jun 2019 14:07:44 +0200
Subject: [PATCH] Check if its possible to parse a chunked response

This patch is a simple check for either read the chunks
of the response of use the standard read method.

Currently if it is not possible to use the 'read_chunked'
method the client fails.
---
 watch/watch.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/watch/watch.py b/watch/watch.py
index 5966eace..04f3c4f5 100644
--- a/watch/watch.py
+++ b/watch/watch.py
@@ -45,10 +45,27 @@ def _find_return_type(func):
 
 def iter_resp_lines(resp):
     prev = ""
-    for seg in resp.read_chunked(decode_content=False):
+    # Check if it is possible to read a chunked
+    # response, if not, use the standard read.
+    try:
+        resp = resp.read_chunked(decode_content=False)
+        for seg in resp:
+            if isinstance(seg, bytes):
+                seg = seg.decode('utf8')
+            seg = prev + seg
+            lines = seg.split("\n")
+            if not seg.endswith("\n"):
+                prev = lines[-1]
+                lines = lines[:-1]
+            else:
+                prev = ""
+            for line in lines:
+                if line:
+                    yield line
+    except AttributeError:
+        seg = resp.read()
         if isinstance(seg, bytes):
             seg = seg.decode('utf8')
-        seg = prev + seg
         lines = seg.split("\n")
         if not seg.endswith("\n"):
             prev = lines[-1]