GFunk911-dataload 0.3.0

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Mike Harris
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,7 @@
+= dataload
+
+Description goes here.
+
+== Copyright
+
+Copyright (c) 2009 Mike Harris. See LICENSE for details.

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:patch: 0
+:major: 0
+:minor: 3

data/lib/dataload/loader.rb

@@ -0,0 +1,133 @@
+require 'rubygems'
+require 'mharris_ext'
+require 'fastercsv'
+require 'activerecord'
+
+class FasterCSV::Row
+  def method_missing(sym,*args,&b)
+    if self[sym.to_s]
+      self[sym.to_s]
+    else
+      super(sym,*args,&b)
+    end
+  end
+end
+
+class Loader
+  fattr(:columns) { [] }
+  attr_accessor :source_filename, :db_ops, :table_name
+  fattr(:source_rows) do
+    res = []
+    FasterCSV.foreach(source_filename, :headers => true) do |row|
+      res << row
+    end
+    res
+  end
+  def target_hash_for_row(row)
+    h = {}
+    columns.each do |col|
+      h[col.target_name] = col.target_value(row)
+    end
+    h
+  end
+  def target_hashes
+    source_rows.map { |x| target_hash_for_row(x) }
+  end
+  def target_column_names
+    columns.map { |x| x.target_name }
+  end
+  def new_struct
+    Struct.new(*target_column_names)
+  end
+  fattr(:migration) do
+    raise "must define table" unless table_name
+    cls = Class.new(ActiveRecord::Migration)
+    class << cls
+      attr_accessor :cols, :table_name
+    end
+    cls.cols = columns
+    cls.table_name = table_name
+    puts "Table: #{table_name}"
+    cls.class_eval do
+      def self.up
+        create_table table_name do |t|
+          cols.each do |col|
+            t.column col.target_name, :string
+          end
+        end
+      end
+    end
+    cls
+  end
+  fattr(:ar) do
+    cls = Class.new(ActiveRecord::Base)
+    cls.send(:set_table_name, table_name)
+    cls
+  end
+  def migrate!
+    ar.find(:first)
+  rescue => exp
+    puts "find failed"
+    puts exp.inspect
+    migration.migrate(:up)
+  end
+  fattr(:ar_objects) do
+    target_hashes.map { |h| ar.new(h) }
+  end
+  def load!
+    ActiveRecord::Base.establish_connection(db_ops)
+    migrate!
+    ar_objects.each { |x| x.save! }
+  end
+end
+
+class Column
+  include FromHash
+  attr_accessor :target_name, :blk
+  def target_value(row)
+    if blk.arity == 1
+      blk.call(row)
+    else
+      row.instance_eval(&blk)
+    end
+  end
+end 
+
+class LoaderDSL
+  fattr(:loader) { Loader.new }
+  def column(name,type,&blk)
+    blk ||= lambda { |x| x.send(name) }
+    loader.columns << Column.new(:target_name => name, :blk => blk)
+  end
+  def method_missing(sym,*args,&b)
+    if [:string, :text, :integer, :float, :decimal, :datetime, :timestamp, :time, :date, :binary, :boolean].include?(sym)
+      column(args.first,sym,&b)
+    else
+      super(sym,*args,&b)
+    end
+  end
+  def source(file)
+    loader.source_filename = file
+  end
+  def database(ops)
+    loader.db_ops = ops
+  end
+  def table(name)
+    loader.table_name = name
+  end
+end
+
+def dataload(&b)
+  dsl = LoaderDSL.new
+  dsl.instance_eval(&b)
+  dsl.loader.load!
+  puts "Row Count: " + dsl.loader.ar.find(:all).size.to_s
+end
+
+dataload do
+  source "source.csv"
+  database :adapter => 'sqlite3', :database => "db.sqlite3", :timeout => 5000
+  table 'foobar'
+  string(:cat) { bar + "_cat" }
+end
+

data/spec/dataload_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe "Dataload" do
+  it "fails" do
+    fail "hey buddy, you should probably rename this file and start specing for real"
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+require 'spec'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'dataload'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification 
+name: GFunk911-dataload
+version: !ruby/object:Gem::Version 
+  version: 0.3.0
+platform: ruby
+authors: 
+- Mike Harris
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-03-12 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: GFunk913@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+- LICENSE
+files: 
+- README.rdoc
+- VERSION.yml
+- lib/dataload
+- lib/dataload/db.sqlite3
+- lib/dataload/loader.rb
+- lib/dataload/source.csv
+- lib/dataload.rb
+- spec/dataload_spec.rb
+- spec/spec_helper.rb
+- LICENSE
+has_rdoc: true
+homepage: http://github.com/GFunk911/dataload
+post_install_message: 
+rdoc_options: 
+- --inline-source
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: TODO
+test_files: []
+