postgres-copy 1.6.0 → 1.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ff5750f062c23bbbaf42c2c4b54e0d10a2a18c45eda5a66e00abd2fa0d3d9cdc
-  data.tar.gz: 10920c6056c38c5cc4a83ba0ace5164abf8a9eb6276e5cef7a4c19e39424de11
+  metadata.gz: 0c9c2751b59517a10e556c5e4cf9cc6c51cafdf1bded29fc3ea493aee37df8f4
+  data.tar.gz: fb386d86afd455b9ebd6b4bbc93f6e2d316b5285fb5e1174ea335284e187c2b5
 SHA512:
-  metadata.gz: 25c750eb0da06933151e7707f67fa52b9973e603d2afe3288241acf34dc61f7004179483339e630e245660f4371c9f1ddb086e9e8f3a80b32c8bc5d180a79937
-  data.tar.gz: 8e537afc3c20801bcac073f208225d8375cc25b14b40ebbe80f904eef79ffded4c00ed672987094a08691d84dbaf1818d71fbf0ec2aeedba8d9d2a611298b68d
+  metadata.gz: 4b72e200f779749dbeaf7930835296709ac7c7f8c3b81bab737428c4d221b79032b1fd3996eafe8d2e9280f64f6b811cba1ab456969d6a35b5744cc8d7b6f477
+  data.tar.gz: e916531647bc072804b7a72987ac416674d9a81b6fd48d4c93c57c046e13ab99e172be607a66d09ceea0590a6573c7f9591a0d1b8e795a3eec9858c07f7f14cf

data/Gemfile.lock

@@ -1,29 +1,29 @@
 PATH
   remote: .
   specs:
-    postgres-copy (1.6.0)
+    postgres-copy (1.7.0)
       activerecord (>= 5.1)
       pg (>= 0.17)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    activemodel (7.0.1)
-      activesupport (= 7.0.1)
-    activerecord (7.0.1)
-      activemodel (= 7.0.1)
-      activesupport (= 7.0.1)
-    activesupport (7.0.1)
+    activemodel (7.0.4)
+      activesupport (= 7.0.4)
+    activerecord (7.0.4)
+      activemodel (= 7.0.4)
+      activesupport (= 7.0.4)
+    activesupport (7.0.4)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
       tzinfo (~> 2.0)
-    concurrent-ruby (1.1.9)
+    concurrent-ruby (1.1.10)
     diff-lcs (1.4.4)
-    i18n (1.9.1)
+    i18n (1.12.0)
       concurrent-ruby (~> 1.0)
-    minitest (5.15.0)
-    pg (1.2.3)
+    minitest (5.16.3)
+    pg (1.4.5)
     rake (11.2.2)
     rdoc (6.2.1)
     rspec (2.99.0)
@@ -34,7 +34,7 @@ GEM
     rspec-expectations (2.99.2)
       diff-lcs (>= 1.1.3, < 2.0)
     rspec-mocks (2.99.4)
-    tzinfo (2.0.4)
+    tzinfo (2.0.5)
       concurrent-ruby (~> 1.0)
 
 PLATFORMS

data/README.md

