rails-archiver 0.1.8 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 22767f51d1fefb73fd2d17b703a0ee846509bb99fc4dd2bed27c06a28fcdfc60
-  data.tar.gz: d04f42ec798d1df81fc3ee76575d4e1658563b73e4087f1e1639553b489b52b6
+  metadata.gz: 3b31d762ea01d15091eebffd02a581b9ae2a20fba31005ec2669b4cf0d350125
+  data.tar.gz: 6f553c01c3a2d02062c74384c09a8e220845907544009f8d64ea97313afba6af
 SHA512:
-  metadata.gz: a9b6176a12fd8fd673ebb726dcaa4dd08f2a3e561718685693870f78d0661b863e7ca03b2ea8a9da8f8e18e73f1128115c20af17f65aee08e070911584473958
-  data.tar.gz: 176e5169d8d4756fdae4ff8fcbe5e8f5bcde4395a262a332321e7cbbdf1e61530e7373b1fd389217d5c747190d4dbb0023aa1aab48c29df5128ac7827a7a502d
+  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
 
@@ -58,7 +64,7 @@ module RailsArchiver
         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
 
@@ -114,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.8'
-  s.date         = '2024-07-03'
+  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.8
+  version: 0.2.0
 platform: ruby
 authors:
 - Daniel Orner
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-07-03 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