Add optional column mapping (#51)

paths.yaml

@@ -75,6 +75,8 @@
                   filename:
                     type: string
                     example: "dataset.csv"
+                  mappings:
+                    type: object
               timespan:
                 type: object
                 properties:

src/service/ArgIO.jl

@@ -31,7 +31,7 @@ function prepare_input(args; context...)
         args[:model] = fetch_model(args[:model_config_id])
     end
     if in(:dataset, keys(args)) && !isa(args[:dataset], String)
-        args[:dataset] = fetch_dataset(args[:dataset]["id"], args[:dataset]["filename"])
+        args[:dataset] = fetch_dataset(args[:dataset]["id"], args[:dataset]["filename"], get(args[:dataset], "mappings", Dict()))
     end
     if in(:model_config_ids, keys(args))
         args[:models] = fetch_model.(map(string, args[:model_ids]))

src/service/AssetManager.jl

@@ -3,7 +3,7 @@ Asset fetching from TDS
 """
 module AssetManager
 
-import DataFrames: DataFrame
+import DataFrames: rename!, DataFrame
 import CSV, Downloads, HTTP
 import OpenAPI.Clients: Client
 import JSON3 as JSON
@@ -31,15 +31,17 @@ end
 """
 Return csv from TDS by ID
 """
-function fetch_dataset(dataset_id::String, filename::String)
+function fetch_dataset(dataset_id::String, filename::String, mappings::Dict=Dict())
     # TODO(five): Select name dynamicially
     url = "$(settings["TDS_URL"])/datasets/$dataset_id/download-url?filename=$filename"
     response = HTTP.get(url, ["Content-Type" => "application/json"])
     body = response.body |> JSON.read ∘ String
     io = IOBuffer()
     Downloads.download(body.url, io)
     seekstart(io)
-    CSV.read(io, DataFrame)
+    dataframe = CSV.read(io, DataFrame)
+    for (from, to) in mappings rename!(dataframe, Symbol(from)=>Symbol(to)) end
+    dataframe
 end
 
 """