csv-query 0.1.11 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e81f7c4a5af21d396b610651b90d8815bf57b93721db4b83ceb50accc2dba055
-  data.tar.gz: 7addf7ffdebdcf96b566812a1e273cab128ded81ec7095b7f93fb26b03b7972d
+  metadata.gz: 6e2eb91c43cde7f52f32a6cb81ff9bfbe1db93d78a5e40ee0faecedec42c4e0d
+  data.tar.gz: a5d750c90a8a2ca068a0ee0767aa16855532a8bbb81ff2d956e1c0d0f8602656
 SHA512:
-  metadata.gz: 9304d5cc064e2d53f1e2756278cbf9a753be72da559ab900b5baac7e76b526fca1b0dc5292bddc413dcdf8fe9271e262347a8aa506d815eed22f853519b657e2
-  data.tar.gz: 26506e5378dcd310bd6ab3de5a927f62859e5ea6fa83dc599ced5596856a85216a6a2d07d8a75b4370f3897ad29ebab2745641e944ae525b3468bc1439421b64
+  metadata.gz: 6e28babdaf65665beaa4d73fec25c86403811a3fa5a0d310033cfe775ad92073132347c4db40d4234daba0d51d5706d00ceb81091c35790c090cdabedd52b4a2
+  data.tar.gz: 9fda521cd933cd080e4fdb50672c5132f7014225c37da0eb78bf4722082722799642aff4d1fc09020d37a8876272d7d043237faccf741aeddd1de7d0d87e396e

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    csv-query (0.1.11)
+    csv-query (0.1.12)
       activerecord
       charlock_holmes
       sqlite3

data/lib/csv/query/version.rb

@@ -1,5 +1,5 @@
 module Csv
   module Query
-    VERSION = "0.1.11"
+    VERSION = "0.1.12"
   end
 end

data/lib/csv/query.rb

@@ -3,7 +3,6 @@ require 'csv'
 require 'sqlite3'
 require 'active_record'
 require 'optparse'
-require 'pp'
 require 'charlock_holmes/string'
 require 'terminal-table'
 
@@ -11,15 +10,20 @@ module Csv
   module Query
     # Your code goes here...
     def self.run!
-      file_path, json =
-        ARGV.getopts(nil, 'file:', 'json').values
+      file_path, json, sync =
+        ARGV.getopts(nil, 'file:', 'json', 'sync').values
 
-      InMemoryAR.new(file_path, json).run!
+      InMemoryAR.new(file_path, json, sync).run!
     end
 
     class InMemoryAR
       # temporary AR obj
       class Record < ActiveRecord::Base
+        def to_h
+          h = self.attributes
+          h.delete("id")
+          h
+        end
       end
 
       # `LP(iphone)`みたいに
@@ -31,10 +35,14 @@ module Csv
         end
       end
 
-      def csv
+      def encoding
         contents = File.read(file_path)
         detection = contents.detect_encoding
-        case detection[:encoding]
+        detection[:encoding]
+      end
+
+      def csv
+        case encoding
         when "Shift_JIS"
           CSV.read(file_path, encoding: "SJIS:UTF-8", headers: true, header_converters: header_converter)
         when "UTF-8"
@@ -48,7 +56,8 @@ module Csv
 
       attr_reader :file_path
 
-      def initialize(file_path, json)
+      def initialize(file_path, json, sync)
+        @sync = sync
         @json = json
         @file_path = file_path
 
@@ -73,18 +82,35 @@ module Csv
         @json
       end
 
+      def sync?
+        @sync
+      end
+
       def run!
         csv.each do |row|
           Record.create!(row.to_h)
         end
 
-        records =
-          InMemoryAR::Record.all
+        records = InMemoryAR::Record.all
+
+        export_as_csv(records) if sync?
+
+        render(records)
+      end
+
+      def export_as_csv records
+        CSV.open(file_path, "wb", encoding: encoding, headers: csv_headers, write_headers: true, force_quotes: true) do |csv|
+          records.each do |record|
+            csv << record.to_h.values
+          end
+        end
+      end
 
+      def render records
         if json_format?
-          puts records.map { |e| e.attributes }.to_json
+          puts records.map { |e| e.to_h }.to_json
         else
-          rows = records.map { |e| e.attributes.values }
+          rows = records.map { |e| e.to_h.values }
           puts Terminal::Table.new :headings => csv_headers, :rows => rows
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: csv-query
 version: !ruby/object:Gem::Version
-  version: 0.1.11
+  version: 0.1.12
 platform: ruby
 authors:
 - naofumi-fujii