csv2psql 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fc32b319264dee9105efb40ad0091403d8ecd6e3
-  data.tar.gz: 6e20b3c108466bf8c171d417156cd25e17afe267
+  metadata.gz: b8617b9028668a6a28cd5185dc0eb6a79effda90
+  data.tar.gz: b7808d63c2ec216005dc407aa8b200638b5344e5
 SHA512:
-  metadata.gz: 8784cbe3de9285e9d78b81b8e69d628488839cafe6557cc8a0602d9c321f49b4fbd36037d01652e2fcceb5ac687bfc428005c34c76232850c61269a60728237f
-  data.tar.gz: 6657977d8372f140ee242be56889840ef97969d8a043244be4d721b8de3da2ac5d04e357fb913f20377d4eab121027c0af9cfe64984142eedad9951ace6880a0
+  metadata.gz: 7ad111c1e05e5b12d191d9e160d248280f76d3768933b03ac41c4ee42c1f0e2962ed5251bade181c52b217fa14da66e1a58acaa6303d1bcb0e778ddb95bebf04
+  data.tar.gz: 9b75d29f5a8bb8c412b719df00aa88056053f0b209a376166b05f8cdf616e1f400d1f2d872d94fa4ba5afbd38f5339657bf35d792d626cc188486596af988537

data/README.md

@@ -56,13 +56,15 @@ SYNOPSIS
     csv2psql [global options] convert [command options]
 
 COMMAND OPTIONS
-    --[no-]create-table - Crate SQL Table before inserts
-    -d, --delimiter=arg - Column delimiter (default: ,)
-    -h, --[no-]header   - Header row included (default: enabled)
-    -q, --quote=arg     - Quoting character (default: ")
-    -s, --separator=arg - Line separator (default: auto)
-    -t, --table=arg     - Table to insert to (default: my_table)
-    --[no-]transaction  - Import in transaction block (default: enabled)
+    --[no-]create-table   - Crate SQL Table before inserts
+    -d, --delimiter=arg   - Column delimiter (default: ,)
+    --[no-]drop-table     - Drop SQL Table before inserts
+    -h, --[no-]header     - Header row included (default: enabled)
+    -q, --quote=arg       - Quoting character (default: ")
+    -s, --separator=arg   - Line separator (default: auto)
+    -t, --table=arg       - Table to insert to (default: my_table)
+    --[no-]transaction    - Import in transaction block (default: enabled)
+    --[no-]truncate-table - Truncate SQL Table before inserts
 ```
 
 ## Example
@@ -126,6 +128,33 @@ INSERT 0 1
 COMMIT
 ```
 
+**Convert CSV - Full load**
+
+```
+csv2psql convert --create-table --drop-table --truncate-table -t test data/sample.csv
+
+BEGIN;
+DROP TABLE IF EXISTS test;
+
+CREATE TABLE test(
+	id TEXT,
+	firstname TEXT,
+	lastname TEXT,
+	address_street TEXT,
+	address_city TEXT,
+	address_details_note TEXT
+)
+WITH (
+  OIDS=FALSE
+);
+
+TRUNCATE test;
+
+INSERT INTO test(id, firstname, lastname, address_street, address_city, address_details_note) VALUES('12345', 'Joe', 'Doe', '#2140 Taylor Street, 94133', 'San Francisco', 'Pool available');
+INSERT INTO test(id, firstname, lastname, address_street, address_city, address_details_note) VALUES('45678', 'Jack', 'Plumber', '#111 Sutter St, 94104', 'San Francisco', 'Korean Deli near to main entrance');
+COMMIT;
+```
+
 ## Contributing to csv2psql
 
 - Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet

data/lib/csv2psql/processor/processor.rb

@@ -34,6 +34,7 @@ module Csv2Psql
     TEMPLATE_DIR = File.join(BASE_DIR, 'templates')
     CREATE_TABLE_TEMPLATE = File.join(TEMPLATE_DIR, 'create_table.sql.erb')
     DROP_TABLE_TEMPLATE = File.join(TEMPLATE_DIR, 'drop_table.sql.erb')
+    HEADER_TEMPLATE = File.join(TEMPLATE_DIR, 'header.sql.erb')
     TRUNCATE_TABLE_TEMPLATE = File.join(TEMPLATE_DIR, 'truncate_table.sql.erb')
 
     def convert(paths, opts = {})
@@ -54,8 +55,22 @@ module Csv2Psql
     end
 
     def create_header(path, row, opts = {})
+      ctx = create_erb_context(path, row, opts)
+      erb = ErbHelper.new
+      erb.process(HEADER_TEMPLATE, ctx)
+    end
+
+    def create_table(path, row, opts = {})
+      ctx = create_erb_context(path, row, opts)
+      erb = ErbHelper.new
+      erb.process(CREATE_TABLE_TEMPLATE, ctx)
+    end
+
+    def create_sql_script(path, row, opts = {})
       return unless @first_row
 
+      puts create_header(path, row, opts)
+
       TABLE_FUNCTIONS.each do |k, v|
         t = DEFAULT_OPTIONS[k]
         t = opts[k] unless opts[k].nil?
@@ -65,12 +80,6 @@ module Csv2Psql
       @first_row = false
     end
 
-    def create_table(path, row, opts = {})
-      ctx = create_erb_context(path, row, opts)
-      erb = ErbHelper.new
-      erb.process(CREATE_TABLE_TEMPLATE, ctx)
-    end
-
     def drop_table(path, row, opts = {})
       ctx = create_erb_context(path, row, opts)
       erb = ErbHelper.new
@@ -148,7 +157,7 @@ module Csv2Psql
     end
 
     def with_row(path, row, opts = {}, &block)
-      create_header(path, row, opts)
+      create_sql_script(path, row, opts)
 
       args = { path: path, row: row }
       block.call(args) if block_given?

data/lib/csv2psql/version.rb

@@ -2,5 +2,5 @@
 
 # Csv2Psql module
 module Csv2Psql
-  VERSION = '0.0.5'
+  VERSION = '0.0.6'
 end

data/templates/header.sql.erb

@@ -0,0 +1,2 @@
+-- Table <%= ctx[:table] + "\n" %>
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: csv2psql
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Tomas Korcak
@@ -208,6 +208,7 @@ files:
 - spec/spec_helper.rb
 - templates/create_table.sql.erb
 - templates/drop_table.sql.erb
+- templates/header.sql.erb
 - templates/truncate_table.sql.erb
 homepage: https://github.com/korczis/csv2psql
 licenses: