Merge pull request #9 from queryverse/eltypeunknown

Eltypeunknown
queryverse · Feb 10, 2019 · 1613f4a · 1613f4a
2 parents 19824af + e1f25ec
commit 1613f4a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -1,3 +1,6 @@
+# TableShowUtils.jl v0.2.4 Release Notes
+* Make printdataresource work with EltypeUnknown
+
 # TableShowUtils.jl v0.2.3 Release Notes
 * Show NA values in gray
 

diff --git a/src/TableShowUtils.jl b/src/TableShowUtils.jl
@@ -207,9 +207,14 @@ own_json_formatter(io, x) = JSON.print(io, x)
 own_json_formatter(io, x::DataValues.DataValue) = DataValues.isna(x) ? JSON.print(io,nothing) : own_json_formatter(io, x[])
 
 function printdataresource(io::IO, source)
-    col_names = String.(fieldnames(eltype(source)))
-    col_types = [fieldtype(eltype(source), i) for i=1:length(col_names)]
-
+    if Base.IteratorEltype(source) isa Base.EltypeUnknown
+        first_el = first(source)
+        col_names = String.(propertynames(first_el))
+        col_types = [fieldtype(typeof(first_el), i) for i=1:length(col_names)]
+    else
+        col_names = String.(fieldnames(eltype(source)))
+        col_types = [fieldtype(eltype(source), i) for i=1:length(col_names)]
+    end
     schema = Dict("fields" => [Dict("name"=>string(i[1]), "type"=>julia_type_to_schema_type(i[2])) for i in zip(col_names, col_types)])
 
     print(io, "{")