module Db

# File lib/db.rb, line 14
def close
      @conn.disconnect
      @conn = nil
end

# File lib/db.rb, line 5
def connect
  begin
    @conn = Sequel.connect('sqlite://docgen')
    @substitutions = @conn.from(:substitutions)
  rescue Exeption => e
    puts "Database connection error: #{e.message}"
  end
end

# File lib/db.rb, line 19
def substitution_text_for set_id, key
  connect unless @conn
  res = @substitutions.where(:set_id => set_id, :key => key).select(:value)
  res.get(:value)
end