pg_copy 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 82d865ae378b95ab956b69ac6231829b6c0c0240
-  data.tar.gz: ef1c2b9b2d0742eb5ef227e36061d8880533d1cd
+  metadata.gz: 231532457b1b447cf36b4a3011c85fb583d544c7
+  data.tar.gz: ef3badfdaafcf3aff0d7ad845193a7d31b7a0f99
 SHA512:
-  metadata.gz: bc015125da0d789f149ed15daf1b24f2ad4f04a21f0dc657b9d5af7ee28c6773ab323acbdb10641ac3705978b987d14f6c1dd9424a4299ac08b02a4b13b21b5d
-  data.tar.gz: 385234498337426f1da91ff29da79e4c0cce219c2320b5e9c92353f97bbf5a32e2f87bd39614fb68f76f795cf88829546e7f1b13b8cce2ef044a28a353923259
+  metadata.gz: 2d822841d50c580d7c83688445c95a6d3149653d350f1062c17aea0b6f8d114d1dcd6f1a60969fcdf403a6c39021de4ccb3062273aa10cc94972365f88fea304
+  data.tar.gz: 4767d61720cc1a29a29460535a1860c036b61ceeae3d7adc175bda509aaeeefa0131a815be08e5698454581e98a913723ed68ca2501495bb42c5189f2fda7451

data/lib/pg_copy.rb

@@ -1,7 +1,18 @@
-require "pg"
-require "pg_copy/version"
-require "pg_copy/copy_to_csv"
+require 'pg'
+require 'pg_copy/version'
+require 'pg_copy/copy_to_csv'
+require 'pg_copy/copy_from_csv'
 
 module PgCopy
-  # Your code goes here...
+  class << self
+    attr_accessor :default_connection
+
+    def copy_to_csv(sql, dest, options = {})
+      PgCopy::CopyToCSV.new(default_connection, sql, dest, options).copy
+    end
+
+    def copy_from_csv(source, table, columns, options = {})
+      PgCopy::CopyFromCSV.new(default_connection, source, table, columns, options).copy
+    end
+  end
 end

data/lib/pg_copy/copy_from_csv.rb

@@ -0,0 +1,39 @@
+require 'csv'
+
+module PgCopy
+  class CopyFromCSV
+    SQL = 'COPY %<table>s (%<columns>s) FROM STDIN WITH CSV'.freeze
+
+    DEFAULTS = {
+      header: false
+    }.freeze
+
+    def initialize(conn, file, table, columns, options = {})
+      @conn = conn
+      @file = file
+      @table = table
+      @columns = columns
+      @options = DEFAULTS.merge(options)
+    end
+
+    def copy
+      sql = format(SQL, table: @conn.escape_identifier(@table),
+                        columns: @columns.join(','))
+
+      csv = CSV.foreach(@file).lazy
+      csv = csv.drop(1) if skip_header?
+
+      @conn.copy_data(sql) do
+        csv.each do |row|
+          @conn.put_copy_data(row.to_csv)
+        end
+      end
+    end
+
+    private
+
+    def skip_header?
+      @options[:header]
+    end
+  end
+end

data/lib/pg_copy/copy_to_csv.rb

@@ -1,15 +1,29 @@
 module PgCopy
   class CopyToCSV
-    SQL = 'COPY (%<sql>s) TO STDOUT WITH CSV'.freeze
+    SQL = <<-SQL.freeze
+      COPY (%<sql>s) TO STDOUT (
+        FORMAT CSV,
+        DELIMITER '%<delimiter>c',
+        HEADER %<header>s,
+        QUOTE '%<quote>c'
+      )
+    SQL
 
-    def initialize(conn, sql, dest)
+    DEFAULTS = {
+      delimiter: ',',
+      header: false,
+      quote: '"'
+    }.freeze
+
+    def initialize(conn, sql, dest, options = {})
       @conn = conn
       @sql = sql
       @dest = dest
+      @options = DEFAULTS.merge(options)
     end
 
     def copy
