infer_model 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 11b33f155e0db1415140efd8c4694aad4499ffdbd2c91cf239bba1d527fd2d40
-  data.tar.gz: 22514951be567f30c8743aa4c53749cb2441411ce77fd6017172b914d2d40629
+  metadata.gz: 89e0007e60d9ffa17f7ceb8f12e157be7c8d3570e3edaf5acd7ae9c3d397478e
+  data.tar.gz: c622cb756b83391f44328e0a85ca2b3cd46df5e7f171757deb2aca36f3f7cd3e
 SHA512:
-  metadata.gz: 57e31a6031eb46c9dee5e121e152a678fe60a464ee478901dd025556bbfcd2f5fba5ad8ca36142bfdb675aaca7c067828df185a34c9b4b8c7537b09bcde675a4
-  data.tar.gz: ef70c7981755957dd0ad8e0494793844b119fd323fc202fbb70019c227dc3f5c196121bb5a27b5016eeb672e0f62eddb186a7f785606af924245cd2ec73752cb
+  metadata.gz: f1a0328a1fa5fb4b3eaf22c7e7fe19e206041f847859e8b7291ebd1b4f1ec520b1e54fd6644f62733e619f6e7990b6571f189951f45266561dd9a365e1af4e43
+  data.tar.gz: 508256b0bb1a0e40f0b5ae7881c141dcea149fa3b74817d37542a77df8496f8cd485938afb9ea2ac085f68ee273b82b038350be0e4384fb8ed0e4edc1d64c234

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.1.2] - 2022-09-15
+
+- feat: Allow renaming attributes in to/migration & from/csv adapters
+
 ## [0.1.1] - 2022-09-13
 
 - Bump dependencies

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    infer_model (0.1.1)
+    infer_model (0.1.2)
       activesupport (~> 7.0)
       dry-initializer (~> 3.0)
       zeitwerk (~> 2.6)

data/lib/infer_model/from/csv.rb

@@ -13,6 +13,7 @@ module InferModel::From
     param :filename
     option :available_types, default: -> { ::InferModel::ValueTypeGuesser::RESULT_TYPES }
     option :multi, default: -> { false }
+    option :rename_attributes, optional: true
     option :csv_options, default: -> { {} }
 
     def call
@@ -26,12 +27,19 @@ module InferModel::From
     def attributes
       csv.by_col!.to_h do |header, contents|
         [
-          header.downcase.to_sym,
+          transform_header(header),
           ::InferModel::CommonTypeGuesser.call(contents, available_types:),
         ]
       end
     end
 
+    def transform_header(header)
+      header = header.downcase.to_sym
+      return header unless rename_attributes
+
+      rename_attributes.to_proc.call(header) || header
+    end
+
     def csv = ::CSV.parse(file_content, **csv_options_with_defaults)
 
     def file_content = File.read(filename)

data/lib/infer_model/to/migration.rb

@@ -10,9 +10,11 @@ module InferModel::To
     TIMESTAMP_FORMAT = "%Y%m%d%H%M%S"
 
     param :model
-    option :target_dir, default: -> { "db/migrate" }
-    option :table_name, optional: true
+    option :attributes_map, optional: true
+    option :filter_on_attribute_mapping, default: -> { true }
     option :rails_version, default: -> { "7.0" }
+    option :table_name, optional: true
+    option :target_dir, default: -> { "db/migrate" }
 
     def call
       FileUtils.mkdir_p(target_dir)
@@ -30,6 +32,16 @@ module InferModel::To
       table_name || model.source_name.pluralize
     end
 
+    def given_or_mapped_attributes
+      return model.attributes unless attributes_map
+
+      if filter_on_attribute_mapping
+        attributes_map.to_h { |old_key, new_key| [new_key, model.attributes[old_key]] }
+      else
+        model.attributes.transform_keys { |key| attributes_map.fetch(key, key) }
+      end
+    end
+
     def migration_content
       <<~RUBY
         # frozen_string_literal: true
@@ -49,13 +61,13 @@ module InferModel::To
     COLUMN_DDL_LINES_WITH_INDENTATION_JOINER = "\n#{"  " * 3}".freeze
 
     def column_ddl_lines
-      column_definitions = model.attributes.map do |key, common_type|
+      column_definitions = given_or_mapped_attributes.map do |key, common_type|
         attribute_and_name = %(t.#{common_type.detected_type} "#{key}")
         non_null_constraint = common_type.non_null_constraint_possible ? "null: false" : nil
 
         [attribute_and_name, non_null_constraint].compact.join(", ")
       end
-      index_definitions = model.attributes.filter_map do |key, common_type|
+      index_definitions = given_or_mapped_attributes.filter_map do |key, common_type|
         next unless common_type.unique_constraint_possible
 
         %(t.index ["#{key}"], unique: true)

data/lib/infer_model/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module InferModel
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: infer_model
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Jay Schneider
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-09-13 00:00:00.000000000 Z
+date: 2022-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport