active_tsv 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a1e82adf6d31ab6578a3e5b5f290682135ab9a3b
-  data.tar.gz: 877883f7b97bb19854e17b801f2b71c054caf50b
+  metadata.gz: bd70390aab63f668f621f2305a840d60169f6e1e
+  data.tar.gz: dd47ec7b43b36397d701db69aadcedc1166a64a8
 SHA512:
-  metadata.gz: 2a7e77ed3e9fc8f04924549bed9327913dc66c381ef863437d107745246fbefe1dae9c6a80a2c86b73866e37a2022a99684bb6a256baa8f2bdf517e695fce613
-  data.tar.gz: 03a7633a7bec5da34353e40a7fba5719c12bada29217a1b5686ed9de47a33e594a38dff57904ef6ad3d784db1ef011d32dd7b61404fd09fdc9dfe4cae5858546
+  metadata.gz: c200088cb1725b2d2357642b99c4aa2102c0e591c8b2c88a4b9ba7c42b2077f907f059c0c3f9ba238b453f882772d3fdd69a70e6cca61119dff17bcede9ab188
+  data.tar.gz: 2ad08364ab1dd69f2937e6f8b4a54f2a55275da21f900a8e1fffa7ac79ac62d05cb57d75b2b9e24bc8cdc719f06e668b975fee4228e346dcb9a805c8cc0c94ac

data/.travis.yml

@@ -2,5 +2,6 @@ language: ruby
 rvm:
   - 2.2.6
   - 2.3.3
+  - 2.4.0
 notifications:
   email: false

data/data/nokeys.tsv

@@ -0,0 +1,3 @@
+1	2	3
+2	3	4
+3	4	5

data/lib/active_tsv/base.rb

@@ -28,6 +28,7 @@ module ActiveTsv
         end
 
         @column_names = nil
+        @custom_column_name = false
         @table_path = path
         column_names.each do |k|
           define_method(k) { @attrs[k] }
@@ -48,7 +49,24 @@ module ActiveTsv
       end
 
       def column_names
-        @column_names ||= open { |csv| csv.gets }
+        @column_names ||= begin
+          @custom_column_name = false
+          open { |csv| csv.gets }
+        end
+      end
+
+      def column_names=(names)
+        @custom_column_name = true
+        column_names = names.map(&:to_s)
+        column_names.each do |k|
+          define_method(k) { @attrs[k] }
+          define_method("#{k}=") { |v| @attrs[k] = v }
+        end
+        @column_names = column_names
+      end
+
+      def custom_column_name?
+        @custom_column_name
       end
 
       def primary_key
@@ -76,7 +94,17 @@ module ActiveTsv
     def initialize(attrs = {})
       case attrs
       when Hash
-        @attrs = attrs
+        h = {}
+        self.class.column_names.each do |name|
+          h[name] = nil
+        end
+        attrs.each do |name, v|
+          unless respond_to?("#{name}=")
+            raise UnknownAttributeError, "unknown attribute '#{name}' for #{self.class}."
+          end
+          h[name.to_s] = v
+        end
+        @attrs = h
       when Array
         @attrs = self.class.column_names.zip(attrs).to_h
       else

data/lib/active_tsv/errors.rb

@@ -2,4 +2,5 @@ module ActiveTsv
   ActiveTsvError = Class.new(StandardError)
   RecordNotFound = Class.new(ActiveTsvError)
   StatementInvalid = Class.new(ActiveTsvError)
+  UnknownAttributeError = Class.new(NoMethodError)
 end

data/lib/active_tsv/relation.rb

@@ -198,7 +198,7 @@ module ActiveTsv
 
       key_to_value_index = @model.column_names.each_with_index.to_h
       @model.open do |csv|
-        csv.gets
+        csv.gets if !@model.custom_column_name?
         csv.each do |value|
           yield value if @where_values.all? do |cond|
             case cond

data/lib/active_tsv/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveTsv
-  VERSION = "0.3.1"
+  VERSION = "0.3.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_tsv
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - ksss
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-01-14 00:00:00.000000000 Z
+date: 2017-01-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -87,6 +87,7 @@ files:
 - data/books.tsv
 - data/names.tsv
 - data/nicknames.tsv
+- data/nokeys.tsv
 - data/passwords.tsv
 - data/user_books.tsv
 - data/users.csv