Replace CSV structure functions with a simpler method

Project.toml

@@ -1,10 +1,11 @@
 name = "DOMO"
 uuid = "72b704fe-efc7-41fd-b723-a3ca9fa252d1"
 authors = ["Michael Johnson"]
-version = "0.1.3"
+version = "0.1.4"
 
 [deps]
 Base64 = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f"
+CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 HTTP = "cd3eb016-35fb-5094-929b-558a96fad6f3"

src/schema.jl

@@ -1,3 +1,5 @@
+using DataFrames: DataFrame
+using CSV
 """
     match_domo_types(type)
 
@@ -44,37 +46,17 @@ function create_dataset_schema(df, name, description)
     return JSON3.write(schema)
 end
 
-function create_csv_string(row, col_num)
-    csv_string = string(ifelse(ismissing(row[col_num]), "", row[col_num]))
-
-    if occursin(",", csv_string)
-        csv_string = "\"" * csv_string * "\""
-    end
-
-    return csv_string
-end
-
 """
-    create_csv_structure(df)
+    dataframe_to_csv
 
-This function creates a CSV, in the form of a single string, from a DataFrame to send to Domo.
+Convert a Julia DataFrame to a CSV string for the API.
 """
-function create_csv_structure(df)
-    csv_data = ""
-
-    for row in eachrow(df), col_num in 1:ncol(df)
-        if col_num < ncol(df) && rownumber(row) < nrow(df)
-            csv_data = csv_data * create_csv_string(row, col_num) * ","
-        elseif col_num == ncol(df) && rownumber(row) < nrow(df)
-            csv_data = csv_data * create_csv_string(row, col_num) * "\n"
-        elseif col_num < ncol(df) && rownumber(row) == nrow(df)
-            csv_data = csv_data * create_csv_string(row, col_num) * ","
-        elseif col_num == ncol(df) && rownumber(row) == nrow(df)
-            csv_data = csv_data * create_csv_string(row, col_num) * "\n"
-        end
-    end
-
-    return csv_data
+function dataframe_to_csv(df::DataFrame)
+    io = IOBuffer()
+    io_file = CSV.write(io, df; header = false)
+    csv_string = String(take!(io_file))
+    close(io)
+    return csv_string
 end
 
 function push_schema_to_domo(dataset_schema)

test/runtests.jl

@@ -21,6 +21,6 @@ include("test-sets/schema-tests.jl")
         @test match_domo_types(types[type]) == expected_types[type]
     end
     # test whether behavior of csv creator is valid
-    @test create_csv_structure(schema_test_mathematicians_dataset) == test_csv_string_math
-    @test create_csv_structure(null_schema_test_df) == test_csv_string_crows
+    @test dataframe_to_csv(schema_test_mathematicians_dataset) == test_csv_string_math
+    @test dataframe_to_csv(null_schema_test_df) == test_csv_string_crows
 end;

test/test-sets/schema-tests.jl

@@ -1,4 +1,4 @@
-import DOMO: match_domo_types, create_dataset_schema, create_csv_structure
+import DOMO: match_domo_types, create_dataset_schema, dataframe_to_csv
 using JSON3
 import DataFrames: DataFrame
 using Dates