feat(Common/xlsx): reading a sheet to a table

Author: tecosaur

Common/ext/XLSXExt.jl

@@ -3,12 +3,25 @@ module XLSXExt
 using XLSX
 import DataToolkitCommon: _read_xlsx, _write_xlsx
 
-function _read_xlsx(from::IO, sheet::Union{String, Int}, range::Union{String, Nothing})
+function _read_xlsx(from::IO, ::Type{Matrix{Any}}, sheet::Union{String, Int}, range::Union{String, Nothing})
     if !isnothing(range)
         XLSX.readdata(from, sheet, range)
     else
         XLSX.readdata(from, sheet)
     end
 end
 
+function _read_xlsx(from::IO, astype::Type, sheet::Union{String, Int}, range::Union{String, Nothing})
+    table = if !isnothing(range)
+        XLSX.readtable(from, sheet, range)
+    else
+        XLSX.readtable(from, sheet)
+    end
+    if astype == XLSX.DataTable
+        table
+    else
+        astype(table)
+    end
+end
+
 end

Common/src/transformers/saveload/xlsx.jl

@@ -5,9 +5,21 @@ function load(loader::DataLoader{:xlsx}, from::IO, ::Type{Matrix})
     @require XLSX
     sheet = @getparam loader."sheet"::Union{String, Int} 1
     range = @getparam loader."range"::Union{String, Nothing}
-    invokelatest(_read_xlsx, from, sheet, range)
+    invokelatest(_read_xlsx, from, Matrix{Any}, sheet, range)
 end
 
+function load(loader::DataLoader{:xlsx}, from::IO, as::Type)
+    @require XLSX
+    sheet = @getparam loader."sheet"::Union{String, Int} 1
+    range = @getparam loader."range"::Union{String, Nothing}
+    invokelatest(_read_xlsx, from, as, sheet, range)
+end
+
+supportedtypes(::Type{DataLoader{:xlsx}}) =
+    [QualifiedType(:DataFrames, :DataFrame),
+     QualifiedType(:XLSX, :DataTable),
+     QualifiedType(:Base, :Matrix)]
+
 createpriority(::Type{DataLoader{:xlsx}}) = 10
 
 function createinteractive(::Type{DataLoader{:xlsx}}, source::String)