Format and add type to function parameter

psrenergy · Nov 29, 2023 · 735aa5d · 735aa5d
1 parent 1359a00
commit 735aa5d
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/src/OpenSQL/OpenSQL.jl b/src/OpenSQL/OpenSQL.jl
@@ -8,6 +8,8 @@ using DataFrames
 
 const DB = SQLite.DB
 
+const ValidOpenSQLDataType = Union{AbstractString, Integer, Real, Bool}
+
 """
 SQLInterface
 """

diff --git a/src/OpenSQL/update.jl b/src/OpenSQL/update.jl
@@ -3,8 +3,8 @@ function update!(
     table::String,
     column::String,
     id::String,
-    val,
-)
+    val::T,
+) where {T <: ValidOpenSQLDataType}
     sanity_check(db, table, column)
     DBInterface.execute(db, "UPDATE $table SET $column = '$val' WHERE id = '$id'")
     return nothing
@@ -14,8 +14,8 @@ function update!(
     db::SQLite.DB,
     table::String,
     column::String,
-    val,
-)
+    val::T,
+) where {T <: ValidOpenSQLDataType}
     sanity_check(db, table, column)
     DBInterface.execute(db, "UPDATE $table SET $column = '$val'")
     return nothing

diff --git a/src/sql_interface.jl b/src/sql_interface.jl
@@ -109,7 +109,7 @@ function link_series_to_file(
 
     if length(get_parms(db, time_series_table, attribute)) == 0
         OpenSQL.create_parameters!(db, time_series_table, Dict(attribute => file_path))
-    else 
+    else
         OpenSQL.update!(db, time_series_table, attribute, file_path)
     end
     return nothing

diff --git a/test/OpenSQL/time_series.jl b/test/OpenSQL/time_series.jl
@@ -130,7 +130,6 @@ function test_time_series_2()
 
     PSRI.close(iow)
 
-
     iow = PSRI.open(
         PSRI.OpenBinary.Writer,
         joinpath(case_path, "cost");
-Original file line number
+Diff line change
@@ Expand Up / @@ -8,6 +8,8 @@ using DataFrames @@
     const DB = SQLite.DB
+    const ValidOpenSQLDataType = Union{AbstractString, Integer, Real, Bool}
     """
     SQLInterface
     """
@@ Expand Down @@