recordx_sqlite 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9323b4c82ce89a36aabab387780e377c0db10ba7
-  data.tar.gz: b7fd2bf0bd1ee50fd4b3ce9da277d3eb084efebc
+  metadata.gz: e069e8ccb9d0cbe316d0ed625a7ae24a5108b2a9
+  data.tar.gz: 3e39c34f538d22ed9fd39ef0848d723027152fd5
 SHA512:
-  metadata.gz: 77b3f0b2d2c57bf6c7bea34c5b0f004b5c7b03d313d9416f795ac08fcbf228d4547157c700b7e94a6be51e4e4f5d7ff1b0a189ad386221e7a53bb0c47f3e013b
-  data.tar.gz: faa156cb198c9fe6d90b8aea54a15c6c720d2463c0aa650e5ead4b49425fe524941b166694ba2ad27b7bb7aede8ea6fcf29d43113f7758664553044d365058f9
+  metadata.gz: 2a5de2afcc95af499e92e65b2d922700bde7d2960a6e9fe5c1ba83f78d6535213475f7e42a42e81968a1a6c8aa5ca44787eb0b3e4cf91580bd06f0d48d7c7a3d
+  data.tar.gz: be52987770f9bfd896718be81f1dc065750088e71da426acd7f22d683c0dd08bbc22d2870b81f452f0b66e1e826027a0821eb702c6c03d058802ccce6a440480

data/lib/recordx_sqlite.rb

@@ -8,36 +8,43 @@ require 'recordx'
 
 class RecordxSqlite
 
-  def initialize(dbfile, table)
-
+  def initialize(dbfile, table: '', primary_key: :id, pk: primary_key)
+    
+    
     @db = SQLite3::Database.new dbfile
+
     @db.results_as_hash = true
     rs = @db.query 'select * from ' + table
 
+    @table, @primary_key = table, pk.to_sym
+    
     @a = rs.map do |h| 
       h2 = h.inject({}) {|r, x| k, v = x; r.merge(k.to_sym => v)}
-      RecordX.new(h2, self, h2[:uid]) 
+      RecordX.new(h2, self, h2[@primary_key]) 
     end
     
-    @table = table
-
   end
 
   def all()
     @a
   end
+  
+  def find(id)
+    @a.find {|x| x.method(@primary_key).call == id}
+  end
 
   def update(id, h={})
 
     col, value = h.to_a.first
-    return if col == :id
+    return if col == @primary_key
 
 s = "
 UPDATE #{@table}
 SET #{col}='#{value}'
-WHERE id='#{id}';"
+WHERE #{@primary_key.to_s}='#{id}';"
 
     @db.execute(s)
 
   end
 end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: recordx_sqlite
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - James Robertson
@@ -31,7 +31,7 @@ cert_chain:
   pewreBQ0QCPuPbV1FPlyX5N5qsGIMHFIYqUlaykOA8156Y6fbIbqQsLwxaSr8WkD
   eojajR/fTTvliw==
   -----END CERTIFICATE-----
-date: 2016-07-16 00:00:00.000000000 Z
+date: 2016-07-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sqlite3