datashift 0.1.0 → 0.2.1

data/lib/generators/csv_generator.rb

@@ -1,36 +1,36 @@
-# Copyright:: (c) Autotelik Media Ltd 2011
-# Author ::   Tom Statter
-# Date ::     Aug 2011
-# License::   MIT
-#
-# Details::   Export a model to CSV
-#
-#
-require 'generator_base'
-
-module DataShift
-
-  class CsvGenerator < GeneratorBase
-
-    attr_accessor :excel, :filename
-  
-    def initialize(filename)
-      @excel = nil
-      @filename = filename
-    end
-
-    # Create CSV file representing supplied Model
-    
-    def generate(model, options = {})
-
-      @filename = options[:filename] if  options[:filename]
-    end
-
-  
-    # Create an Csv file representing supplied Model
-
-    def export(items, options = {})
-    end
-
-  end
-end
+# Copyright:: (c) Autotelik Media Ltd 2011
+# Author ::   Tom Statter
+# Date ::     Aug 2011
+# License::   MIT
+#
+# Details::   Export a model to CSV
+#
+#
+require 'generator_base'
+
+module DataShift
+
+  class CsvGenerator < GeneratorBase
+
+    attr_accessor :excel, :filename
+  
+    def initialize(filename)
+      @excel = nil
+      @filename = filename
+    end
+
+    # Create CSV file representing supplied Model
+    
+    def generate(model, options = {})
+
+      @filename = options[:filename] if  options[:filename]
+    end
+
+  
+    # Create an Csv file representing supplied Model
+
+    def export(items, options = {})
+    end
+
+  end
+end

data/lib/generators/excel_generator.rb

