added csv loading functions

lib/csv_db.rb

@@ -1,3 +1,4 @@
+#require 'FasterCSV'
 module CsvDb
   module Helper
     def self.loader
@@ -14,13 +15,34 @@ def self.extension
   end
 
   class Load < SerializationHelper::Load
+    def self.load_documents(io, truncate = true)
+      puts io.to_s
+      tables = {}
+      curr_table = nil
+      io.each do |line|
+        if /BEGIN_CSV_TABLE_DECLARATION(.+)END_CSV_TABLE_DECLARATION/ =~ line
+          curr_table = $1
+          tables[curr_table] = {}
+        else
+          if tables[curr_table]["columns"]
+            tables[curr_table]["records"] << FasterCSV.parse(line)[0]
+          else
+            tables[curr_table]["columns"] = FasterCSV.parse(line)[0]
+            tables[curr_table]["records"] = []
+          end
+        end
+      end
 
+      tables.each_pair do |table_name, contents|
+        load_table(table_name, contents, truncate)
+      end
+    end
   end
 
   class Dump < SerializationHelper::Dump
 
     def self.before_table(io,table)
-      io.write "#{table}:"
+      io.write "BEGIN_CSV_TABLE_DECLARATION#{table}END_CSV_TABLE_DECLARATION\n"
     end
 
     def self.dump(io)

lib/yaml_db.rb

@@ -57,6 +57,8 @@ class Load < SerializationHelper::Load
     def self.load_documents(io, truncate = true) 
         YAML.load_documents(io) do |ydoc|
           ydoc.keys.each do |table_name|
+            puts table_name
+            puts ydoc[table_name].inspect
             next if ydoc[table_name].nil?
             load_table(table_name, ydoc[table_name], truncate)
           end

tasks/yaml_db_tasks.rake

@@ -31,7 +31,8 @@ namespace :db do
 		desc "Load contents of db/data.extension (defaults to yaml) into database"
 		task :load => :environment do
             format_class = ENV['class'] || "YamlDb::Helper"
-			SerializationHelper::Base.new(format_class.constantize).load db_dump_data_file
+            helper = format_class.constantize
+			SerializationHelper::Base.new(helper).load (db_dump_data_file helper.extension)
 		end
 
 		desc "Load contents of db/data_dir into database"