Return only leaf field names in the record fields and don't recurse into REPEATED fields

mxmzdlv · mxmzdlv · commit 94d82104b4ea · 2018-08-18T20:46:45.000+12:00
diff --git a/pybigquery/sqlalchemy_bigquery.py b/pybigquery/sqlalchemy_bigquery.py
@@ -240,16 +240,16 @@ def _get_columns_helper(self, columns, cur_columns):
         """
         results = []
         for col in columns:
-            if col.field_type == 'RECORD':
+            if col.field_type == 'RECORD' and col.mode != 'REPEATED':
                 cur_columns.append(col)
                 results += self._get_columns_helper(col.fields, cur_columns)
                 cur_columns.pop()
-
-            results += [SchemaField(name='.'.join(col.name for col in cur_columns + [col]),
-                                    field_type=col.field_type,
-                                    mode=col.mode,
-                                    description=col.description,
-                                    fields=col.fields)]
+            else:
+                results += [SchemaField(name='.'.join(col.name for col in cur_columns + [col]),
+                                        field_type=col.field_type,
+                                        mode=col.mode,
+                                        description=col.description,
+                                        fields=col.fields)]
         return results
 
     def get_columns(self, connection, table_name, schema=None, **kw):
