activerecord-archiver 0.0.0 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0bea1df0fa0ebd3cdf2f49f5c2773357810a8dc6
-  data.tar.gz: d2153f6d8ff024cce3bdbcc71df74db5acb99d2f
+  metadata.gz: d643cb8c889c10a98f78294e99e0a3efa6e8e08b
+  data.tar.gz: 994f7221f37d80e5469c56d8882d1a2a75544530
 SHA512:
-  metadata.gz: f9aec4a4adccee5eba5488e7947323a51ac97e7c18395ff02d2a1cd0ac8bd7b7ae6d1fee25e8fe2f37fbd1af892494d20b4aad62037a5ec8fea5530ad65d37cf
-  data.tar.gz: 2f3fb76e833265d4a63e194fb710811f7f334748e50026916f23c71cff36a015cdea4b153542883d0bc2e47e0ee7a79c26c21143cc2354f778d5df0b9f714670
+  metadata.gz: 8e3c4233857329b4fbccbc38672ef088692592588121e164e871b4b6e74ca3767169cd1ef605c9e3c6c6861f5b98379c0fc9ebaa7a6611ca195ae29a4f7d31bd
+  data.tar.gz: 2b47faa160584694371ad67167d1e1639ea1e6cf609f28255ca412a5506cb494756df96662cc95474d98f827b91650be2752a1c90774ef3b7d3bbda57bcf8621

data/lib/activerecord-archiver/archiver.rb

@@ -76,4 +76,25 @@ class ActiveRecordArchiver
       raise "#{model} is not an activerecord model"
     end
   end
+  
+  def self.relation_from_key model, attribute
+    model.reflections.values.detect{|ref| ref.foreign_key == attribute}
+  end
+  
+  def self.cols_for_model model, all_models
+    model.columns.map do |col|
+      if col.primary
+        # omit primary keys
+        nil
+      elsif (relation = relation_from_key(model, col.name))
+        # include belongs_to relations to included models
+        if all_models.include? relation.klass
+          relation.name
+        else nil end
+      else
+        warn "Warning: #{model}##{col.name} included in export" if col.name =~ /_id$/
+        col.name
+      end
+    end.compact
+  end
 end

data/lib/activerecord-archiver/export.rb

@@ -36,9 +36,9 @@ Example:
 =end
 
 class ActiveRecordArchiver
-  def self.export models_hash
+  def self.export *args
     
-    @models_hash = models_hash
+    @models_hash = models_hash_from_args(args)
     
     result = {}
     
@@ -69,8 +69,6 @@ class ActiveRecordArchiver
               if placeholder
                 rec[relation_foreign_key(model, attribute)] = placeholder
               end
-            else
-              raise "#{record} belongs_to #{attribute} which is not included in the export"
             end
           else
             raise "#{attribute} is not an attribute or belongs_to relation of #{model}"
@@ -83,4 +81,27 @@ class ActiveRecordArchiver
     # encode
     JSON.dump result
   end
+  
+  def self.models_hash_from_args args
+    options = args.extract_options!
+    args.each do |arg|
+      options[arg] = :all
+    end
+    
+    models = options.each_pair.map do |collection, _|
+      collection.first.class
+    end
+    
+    models_hash = {}
+    
+    options.each_pair do |collection, cols|
+      models_hash[collection.first.class] =
+        [collection,
+         if cols == :all
+           cols_for_model(collection.first.class, models)
+         else cols end]
+    end
+    
+    models_hash
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-archiver
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.1.0
 platform: ruby
 authors:
 - Sam Auciello
@@ -9,7 +9,49 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2014-02-05 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: |
   ActiveRecord-Archiver is a simple tool for taking a subset of the records in one environment
   and exporting them for use in another environment.