@@ -1,123 +1,123 @@
-# Copyright:: (c) Autotelik Media Ltd 2011
-# Author ::   Tom Statter
-# Date ::     Aug 2011
-# License::   MIT
-#
-# Details::   Export a model to Excel '97(-2007) file format.
-#
-# TOD : Can we switch between .xls and XSSF (POI implementation of Excel 2007 OOXML (.xlsx) file format.)
-#
-#
-module DataShift
-
-  require 'generator_base'
-      
-  if(Guards::jruby?)
-
-    require 'jruby/jexcel_file'
-
-    class ExcelGenerator < GeneratorBase
-
-      attr_accessor :filename
-  
-      def initialize(filename)
-        @filename = filename
-      end
-
-      # Create an Excel file template (header row) representing supplied Model
-    
-      def generate(model, options = {})
-        MethodDictionary.find_operators( model )
-
-        @filename = options[:filename] if  options[:filename]
-
-        excel = JExcelFile.new()
-
-        if(options[:sheet_name] )
-          excel.create_sheet( options[:sheet_name] ) 
-        else
-          excel.create_sheet( model.name )
-        end
-        
-        raise "Failed to create Excel WorkSheet for #{model.name}" unless excel.sheet
-
-        excel.set_headers(MethodDictionary.assignments[model])
-
-        excel.save( @filename )
-      end
-
-  
-      # Create an Excel file from list of ActiveRecord objects
-      def export(records, options = {})
-
-        excel = JExcelFile.new()
-
-        if(options[:sheet_name] )
-          excel.create_sheet( options[:sheet_name] ) 
-        else
-          excel.create_sheet( records.first.class.name )
-        end
-      
-        excel.ar_to_headers(records)
-        
-        excel.ar_to_xls(records)
-
-        excel.save( filename() )
-      end
-      
-      # Create an Excel file from list of ActiveRecord objects
-      # Specify which associations to export via :with or :exclude
-      # Possible values are : [:assignment, :belongs_to, :has_one, :has_many]
-      #
-      def export_with_associations(klass, items, options = {})
-
-        excel = JExcelFile.new()
-
-        if(options[:sheet_name] )
-          excel.create_sheet( options[:sheet_name] ) 
-        else
-          excel.create_sheet( items.first.class.name )
-        end
-        
-        MethodDictionary.find_operators( klass )
-         
-        MethodDictionary.build_method_details( klass )
-          
-        work_list = (options[:with]) ? options[:with] : [:assignments, :belongs_to, :has_one, :has_many]
-        
-        headers = []
-        
-        work_list.each do |mdtype|
-          method_details = MethodDictionary.send("#{mdtype}_for", klass)
-        
-          method_details.each {|md| headers << "#{md.operator}" }
-        end
-        
-        excel.set_headers( headers )
-                
-        data = []
-        
-        items.each do |record|
-          
-          MethodMapper.method_details[klass].each do |method_detail|   
-            if(method_detail.operator_type == :assignment)
-              data << record.send( method_detail.operator )
-            end
-          end
-        end
-        
-        excel.set_row(2,1,items)
-
-        excel.save( filename() )
-      end
-    end # ExcelGenerator
-
-  else
-    class  ExcelGenerator < GeneratorBase
-      def initialize
-        raise DataShift::BadRuby, "Please install and use JRuby for working with .xls files"
-      end
-    end
-  end # jruby
-  
+# Copyright:: (c) Autotelik Media Ltd 2011
+# Author ::   Tom Statter
+# Date ::     Aug 2011
+# License::   MIT
+#
+# Details::   Export a model to Excel '97(-2007) file format.
+#
+# TOD : Can we switch between .xls and XSSF (POI implementation of Excel 2007 OOXML (.xlsx) file format.)
+#
+#
+module DataShift
+
+  require 'generator_base'
+      
+  if(Guards::jruby?)
+
+    require 'jruby/jexcel_file'
+
+    class ExcelGenerator < GeneratorBase
+
+      attr_accessor :filename
+  
+      def initialize(filename)
+        @filename = filename
+      end
+
+      # Create an Excel file template (header row) representing supplied Model
+    
+      def generate(model, options = {})
+        MethodDictionary.find_operators( model )
+
+        @filename = options[:filename] if  options[:filename]
+
+        excel = JExcelFile.new()
+
+        if(options[:sheet_name] )
+          excel.create_sheet( options[:sheet_name] ) 
+        else
+          excel.create_sheet( model.name )
+        end
+        
+        raise "Failed to create Excel WorkSheet for #{model.name}" unless excel.sheet
+
+        excel.set_headers(MethodDictionary.assignments[model])
+
+        excel.save( @filename )
+      end
+
+  
+      # Create an Excel file from list of ActiveRecord objects
+      def export(records, options = {})
+
+        excel = JExcelFile.new()
+
+        if(options[:sheet_name] )
+          excel.create_sheet( options[:sheet_name] ) 
+        else
+          excel.create_sheet( records.first.class.name )
+        end
+      
+        excel.ar_to_headers(records)
+        
+        excel.ar_to_xls(records)
+
+        excel.save( filename() )
+      end
+      
+      # Create an Excel file from list of ActiveRecord objects
+      # Specify which associations to export via :with or :exclude
+      # Possible values are : [:assignment, :belongs_to, :has_one, :has_many]
+      #
+      def export_with_associations(klass, items, options = {})
+
+        excel = JExcelFile.new()
+
+        if(options[:sheet_name] )
+          excel.create_sheet( options[:sheet_name] ) 
+        else
+          excel.create_sheet( items.first.class.name )
+        end
+        
+        MethodDictionary.find_operators( klass )
+         
+        MethodDictionary.build_method_details( klass )
+          
+        work_list = (options[:with]) ? options[:with] : [:assignments, :belongs_to, :has_one, :has_many]
+        
+        headers = []
+        
+        work_list.each do |mdtype|
+          method_details = MethodDictionary.send("#{mdtype}_for", klass)
+        
+          method_details.each {|md| headers << "#{md.operator}" }
+        end
+        
+        excel.set_headers( headers )
+                
+        data = []
+        
+        items.each do |record|
+          
+          MethodMapper.method_details[klass].each do |method_detail|   
+            if(method_detail.operator_type == :assignment)
+              data << record.send( method_detail.operator )
+            end
+          end
+        end
+        
+        excel.set_row(2,1,items)
+
+        excel.save( filename() )
+      end
+    end # ExcelGenerator
+
+  else
+    class  ExcelGenerator < GeneratorBase
+      def initialize
+        raise DataShift::BadRuby, "Please install and use JRuby for working with .xls files"
+      end
+    end
+  end # jruby
+  
 end # DataShift

data/lib/generators/generator_base.rb

@@ -1,14 +1,14 @@
-# Copyright:: (c) Autotelik Media Ltd 2011
-# Author ::   Tom Statter
-# Date ::     Aug 2010
-# License::   MIT
-#
-#  Details::  Base class for generators, which provide serivrs to describe a Model in an external format
-#
-module DataShift
-
-  class GeneratorBase
-
-  end
-
+# Copyright:: (c) Autotelik Media Ltd 2011
+# Author ::   Tom Statter
+# Date ::     Aug 2010
+# License::   MIT
+#
+#  Details::  Base class for generators, which provide serivrs to describe a Model in an external format
+#
+module DataShift
+
+  class GeneratorBase
+
+  end
+
 end

data/lib/helpers/core_ext/to_b.rb

@@ -1,24 +1,24 @@
-class Object
-  def to_b
-    case self
-    when true, false then self
-    when nil then false
-    else
-      to_i != 0
-    end
-  end
-end
-
-class String
-  TRUE_REGEXP = /^(yes|true|on|t|1|\-1)$/i.freeze
-  FALSE_REGEXP = /^(no|false|off|f|0)$/i.freeze
-
-  def to_b
-    case self
-    when TRUE_REGEXP then true
-    when FALSE_REGEXP then false
-    else
-      to_i != 0
-    end
-  end
-end
+class Object
+  def to_b
+    case self
+    when true, false then self
+    when nil then false
+    else
+      to_i != 0
+    end
+  end
+end
+
+class String
+  TRUE_REGEXP = /^(yes|true|on|t|1|\-1)$/i.freeze
+  FALSE_REGEXP = /^(no|false|off|f|0)$/i.freeze
+
+  def to_b
+    case self
+    when TRUE_REGEXP then true
+    when FALSE_REGEXP then false
+    else
+      to_i != 0
+    end
+  end
+end

data/lib/helpers/spree_helper.rb

