rails-archiver 0.1.7 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 631c54ba5ee068f6e555a6e5402a6543af7b117c
-  data.tar.gz: 4d194e22a65b774c3dcce89fe176a9967b834a60
+SHA256:
+  metadata.gz: 3b31d762ea01d15091eebffd02a581b9ae2a20fba31005ec2669b4cf0d350125
+  data.tar.gz: 6f553c01c3a2d02062c74384c09a8e220845907544009f8d64ea97313afba6af
 SHA512:
-  metadata.gz: 171728f0a785320566a37c212e3758b6de78c9d2d3304e7b477b4f7b383d182df87152c2c1ff19aae6af8f3ad20066ef2c1920f8ed2966b64bafb130855043ee
-  data.tar.gz: 79917bf2e130af7934a1b603bed51d542a69bf038cff449614b2ff6d2f332164a946e11559d618622d5e96535021e0921241d9a161b50f10974bdc3ed6cae574
+  metadata.gz: 11961ec8a0cacd1101792d3beade62fa79eb224a3c254edcb42a67a0a0f16ce423f7fefec0a5adb1b535ee5a33864e6dd5e02d8e8df22b83f1f490925741e9ae
+  data.tar.gz: 3f37961ca490f99900f3de97e0e76d4b6ae83e9fa7977ebadedfb55784102b8d33dc8975aead38af477d3732350774212b8b11f323ae3a093a05aaca34a4811f

data/.gitignore

@@ -0,0 +1,3 @@
+### Ruby template
+*.gem
+.idea

data/CHANGELOG.md

@@ -0,0 +1,13 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## UNRELEASED
+
+# [0.2.0] - 2025-01-20
+- Added support for enum columns - should be able to handle both keys and values.
+- Refactored the `unarchiver` method to more easily subclass `Unarchiver`.
+- For `belongs_to` associations, replace foreign key attributes with actual reference to the relevant object while archiving to avoid issues with validation.

data/lib/rails-archiver/archiver.rb

@@ -128,9 +128,18 @@ module RailsArchiver
       @logger.info("Finished deleting from #{table}")
     end
 
+    # @return [Class]
+    def unarchiver_class
+      RailsArchiver::Unarchiver
+    end
+
     # @return [RailsArchiver::Unarchiver]
-    def unarchiver
-      unarchiver = RailsArchiver::Unarchiver.new(@model, :logger => @logger)
+    # @param crash_on_errors [Boolean]
+    def unarchiver(crash_on_errors: false)
+      unarchiver = unarchiver_class.new(@model,
+                                        logger: @logger,
+                                        crash_on_errors: crash_on_errors
+      )
       unarchiver.transport = self.transport
       unarchiver
     end

data/lib/rails-archiver/unarchiver.rb

@@ -11,18 +11,24 @@ module RailsArchiver
     attr_accessor :errors, :transport
 
     # @param model [ActiveRecord::Base]
-    # @param options [Hash]
-    #   * logger [Logger]
-    #   * new_copy [Boolean] if true, create all new objects instead of
-    #                        replacing existing ones.
-    #   * crash_on_errors [Boolean] if true, do not do any imports if any
-    #                               models' validations failed.
-    def initialize(model, options={})
+    # @param logger [Logger]
+    # @param new_copy [Boolean] if true, create all new objects instead of
+    #                           replacing existing ones.
+    # @param crash_on_errors [Boolean] if true, do not do any imports if any
+    #                           models' validations failed.
+    # @param transport [Symbol|RailsArchiver::Transport::Base]
+    def initialize(model,
+                   transport: :in_memory,
+                   logger: Logger.new(STDOUT),
+                   new_copy: false, crash_on_errors: false)
       @model = model
-      @logger = options.delete(:logger) || Logger.new(STDOUT)
-      @options = options
+      @logger = logger
+      @options = {
+        new_copy: new_copy,
+        crash_on_errors: crash_on_errors
+      }
       # Transport for downloading
-      self.transport = _get_transport(options.delete(:transport) || :in_memory)
+      self.transport = _get_transport(transport)
       self.errors = []
     end
 
@@ -40,7 +46,7 @@ module RailsArchiver
       if @model
         @model.reload
         if @model.attribute_names.include?('archived')
-          @model.update_attribute(:archived, false)
+          @model.class.where(id: @model.id).update_all(:archived => false)
         end
       end
       @logger.info("#{source} load complete!")
@@ -51,10 +57,14 @@ module RailsArchiver
     def load_classes(hash)
       full_hash = hash.with_indifferent_access
       full_hash.each do |key, vals|
-        save_models(key.constantize, vals)
+        begin
+          save_models(key.constantize, vals)
+        rescue NameError => e
+          @logger.error("Could not save models for #{key}: #{e.message}")
+        end
       end
       if @options[:crash_on_errors] && self.errors.any?
-        raise ImportError.new("Errors occurred during load - please see 'errors' method for more details")
+        raise ImportError.new("Errors occurred during load: #{self.errors.join("\n")}")
       end
     end
 
@@ -110,13 +120,52 @@ module RailsArchiver
       end
 
       model
+    rescue => e
+      self.errors << "Error importing class #{klass.name}: #{e.message}"
     end
 
     private
 
+    # @param model [ActiveRecord::Base]
+    # @param field [String]
+    # @param attrs [Hash]
+    def assign_enum_value(model, field, attrs)
+      entries = model.class.public_send(field.pluralize)
+      is_integer = entries.values.first.is_a?(Integer)
+      value = attrs[field]
+      valid = if entries.keys.include?(value)
+                true
+              elsif is_integer && entries.values.include?(value.to_i)
+                true
+              else
+                entries.values.include?(value)
+              end
+
+      return if valid
+
+      @logger.warn("Invalid value for #{field}: #{attrs[field]}")
+      new_value = model.class.columns.find { |a| a.name == field }.default
+      new_value = new_value.to_i if is_integer && new_value
+      attrs[field] = new_value
+    end
+
     # @param model [ActiveRecord::Base]
     # @param attrs [Hash]
     def _assign_attributes(model, attrs)
+      # Handle integer enum values that were turned into strings
+      if model.class.respond_to?(:attribute_types)
+        enums = model.class.attribute_types.select { |_, v| v.is_a?(ActiveRecord::Enum::EnumType) }
+        enums.keys.each do |field|
+          assign_enum_value(model, field, attrs)
+        end
+      end
+      model.class.reflect_on_all_associations(:belongs_to).each do |assoc|
+        if attrs[assoc.foreign_key].present?
+          record = assoc.klass.find_by(assoc.association_primary_key => attrs[assoc.foreign_key])
+          attrs[assoc.name] = record if record
+        end
+      end
+
       if model.method(:attributes=).arity == 1
         model.attributes = attrs
       else

data/rails-archiver.gemspec

@@ -1,8 +1,8 @@
 Gem::Specification.new do |s|
   s.name         = 'rails-archiver'
   s.require_paths = %w(. lib lib/rails-archiver)
-  s.version      = '0.1.7'
-  s.date         = '2018-06-20'
+  s.version      = '0.2.0'
+  s.date         = '2025-01-20'
   s.summary      = 'Fully archive a Rails model'
   s.description  = <<-EOF
   EOF

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails-archiver
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.2.0
 platform: ruby
 authors:
 - Daniel Orner
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2018-06-20 00:00:00.000000000 Z
+date: 2025-01-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -44,6 +44,8 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
+- CHANGELOG.md
 - README.md
 - lib/rails-archiver.rb
 - lib/rails-archiver/archiver.rb
@@ -56,7 +58,7 @@ homepage: https://github.com/dorner/rails-archiver
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - "."
@@ -73,9 +75,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.14
-signing_key: 
+rubygems_version: 3.4.10
+signing_key:
 specification_version: 4
 summary: Fully archive a Rails model
 test_files: []