Class: Database

Inherits:
Object
  • Object
show all
Defined in:
lib/csvr/sqlitedb.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(db, table, headers, rows) ⇒ Database

Returns a new instance of Database.



7
8
9
10
11
12
 
# File 'lib/csvr/sqlitedb.rb', line 7

def initialize(db, table, headers, rows)
  @db = db
  @table = table
  @headers = headers
  @rows = rows
end

Instance Attribute Details

#db ⇒ Object (readonly)

Returns the value of attribute db.



5
6
7
 
# File 'lib/csvr/sqlitedb.rb', line 5

def db
  @db
end

#headers ⇒ Object (readonly)

Returns the value of attribute headers.



5
6
7
 
# File 'lib/csvr/sqlitedb.rb', line 5

def headers
  @headers
end

#rows ⇒ Object (readonly)

Returns the value of attribute rows.



5
6
7
 
# File 'lib/csvr/sqlitedb.rb', line 5

def rows
  @rows
end

#table ⇒ Object (readonly)

Returns the value of attribute table.



5
6
7
 
# File 'lib/csvr/sqlitedb.rb', line 5

def table
  @table
end

Instance Method Details

#create ⇒ Object 



14
15
16
17
18
19
20
21
22
 
# File 'lib/csvr/sqlitedb.rb', line 14

def create

  puts "Inserting #{@rows.size} rows into database"

  db = SQLite3::Database.open("#{@db}.db")
  db.execute "CREATE TABLE IF NOT EXISTS #{@table}(#{@headers})"
  @rows.each { |row| db.execute "INSERT INTO #{@table} #{row}" }
  db.close
end