diff --git a/src/database.rb b/src/database.rb index 426deb3..a45079f 100644 --- a/src/database.rb +++ b/src/database.rb @@ -1,5 +1,7 @@ DEFAULT_DB_PATH = "db/main.db" +require_relative "db_models.rb" + def db_handle db = SQLite3::Database.new DEFAULT_DB_PATH db.results_as_hash = true diff --git a/src/db_models.rb b/src/db_models.rb new file mode 100644 index 0000000..20b5b22 --- /dev/null +++ b/src/db_models.rb @@ -0,0 +1,6 @@ +class TableModel + attr_reader :table_name + def initialize(table_name) + @table_name = table_name + end +end