crhym3-imexport 0.1.0

data/Manifest

@@ -0,0 +1,6 @@
+Rakefile
+imexport.gemspec
+Manifest
+CHANGELOG
+lib/imexport.rb
+README.rdoc

data/README.rdoc

@@ -0,0 +1,4 @@
+== Description
+
+TODO
+

data/Rakefile

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'rake'
+require 'echoe'
+
+Echoe.new('imexport', '0.1.0') do |p|
+  p.description     = "Simple import from a text file generated by mysql -E ..."
+  p.url             = "http://github.com/crhym3/imexport"
+  p.author          = "alex"
+  p.email           = "alex@digns.com"
+  p.ignore_pattern = ["tmp/*", "script/*"]
+  p.development_dependencies = []
+end
+
+Dir["#{File.dirname(__FILE__)}/tasks/*.rake"].sort.each { |ext| load ext }
+

data/imexport.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{imexport}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["alex"]
+  s.date = %q{2009-04-21}
+  s.description = %q{Simple import from a text file generated by mysql -E ...}
+  s.email = %q{alex@digns.com}
+  s.extra_rdoc_files = ["CHANGELOG", "lib/imexport.rb", "README.rdoc"]
+  s.files = ["Rakefile", "imexport.gemspec", "Manifest", "CHANGELOG", "lib/imexport.rb", "README.rdoc"]
+  s.has_rdoc = true
+  s.homepage = %q{http://github.com/crhym3/imexport}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Imexport", "--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{imexport}
+  s.rubygems_version = %q{1.3.2}
+  s.summary = %q{Simple import from a text file generated by mysql -E ...}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

data/lib/imexport.rb

@@ -0,0 +1,104 @@
+module ImExport
+  def self.import(file_name, options = {})
+  
+    ## e.g. :seminar => "seminar" => "Seminar" => Seminar
+    model = Kernel.const_get(options[:class_name].to_s.classify)
+    
+    ## used to check for an existing record. so we do update_attributes()
+    # instead of save() in that case
+    find_method = "find_by_#{options[:find_by]}"
+    
+    ## helps to distinguish between real columns and column content
+    # could be something like "COLUMN_"
+    columns_prefix = options[:db_columns_prefix].to_s
+    
+    ## for each line, this is how we check for a new column or cont.
+    # from the previous line
+    scan_regexp = Regexp.new("^\\s*#{columns_prefix}(\\w+)\\: (.*)")
+    
+    ## table columns --> model attributes mapping
+    # if an attribute is not specified and present as a column,
+    # it'll try figure it out:
+    # { 'title' => :title } - works w/o expicit mapping
+    attr_map = options[:map] || {}
+    
+    ## Read the file
+    # we assume it's been created with vertial columns layout (mysql -E ...)
+    IO.foreach(file_name) do |line|
+      ## mysql -E ... does this:
+      # ********* 1. row **********
+      # column: value
+      # another_column: value
+      # ********* 2. row **********
+      unless (line =~ /^\*+ \d+\. row \*+$/).nil?
+        ## We've got a new row
+        unless block_given?
+          ## Save previously created object ...
+          if !@model.nil? and @model.valid?
+            if (s = model.send("#{find_method}", @model.title))
+              s.update_attributes(@model.attributes.reject{|k,v| v.nil?})
+            else
+              @model.save
+            end
+          else 
+            $stderr.puts "\n>>> ERRORS while storing #{options[:class_name]}: #{@model.errors.full_messages.join('; ')}\n#{@model.inspect}" unless @model.nil?
+          end
+        else
+          ## or pass it to a block
+          # in this case we don't do any validations
+          yield(@model)
+        end
+        
+        # ... then create a new one
+        #$stderr.puts "\n###################################"
+        @model = model.new
+        @last_column_name = nil
+        @last_column_content = nil
+        next
+      end
+      
+      ## Cont. of the same row.
+      ## It's one table column at a time (or more lines).
+      ## Parse it and add set the corresponding attribute in the model
+      line.chomp!
+      column = line.scan(scan_regexp)
+      #$stderr.puts column.inspect
+      
+      unless column.size > 0
+        ## this is a column continuation of the previous line
+        @last_column_content << '<br/>' << line
+        next
+      end
+      
+      ## this is a new model attribute
+      # set last attribute in the model if defined
+      # we should have at least two items in the array
+      column.flatten!
+      @last_column_name = column.first
+      @last_column_content = column[1].gsub(/\t+/, " ").strip.gsub(/^\n+$/, "").gsub(/\n+/, "<br/>")
+      
+      # in column-to-model map
+      if attr_map.include?(@last_column_name)
+        mattr = attr_map[@last_column_name]
+        case
+          when mattr.kind_of?(Symbol)
+            @model.send("#{mattr.to_s}=", @last_column_content)
+          when mattr.kind_of?(Proc)
+            mattr.call(@last_column_content, @model)
+          when mattr.kind_of?(Hash)
+            @model.send("#{mattr.keys.first}=", mattr.values.first.call(@last_column_content))
+          else
+            $stderr.puts "WARNING: don't know how to handle #{mattr.inspect}"
+        end
+        
+      # simple auto-mapping  
+      elsif @model.respond_to?(@last_column_name.to_sym)
+        @model.send("#{@last_column_name}=", @last_column_content)
+        
+      # otherwise we don't know how to do it
+      else
+        $stderr.puts "WARNING: don't know how to set :#{@last_column_name}"
+      end
+    end # IO.foreach
+  end
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification 
+name: crhym3-imexport
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- alex
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-04-21 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Simple import from a text file generated by mysql -E ...
+email: alex@digns.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- CHANGELOG
+- lib/imexport.rb
+- README.rdoc
+files: 
+- Rakefile
+- imexport.gemspec
+- Manifest
+- CHANGELOG
+- lib/imexport.rb
+- README.rdoc
+has_rdoc: true
+homepage: http://github.com/crhym3/imexport
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Imexport
+- --main
+- README.rdoc
+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: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: imexport
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 3
+summary: Simple import from a text file generated by mysql -E ...
+test_files: []
+