Merge pull request bryanyang0528#95 from HHArrowood/93-bugfix-pep-479-stopiteration

Handle StopIteration Error

Author: bryanyang0528

Diff for: ksql/utils.py

@@ -83,7 +83,7 @@ def parse_columns(columns_str):
 def process_row(row, column_names):
     row = row.replace(",\n", "").replace("]\n", "")
     row_obj = json.loads(row)
-    if 'finalMessage' in row_obj:
+    if "finalMessage" in row_obj:
         return None
     column_values = row_obj["row"]["columns"]
     index = 0
@@ -100,12 +100,14 @@ def process_query_result(results, return_objects=None):
         yield from results
 
     # parse rows into objects
-    header = next(results)
+    try:
+        header = next(results)
+    except StopIteration:
+        return
     columns = parse_columns(header)
 
     for result in results:
         row_obj = process_row(result, columns)
         if row_obj is None:
             return
         yield row_obj
-