@@ -1,155 +1,153 @@
-# Copyright:: (c) Autotelik Media Ltd 2011
-# Author ::   Tom Statter
-# Date ::     Aug 2011
-# License::   MIT
-#
-# Details::   Spree Helper mixing in Support for testing or loading Rails Spree e-commerce.
-# 
-#             The Spree version you want to test should be picked up from the Gemfile
-# 
-#             Since datashift gem is not a Rails app or a Spree App, provides utilities to internally
-#             create a Spree Database, and to load Spree components, enabling standalone testing.
-#
-# =>          Has been tested with 
-# 
-# # =>        TODO - Can we move to a Gemfile/bunlder
-#             require 'rubygems'
-#             gemfile = File.expand_path("<%= gemfile_path %>", __FILE__)
-#
-#             ENV['BUNDLE_GEMFILE'] = gemfile
-#             require 'bundler'
-#             Bundler.setup
-#
-# =>          TODO - See if we can improve DB creation/migration ....
-#             N.B Some or all of Spree Tests may fail very first time run,
-#             as the database is auto generated
-# =>          
-module Spree
-
-  def self.root
-    Gem.loaded_specs['spree_core'] ? Gem.loaded_specs['spree_core'].full_gem_path  : ""
-  end
-
-  def self.lib_root
-    File.join(root, 'lib')
-  end
-
-  def self.app_root
-    File.join(root, 'app')
-  end
-
-  def self.load()
-    gem 'spree'
-  end
-    
-  def self.boot
- 
-    require 'spree'
-    require 'spree_core'
-
-    #require 'rake'
-    #require 'rubygems/package_task'
-    #require 'thor/group'
-    require File.expand_path( lib_root + '/generators/spree/install/install_generator')
-    require 'spree/core/testing_support/common_rake'
-
-    
-    Spree::SandboxGenerator.start ["--lib_name=spree", "--database=#{ENV['DB_NAME']}"]
-    Spree::InstallGenerator.start ["--auto-accept"]
-
-    cmd = "bundle exec rake assets:precompile:nondigest"; 
-    puts cmd; system cmd
-
-    
-    return
-    
- 
-    # TODO how to check gem version actually loaded and do conditional
-    #
-    #if(PRE 1)
-      #require 'spree_core/preferences/model_hooks'
-    #
-    ## Initialize preference system
-    # ActiveRecord::Base.class_eval do
-    #   include Spree::Preferences
-    #   include Spree::Preferences::ModelHooks
-    # end
-    #end
-    
-    gem 'paperclip'
-    gem 'nested_set'
-
-    require 'nested_set'
-    require 'paperclip'
-    require 'acts_as_list'
-
-    CollectiveIdea::Acts::NestedSet::Railtie.extend_active_record
-    ActiveRecord::Base.send(:include, Paperclip::Glue)
-
-    gem 'activemerchant'
-    require 'active_merchant'
-    require 'active_merchant/billing/gateway'
-
-    ActiveRecord::Base.send(:include, ActiveMerchant::Billing)
-
-     
-    #require 'scopes'
-    
-    # Not sure how Rails manages this seems lots of circular dependencies so
-    # keep trying stuff till no more errors
-    
-    Dir[lib_root + '/*.rb'].each do |r|
-      begin
-        require r if File.file?(r)  
-      rescue => e
-      end
-    end
-
-    Dir[lib_root + '/**/*.rb'].each do |r|
-      begin
-        require r if File.file?(r) && ! r.include?('testing')  && ! r.include?('generators')
-      rescue => e
-      end
-    end
-    
-    load_models( true )
-
-    Dir[lib_root + '/*.rb'].each do |r|
-      begin
-        require r if File.file?(r)  
-      rescue => e
-      end
-    end
-
-    Dir[lib_root + '/**/*.rb'].each do |r|
-      begin
-        require r if File.file?(r) && ! r.include?('testing')  && ! r.include?('generators')
-      rescue => e
-      end
-    end
-
-    #if(PRE 1)require 'product'
-    #require 'lib/product_filters'
-    
-    
-    load_models( true )
-
-  end
-
-  def self.load_models( report_errors = nil )
-    puts 'Loading Spree models from', root
-    Dir[root + '/app/models/**/*.rb'].each {|r|
-      begin
-        require r if File.file?(r)
-      rescue => e
-        puts("WARNING failed to load #{r}", e.inspect) if(report_errors == true)
-      end
-    }
-  end
-
-  def self.migrate_up
-    load
-    ActiveRecord::Migrator.up( File.join(root, 'db/migrate') )
-  end
-
-end
+# Copyright:: (c) Autotelik Media Ltd 2011
+# Author ::   Tom Statter
+# Date ::     Aug 2011
+# License::   MIT
+#
+# Details::   Spree Helper mixing in Support for testing or loading Rails Spree e-commerce.
+# 
+#             The Spree version you want to test should be picked up from the Gemfile
+# 
+#             Since datashift gem is not a Rails app or a Spree App, provides utilities to internally
+#             create a Spree Database, and to load Spree components, enabling standalone testing.
+#
+# =>          Has been tested with  0.7
+# 
+# # =>        TODO - Can we move to a Gemfile/bunlder
+#             require 'rubygems'
+#             gemfile = File.expand_path("<%= gemfile_path %>", __FILE__)
+#
+#             ENV['BUNDLE_GEMFILE'] = gemfile
+#             require 'bundler'
+#             Bundler.setup
+#
+# =>          TODO - See if we can improve DB creation/migration ....
+#             N.B Some or all of Spree Tests may fail very first time run,
+#             as the database is auto generated
+# =>          
+module Spree
+
+  def self.root
+    Gem.loaded_specs['spree_core'] ? Gem.loaded_specs['spree_core'].full_gem_path  : ""
+  end
+
+  def self.lib_root
+    File.join(root, 'lib')
+  end
+
+  def self.app_root
+    File.join(root, 'app')
+  end
+
+  def self.load()
+    gem 'spree'
+  end
+    
+  def self.boot
+ 
+    require 'spree'
+    require 'spree_core'
+
+    if(Gem.loaded_specs['spree'].version.version.to_f < 1)
+      boot_pre_1
+    else
+      require File.expand_path( lib_root + '/generators/spree/install/install_generator')
+      require 'spree/core/testing_support/common_rake'
+
+
+      Spree::SandboxGenerator.start ["--lib_name=spree", "--database=#{ENV['DB_NAME']}"]
+      Spree::InstallGenerator.start ["--auto-accept"]
+
+      cmd = "bundle exec rake assets:precompile:nondigest"; 
+      puts cmd; system cmd
+    end
+  end
+
+  def self.boot_pre_1
+ 
+    require 'rake'
+    require 'rubygems/package_task'
+    require 'thor/group'
+
+    require 'spree_core/preferences/model_hooks'
+    #
+    # Initialize preference system
+    ActiveRecord::Base.class_eval do
+      include Spree::Preferences
+      include Spree::Preferences::ModelHooks
+    end
+ 
+    gem 'paperclip'
+    gem 'nested_set'
+
+    require 'nested_set'
+    require 'paperclip'
+    require 'acts_as_list'
+
+    CollectiveIdea::Acts::NestedSet::Railtie.extend_active_record
+    ActiveRecord::Base.send(:include, Paperclip::Glue)
+
+    gem 'activemerchant'
+    require 'active_merchant'
+    require 'active_merchant/billing/gateway'
+
+    ActiveRecord::Base.send(:include, ActiveMerchant::Billing)
+  
+    require 'scopes'
+    
+    # Not sure how Rails manages this seems lots of circular dependencies so
+    # keep trying stuff till no more errors
+    
+    Dir[lib_root + '/*.rb'].each do |r|
+      begin
+        require r if File.file?(r)  
+      rescue => e
+      end
+    end
+
+    Dir[lib_root + '/**/*.rb'].each do |r|
+      begin
+        require r if File.file?(r) && ! r.include?('testing')  && ! r.include?('generators')
+      rescue => e
+      end
+    end
+    
+    load_models( true )
+
+    Dir[lib_root + '/*.rb'].each do |r|
+      begin
+        require r if File.file?(r)  
+      rescue => e
+      end
+    end
+
+    Dir[lib_root + '/**/*.rb'].each do |r|
+      begin
+        require r if File.file?(r) && ! r.include?('testing')  && ! r.include?('generators')
+      rescue => e
+      end
+    end
+
+  #  require 'lib/product_filters'
+     
+    load_models( true )
+
+  end
+  
+  def self.load_models( report_errors = nil )
+    puts 'Loading Spree models from', root
+    Dir[root + '/app/models/**/*.rb'].each {|r|
+      begin
+        require r if File.file?(r)
+      rescue => e
+        puts("WARNING failed to load #{r}", e.inspect) if(report_errors == true)
+      end
+    }
+  end
+
+  def self.migrate_up
+    load
+    boot
+    ActiveRecord::Migrator.up( File.join(root, 'db/migrate') )
+  end
+
+end