relational_exporter 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 93369bce5ffa225e191684a1b1655bede21b40ce
-  data.tar.gz: 4b811e535dedd706361921e85d204938436c8912
+  metadata.gz: 27528505c9ae963ac90a8dec1a21aadba17fecd7
+  data.tar.gz: e84bb8298c019cc46dca592d3da020bf03f6c564
 SHA512:
-  metadata.gz: 9eae62da777df4f3991a206323d80351cced31ff9d86a2547d884ea1c92058c16198ded18819f877769fdd9983145b960ea0a9dd063001e2831e310c3506e83e
-  data.tar.gz: 5c2d7546a87f69f34a481873769b619bab48cf6cbc751f5b75162f703ff88e1a9eb28928a8428832e5efa8af62a2e97b6bdf6ae185df4e9c97617ffc10ce625e
+  metadata.gz: 12fc7ca1f8478f07f1c76f23fa05e91a4d89620f9656ab72d133c768c2b5e5cc8fbe8397f3e2742a9f35b47ef3e7cf28d271205334df573c8aacbb8f7201ae01
+  data.tar.gz: 202c03846dcc21b7cd8f0c367d6ff841c346cc6cd5e8f6f79af0834966f0ee06a5e8d9fad5aa50e04f7d72ce22b0ea4a62b746863f4f022a96813e366a0e41a8

data/README.md

@@ -78,9 +78,14 @@ Or install it yourself as:
   }
 
   # Run the export!
-  r = RelationalExporter::Runner.new schema: my_schema, connection_config: my_conn
+  r = RelationalExporter::Runner.new schema: my_schema, connection_config: my_conn, logger: Logger.new(STDERR)
   r.export(my_output_config) do |record|
-    # modify the record and it's models/associations
+    # examine the record and it's associations using any
+    # Active Record models that were defined in your schema!
+    puts record.class # => Person
+
+    # filter out the record if desired
+    record = nil if record.name == 'Bob'
   end
 ```
 

data/lib/relational_exporter.rb

@@ -6,11 +6,10 @@ require 'relational_exporter/active_record_extension'
 
 module RelationalExporter
   class Runner
-    attr_accessor :schema
+    attr_accessor :schema, :logger
 
     def initialize(options={})
-      # TODO - disable when not byebugging!!!
-      ActiveRecord::Base.logger = Logger.new(STDOUT)
+      @logger = options[:logger] ? options[:logger] : Logger.new(STDERR)
 
       @connection_config = options[:connection_config]
       begin
@@ -26,6 +25,8 @@ module RelationalExporter
     end
 
     def export(output_config, &block)
+      ActiveRecord::Base.logger = @logger
+
       output_config = Hashie::Mash.new output_config
 
       main_klass = output_config.output.model.to_s.classify.constantize
@@ -35,7 +36,16 @@ module RelationalExporter
       header_row = []
       max_associations = {}
 
-      ::CSV.open('/tmp/test.csv', 'wb', headers: true) do |csv|
+      csv_options = {headers: true}
+      if output_config.file_path.blank?
+        csv_method = :instance
+        csv_args = [STDOUT, csv_options]
+      else
+        csv_method = :open
+        csv_args = [output_config.file_path, 'wb', csv_options]
+      end
+
+      ::CSV.send(csv_method, *csv_args) do |csv|
         main_klass.find_all_by_scope(output_config.output.scope.as_json).find_in_batches do |batch|
           batch.each do |single|
             if block_given?
@@ -47,15 +57,7 @@ module RelationalExporter
             row = []
 
             # Add main record headers
-            if !main_klass.active_model_serializer.blank?
-              main_attributes = main_klass.active_model_serializer.new(single).as_json(root: false) rescue nil
-            elsif defined?(BaseSerializer)
-              main_attributes = BaseSerializer.new(single).as_json(root: false) rescue nil
-            end
-
-            main_attributes = single.attributes if main_attributes.nil?
-
-            main_attributes.each do |field, value|
+            serialized_attributes(single).each do |field, value|
               header_row << [main_klass.to_s.underscore, field].join('_').classify if csv.header_row?
               row << value
             end
@@ -77,7 +79,7 @@ module RelationalExporter
 
               foreign_key = main_klass.reflections[association_accessor].foreign_key rescue nil
 
-              fields = association_klass.first.attributes.keys
+              fields = serialized_attributes(association_klass).keys
 
               fields.reject! {|v| v == foreign_key } if foreign_key
 
@@ -93,6 +95,8 @@ module RelationalExporter
                   max_associated = 1
                 end
 
+                max_associated = 0 if max_associated.nil?
+
                 max_associations[association_accessor] = max_associated
 
                 max_associated.times do |i|
@@ -107,7 +111,7 @@ module RelationalExporter
 
             csv << header_row if csv.header_row?
             if row.count != header_row.count
-              puts "OH SHIT, this row is not right!"
+              @logger.error "Encountered invalid row, skipping."
             end
             csv << row
           end
@@ -117,10 +121,28 @@ module RelationalExporter
 
     private
 
+    def serialized_attributes(object)
+      return nil if object.nil?
+
+      klass, model = object.is_a?(Class) ? [object, object.first] : [object.class, object]
+
+      return {} if model.nil?
+
+      if model.respond_to?(:active_model_serializer) && !model.active_model_serializer.nil?
+        serialized = model.active_model_serializer.new(model).as_json(root: false) rescue nil
+      # elsif defined?(BaseSerializer)
+        # serialized = BaseSerializer.new(model).as_json(root: false) rescue nil
+      end
+
+      serialized = model.attributes if serialized.nil?
+      serialized
+    end
+
     def get_row_arr(records, fields, max_count=1, &block)
       max_count.times do |i|
+        record = serialized_attributes records[i]
         fields.each do |field|
-          val = records[i][field] rescue nil
+          val = record[field] rescue nil
           yield val
         end
       end

data/lib/relational_exporter/version.rb

@@ -1,3 +1,3 @@
 module RelationalExporter
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: relational_exporter
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Andrew Hammond
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-03 00:00:00.000000000 Z
+date: 2014-02-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: hashie