getproperty overloading

Author: bramtayl

src/library.jl

@@ -28,7 +28,7 @@ function get_column(source_row, column_name)
     SourceCode(source_row.source, Expr(:call, getproperty, source_row, column_name))
 end
 function model_row_dispatch(::typeof(getproperty), source_tables::SourceTables, table_name)
-    source = source_tables.source
+    source = get_source(source_tables)
     column_names = get_column_names(source, table_name)
     NamedTuple{column_names}(partial_map(
         get_column,

src/source.jl

@@ -23,28 +23,24 @@ function get_column_names(source::DB, table_name)
 end
 export get_column_names
 
+
+"""
+    struct SourceTables{Source}
+
+`source` must support [`get_table_names`](@ref) and [`get_column_names`](@ref).
+"""
 struct SourceTables{Source}
     source::Source
 end
+export SourceTables
 
-function get_table(source, table_name::Symbol)
-    SourceCode(source,
-        Expr(:call, getproperty, SourceTables(source), table_name)
-    )
-end
-
-"""
-    get_tables(source)
+get_source(source_tables::SourceTables) = getfield(source_tables, :source)
 
-`source` must support [`get_table_names`](@ref).
-"""
-function get_tables(source)
-    table_names = get_table_names(source)
-    NamedTuple{table_names}(
-        partial_map(get_table, source, table_names)
+function getproperty(source_tables::SourceTables, table_name::Symbol)
+    SourceCode(get_source(source_tables),
+        Expr(:call, getproperty, source_tables, table_name)
     )
 end
-export get_tables
 
 struct SourceRow{Source}
     source::Source

test/runtests.jl

@@ -1,11 +1,11 @@
 using Query
 using Test
-using QuerySQLite
+using QuerySQLite: SourceTables
 using SQLite: DB
 using QueryTables
 
 filename = joinpath(@__DIR__, "Chinook_Sqlite.sqlite")
-database = get_tables(DB(filename))
+database = SourceTables(DB(filename))
 
 @testset "QuerySQLite" begin