infer_model 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: beae3b14bb858f6f8d597788daa62d18ab9246d4ccc81e8d862f08b5e6bcf99f
-  data.tar.gz: 9856dc12362b8f109260e272b2a392d61a3a54c22f02085ffe3146d35acd5ebf
+  metadata.gz: 89e0007e60d9ffa17f7ceb8f12e157be7c8d3570e3edaf5acd7ae9c3d397478e
+  data.tar.gz: c622cb756b83391f44328e0a85ca2b3cd46df5e7f171757deb2aca36f3f7cd3e
 SHA512:
-  metadata.gz: 49929b9598e50253e0eff456bd7a73894d782d95fb82fe90f5dd5c73bcd13eb2a59b4cb86da983b735b236652d9944f10370c35d6dd60265e0183f007cb0a177
-  data.tar.gz: 61c7a94a1f691dad8f6e3999cfcb876472e92c0ca4bf566f1e33d9d567d5835531a97743425317900fa0067664deaeab5ea244e5f3101c13b46ab338b9e8ac3f
+  metadata.gz: f1a0328a1fa5fb4b3eaf22c7e7fe19e206041f847859e8b7291ebd1b4f1ec520b1e54fd6644f62733e619f6e7990b6571f189951f45266561dd9a365e1af4e43
+  data.tar.gz: 508256b0bb1a0e40f0b5ae7881c141dcea149fa3b74817d37542a77df8496f8cd485938afb9ea2ac085f68ee273b82b038350be0e4384fb8ed0e4edc1d64c234

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## [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
+
 ## [0.1.0] - 2022-09-04
 
 - Initial release

data/Gemfile

@@ -4,5 +4,3 @@ source "https://rubygems.org"
 
 # Specify your gem's dependencies in infer_model.gemspec
 gemspec
-
-gem "rake"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    infer_model (0.1.0)
+    infer_model (0.1.2)
       activesupport (~> 7.0)
       dry-initializer (~> 3.0)
       zeitwerk (~> 2.6)
@@ -9,7 +9,7 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    activesupport (7.0.3.1)
+    activesupport (7.0.4)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
@@ -41,13 +41,13 @@ GEM
       rspec-mocks (~> 3.11.0)
     rspec-core (3.11.0)
       rspec-support (~> 3.11.0)
-    rspec-expectations (3.11.0)
+    rspec-expectations (3.11.1)
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.11.0)
     rspec-mocks (3.11.1)
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.11.0)
-    rspec-support (3.11.0)
+    rspec-support (3.11.1)
     rubocop (1.36.0)
       json (~> 2.3)
       parallel (~> 1.10)
@@ -60,7 +60,7 @@ GEM
       unicode-display_width (>= 1.4.0, < 3.0)
     rubocop-ast (1.21.0)
       parser (>= 3.1.1.0)
-    rubocop-performance (1.14.3)
+    rubocop-performance (1.15.0)
       rubocop (>= 1.7.0, < 2.0)
       rubocop-ast (>= 0.4.0)
     ruby-progressbar (1.11.0)
@@ -77,10 +77,10 @@ DEPENDENCIES
   dotenv (~> 2.8)
   infer_model!
   pry (~> 0.14)
-  rake
+  rake (~> 13.0)
   rspec (~> 3.11)
   rubocop (~> 1.36)
   rubocop-performance (~> 1.14)
 
 BUNDLED WITH
-   2.3.17
+   2.3.11

data/infer_model.gemspec

@@ -38,6 +38,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "zeitwerk", "~> 2.6"
   spec.add_development_dependency "dotenv", "~> 2.8"
   spec.add_development_dependency "pry", "~> 0.14"
+  spec.add_development_dependency "rake", "~> 13.0"
   spec.add_development_dependency "rspec", "~> 3.11"
   spec.add_development_dependency "rubocop", "~> 1.36"
   spec.add_development_dependency "rubocop-performance", "~> 1.14"

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.0"
+  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.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Jay Schneider
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-09-10 00:00:00.000000000 Z
+date: 2022-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -80,6 +80,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.14'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement