download hepers

Author: waralex

Project.toml

@@ -4,6 +4,7 @@ authors = ["Chris Parmer <[email protected]>", "Alexandr Romanenko <waralex@gmail
 version = "1.0.0"
 
 [deps]
+Base64 = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f"
 CodecZlib = "944b1d66-785c-5afd-91f1-9de20f533193"
 DashBase = "03207cf0-e2b3-4b91-9ca8-690cf0fb507e"
 DashCoreComponents = "1b08a953-4be3-4667-9a23-9da06441d987"

src/Dash.jl

@@ -25,6 +25,7 @@ include("resources/application.jl")
 include("handlers.jl")
 include("server.jl")
 include("init.jl")
+include("components_utils/_components_utils.jl")
 
 @doc """
     module Dash
@@ -109,4 +110,4 @@ end
 
 
 
-end # module
+end # module

src/components_utils/_components_utils.jl

@@ -0,0 +1 @@
+include("express.jl")

src/components_utils/express.jl

@@ -0,0 +1,105 @@
+import Base64
+export dcc_send_file, dcc_send_string, dcc_send_bytes
+"""
+    dbc_send_file(path::AbstractString, filename = nothing; type = nothing)
+
+Convert a file into the format expected by the Download component.
+
+# Arguments
+- `path` - path to the file to be sent
+- `filename` - name of the file, if not provided the original filename is used
+- `type` - type of the file (optional, passed to Blob in the javascript layer)
+"""
+function dcc_send_file(path, filename = nothing; type = nothing)
+    filename = isnothing(filename) ? basename(path) : filename
+    return dcc_send_bytes(read(path), filename, type = type)
+end
+
+"""
+    dcc_send_bytes(src::AbstractVector{UInt8}, filename; type = nothing)
+    dcc_send_bytes(src::Function, filename; type = nothing)
+
+Convert vector of bytes into the format expected by the Download component.
+
+# Examples
+
+Sending binary content
+```julia
+file_data = read("path/to/file")
+callback!(app, Output("download", "data"), Input("download-btn", "n_clicks"), prevent_initial_call = true) do n_clicks
+    return dcc_send_bytes(file_data, "filename.fl")
+end
+```
+
+Sending `DataFrame` in `Arrow` format
+```julia
+using DataFrames, Arrow
+...
+df = DataFrame(...)
+callback!(app, Output("download", "data"), Input("download-btn", "n_clicks"), prevent_initial_call = true) do n_clicks
+    return dcc_send_bytes("df.arr") do io
+        Arrow.write(io, df)
+    end
+end
+```
+"""
+function dcc_send_bytes(src::AbstractVector{UInt8}, filename; type = nothing)
+
+    return Dict(
+        :content => Base64.base64encode(src),
+        :filename => filename,
+        :type => type,
+        :base64 => true
+    )
+end
+
+function dcc_send_bytes(src::Function, filename; type = nothing)
+    io = IOBuffer()
+    src(io)
+    return dcc_send_bytes(take!(io), filename, type = type)
+end
+
+"""
+    dcc_send_data(src::AbstractString, filename; type = nothing)
+    dcc_send_data(src::Function, filename; type = nothing)
+
+Convert string into the format expected by the Download component.
+`src` is a string or function that takes a single argument - io and writes data to it
+
+# Examples
+
+Sending string content
+```julia
+text_data = "this is the test"
+callback!(app, Output("download", "data"), Input("download-btn", "n_clicks"), prevent_initial_call = true) do n_clicks
+    return dcc_send_string(text_data, "text.txt")
+end
+```
+
+Sending `DataFrame` in `CSV` format
+```julia
+using DataFrames, CSV
+...
+df = DataFrame(...)
+callback!(app, Output("download", "data"), Input("download-btn", "n_clicks"), prevent_initial_call = true) do n_clicks
+    return dcc_send_string("df.csv") do io
+        CSV.write(io, df)
+    end
+end
+```
+"""
+function dcc_send_string(src::AbstractString, filename; type = nothing)
+
+    return Dict(
+        :content => src,
+        :filename => filename,
+        :type => type,
+        :base64 => false
+    )
+end
+
+function dcc_send_string(src::Function, filename; type = nothing)
+    io = IOBuffer()
+    src(io)
+    return dcc_send_string(String(take!(io)), filename, type = type)
+end

test/components_utils.jl

@@ -0,0 +1,53 @@
+using Test
+using Dash
+using Base64
+@testset "Send String" begin
+    data = "test string"
+    res = dcc_send_string(data, "test_file.csv"; type = "text/csv")
+    @test res[:content] == data
+    @test res[:filename] == "test_file.csv"
+    @test res[:type] == "text/csv"
+    @test !res[:base64]
+
+    data2 = "test 2 string"
+    res = dcc_send_string("test_file.csv"; type = "text/csv") do io
+        write(io, data2)
+    end
+    @test res[:content] == data2
+    @test res[:filename] == "test_file.csv"
+    @test res[:type] == "text/csv"
+    @test !res[:base64]
+end
+@testset "Send Bytes" begin
+    data = "test string"
+    res = dcc_send_bytes(Vector{UInt8}(data), "test_file.csv"; type = "text/csv")
+    @test res[:content] == Base64.base64encode(data)
+    @test res[:filename] == "test_file.csv"
+    @test res[:type] == "text/csv"
+    @test res[:base64]
+
+    data2 = "test 2 string"
+    res = dcc_send_bytes("test_file.csv"; type = "text/csv") do io
+        write(io, data2)
+    end
+    @test res[:content] == Base64.base64encode(data2)
+    @test res[:filename] == "test_file.csv"
+    @test res[:type] == "text/csv"
+    @test res[:base64]
+end
+@testset "Send File" begin
+
+    file = "assets/test.png"
+    res = dcc_send_file(file, nothing; type = "text/csv")
+    @test res[:content] == Base64.base64encode(read(file))
+    @test res[:filename] == "test.png"
+    @test res[:type] == "text/csv"
+    @test res[:base64]
+
+    file = "assets/test.png"
+    res = dcc_send_file(file, "ttt.jpeg"; type = "text/csv")
+    @test res[:content] == Base64.base64encode(read(file))
+    @test res[:filename] == "ttt.jpeg"
+    @test res[:type] == "text/csv"
+    @test res[:base64]
+end

test/runtests.jl

@@ -1,4 +1,4 @@
-include("env.jl")
+#=include("env.jl")
 include("resources.jl")
 include("devtools.jl")
 include("dash_creation.jl")
@@ -8,5 +8,6 @@ include("utils.jl")
 include("context.jl")
 include("core.jl")
 include("misc.jl")
-include("callbacks.jl")
-#include("dev.jl")
+include("callbacks.jl")=#
+include("components_utils.jl")
+#include("dev.jl")