-      sql = format(SQL, sql: @sql)
+      sql = format(SQL, sql: @sql, **@options)
 
       File.open(@dest, 'wb') do |f|
         @conn.copy_data(sql) do

data/lib/pg_copy/version.rb

@@ -1,3 +1,3 @@
 module PgCopy
-  VERSION = "0.0.1"
+  VERSION = '0.0.2'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_copy
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Lenon Marcel
@@ -11,9 +11,9 @@ authors:
 - Diogo Lisboa
 - Pablo Silva
 autorequire: 
-bindir: exe
+bindir: bin
 cert_chain: []
-date: 2016-02-16 00:00:00.000000000 Z
+date: 2016-02-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -71,6 +71,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A wrapper for PostgreSQL's COPY command
 email:
 - lenon.marcel@gmail.com
@@ -83,18 +97,11 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- ".rspec"
-- ".travis.yml"
-- Gemfile
 - README.md
-- Rakefile
-- bin/console
-- bin/setup
 - lib/pg_copy.rb
+- lib/pg_copy/copy_from_csv.rb
 - lib/pg_copy/copy_to_csv.rb
 - lib/pg_copy/version.rb
-- pg_copy.gemspec
 homepage: https://github.com/locaweb/pg_copy
 licenses: []
 metadata: {}

data/.gitignore

@@ -1,10 +0,0 @@
-/.bundle/
-/.yardoc
-/Gemfile.lock
-/_yardoc/
-/coverage/
-/doc/
-/pkg/
-/spec/reports/
-/tmp/
-/spec/database.yml

data/.rspec

@@ -1,2 +0,0 @@
---format documentation
---color

data/.travis.yml

@@ -1,4 +0,0 @@
-language: ruby
-rvm:
-  - 2.3.0
-before_install: gem install bundler -v 1.11.2

data/Gemfile

@@ -1,4 +0,0 @@
-source 'https://rubygems.org'
-
-# Specify your gem's dependencies in pg_copy.gemspec
-gemspec

data/Rakefile

@@ -1,6 +0,0 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-task :default => :spec

data/bin/console

@@ -1,14 +0,0 @@
-#!/usr/bin/env ruby
-
-require "bundler/setup"
-require "pg_copy"
-
-# You can add fixtures and/or initialization code here to make experimenting
-# with your gem easier. You can also use a different console, if you like.
-
-# (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
-
-require "irb"
-IRB.start

data/bin/setup

@@ -1,8 +0,0 @@
-#!/usr/bin/env bash
-set -euo pipefail
-IFS=$'\n\t'
-set -vx
-
-bundle install
-
-# Do any other automated setup that you need to do here

data/pg_copy.gemspec

@@ -1,37 +0,0 @@
-# coding: utf-8
-lib = File.expand_path('../lib', __FILE__)
-$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'pg_copy/version'
-
-Gem::Specification.new do |spec|
-  spec.name          = "pg_copy"
-  spec.version       = PgCopy::VERSION
-
-  spec.authors = ['Lenon Marcel',
-                  'Rafael Timbó',
-                  'Rodolfo Liviero',
-                  'Italo Moraes',
-                  'Diogo Lisboa',
-                  'Pablo Silva']
-  spec.email = ['lenon.marcel@gmail.com',
-                'rafaeltimbosoares@gmail.com',
-                'rodolfoliviero@gmail.com',
-                'itu.moraes@gmail.com',
-                'diogoslisboa@gmail.com',
-                'phsilbr@gmail.com']
-
-  spec.summary       = "A wrapper for PostgreSQL's COPY command"
-  spec.description   = "A wrapper for PostgreSQL's COPY command"
-  spec.homepage      = 'https://github.com/locaweb/pg_copy'
-
-  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  spec.bindir        = "exe"
-  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
-
-  spec.add_development_dependency "bundler", "~> 1.11"
-  spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "rspec", "~> 3.0"
-
-  spec.add_dependency "pg"
-end