@@ -1,4 +1,4 @@
-# postgres-copy 
+# postgres-copy
 
 ![Ruby](https://github.com/diogob/postgres-copy/workflows/Ruby/badge.svg)
 
@@ -207,6 +207,19 @@ This is useful for removing byte order marks when matching column headers.
 User.copy_from "/tmp/users_with_byte_order_mark.csv", :encoding => 'bom|utf-8'
 ```
 
+### Using PostgresCopy::WithTempTable.generate
+
+Based on [nfedyashev](https://github.com/nfedyashev)'s [comment](https://github.com/diogob/postgres-copy/issues/51):
+
+```ruby
+    PostgresCopy::WithTempTable.generate do |t|
+      columns.each do |column_name|
+        t.string column_name.to_sym
+      end
+    end
+```
+
+This auto-generates an id column, but the temp table creation is configurable.
 
 ### Using the CSV Responder
 If you want to make the result of a COPY command available to download this gem provides a CSV responder that, in conjunction with [inherited_resources](https://github.com/josevalim/inherited_resources), is a very powerfull tool. BTW, do not try to use the responder without inherited_resources.

data/lib/postgres-copy/acts_as_copy_target.rb

@@ -108,7 +108,7 @@ module PostgresCopy
                   quoted_table_name
                 end
 
-        columns_list = columns_list.map{|c| options[:map][c.to_s] } if options[:map]
+        columns_list = columns_list.map{|c| options[:map][c.to_s] || c.to_s } if options[:map]
         columns_string = columns_list.size > 0 ? "(\"#{columns_list.join('","')}\")" : ""
         connection.raw_connection.copy_data %{COPY #{table} #{columns_string} FROM STDIN #{options_string}} do
           if options[:format] == :binary

data/lib/postgres-copy/with_temp_table.rb

@@ -0,0 +1,18 @@
+module PostgresCopy
+  module WithTempTable
+    def self.generate(connection = ActiveRecord::Base.connection, base_klass:
+      ActiveRecord::Base, temp_table_name: nil, create_table_opts: {id: :bigint})
+      raise "You have to pass a table schema definition block!" unless block_given?
+      table_name = temp_table_name || "temp_table_#{SecureRandom.hex}"
+
+      connection.create_table table_name, temporary: true, **create_table_opts do |t|
+        yield t
+      end
+
+      klass = Class.new(base_klass) do
+        acts_as_copy_target
+        self.table_name = table_name
+      end
+    end
+  end
+end

data/postgres-copy.gemspec

@@ -5,7 +5,7 @@ $:.unshift lib unless $:.include?(lib)
 
 Gem::Specification.new do |s|
   s.name                  = "postgres-copy"
-  s.version               = "1.6.0"
+  s.version               = "1.7.0"
   s.platform              = Gem::Platform::RUBY
   s.required_ruby_version = ">= 1.9.3"
   s.authors               = ["Diogo Biazus"]

data/spec/copy_from_spec.rb

@@ -14,6 +14,12 @@ describe "COPY FROM" do
     TestModel.order(:id).map{|r| r.attributes}.should == [{'id' => 1, 'data' => 'test data 1'}]
   end
 
+  it "should import from file if path is with a field_map" do
+    TestModel.copy_from File.expand_path('spec/fixtures/comma_with_header_to_map.csv'), map: {'ref' => 'id'}
+    TestModel.order(:id).map{|r| r.attributes}.should == [{'id' => 1, 'data' => 'test data 1'}]
+  end
+
+
   it "should import from IO without field_map" do
     TestModel.copy_from File.open(File.expand_path('spec/fixtures/comma_with_header.csv'), 'r')
     TestModel.order(:id).map{|r| r.attributes}.should == [{'id' => 1, 'data' => 'test data 1'}]

data/spec/fixtures/comma_with_header_to_map.csv

@@ -0,0 +1,2 @@
+ref,data
+1,test data 1

data/spec/postgres-copy/with_temp_table_spec.rb

@@ -0,0 +1,17 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+require 'postgres-copy/with_temp_table'
+
+describe '.generate' do
+  subject(:generate) {
+    PostgresCopy::WithTempTable.generate do |t|
+      t.string :data
+    end
+  }
+
+  it {
+    generate.copy_from 'spec/fixtures/comma_with_header.csv'
+    data = generate.all.first
+    expect(data.id).to eq(1)
+    expect(data.data).to eq('test data 1')
+  }
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: postgres-copy
 version: !ruby/object:Gem::Version
-  version: 1.6.0
+  version: 1.7.0
 platform: ruby
 authors:
 - Diogo Biazus
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-02-07 00:00:00.000000000 Z
+date: 2022-11-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -113,6 +113,7 @@ files:
 - lib/postgres-copy.rb
 - lib/postgres-copy/acts_as_copy_target.rb
 - lib/postgres-copy/csv_responder.rb
+- lib/postgres-copy/with_temp_table.rb
 - postgres-copy.gemspec
 - spec/copy_from_binary_spec.rb
 - spec/copy_from_spec.rb
@@ -127,6 +128,7 @@ files:
 - spec/fixtures/comma_with_header_and_scope.csv
 - spec/fixtures/comma_with_header_empty_values_at_the_end.csv
 - spec/fixtures/comma_with_header_multi.csv
+- spec/fixtures/comma_with_header_to_map.csv
 - spec/fixtures/comma_without_header.csv
 - spec/fixtures/extra_field.rb
 - spec/fixtures/reserved_word_model.rb
@@ -146,12 +148,13 @@ files:
 - spec/fixtures/tab_with_two_lines.tsv
 - spec/fixtures/test_extended_model.rb
 - spec/fixtures/test_model.rb
+- spec/postgres-copy/with_temp_table_spec.rb
 - spec/spec.opts
 - spec/spec_helper.rb
 homepage: http://github.com/diogob/postgres-copy
 licenses: []
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -166,8 +169,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
-signing_key:
+rubygems_version: 3.1.6
+signing_key: 
 specification_version: 4
 summary: Put COPY command functionality in ActiveRecord's model class
 test_files: []