christiank-turntable 0.6.3 → 0.6.4

data/bin/sqlite3-to-turntable

@@ -0,0 +1,40 @@
+#!/usr/bin/ruby
+#
+# sqlite3-to-turntable
+# Christian Koch <ckoch002@student.ucr.edu>
+#
+
+require 'rubygems'
+require 'sqlite3'
+require 'turntable'
+
+# There must be 2 arguments
+if ARGV.length != 2
+  puts 'Usage: sqlite3-to-turntable [sqlite3 database] [column name]'
+  exit 1
+end
+
+# Find out about the SQLite3 database
+db_filename    = ARGV[0].split('.').first
+sqlite_db      = SQLite3::Database.new ARGV[0]
+
+sqlite_db_columns = sqlite_db.query("select * from #{ARGV[1]}") { |result| result.columns }.dup
+sqlite_db_rows    = sqlite_db.execute "select * from #{ARGV[1]}"
+
+p sqlite_db_columns
+
+# Turntable ids and SQL ids are different...
+if sqlite_db_columns.index 'id'
+  sqlite_db_columns[sqlite_db_columns.index('id')] = 'old_id'
+end
+
+# Convert the remaining column names to symbols
+sqlite_db_columns.collect! { |column| column.to_sym }
+
+# Create the new Turntable
+turntable_db = Turntable.new "#{db_filename}.turn", sqlite_db_columns
+
+# Fill up the Turntable
+sqlite_db_rows.each { |row| turntable_db.insert_array row }
+
+puts "Created #{turntable_db.filename}"

data/turntable.rb

@@ -1,5 +1,5 @@
 #
-# turntable v0.6.3
+# turntable v0.6.4
 # Christian Koch <ckoch002@student.ucr.edu>
 #
 
@@ -8,16 +8,33 @@ require 'ostruct'
 
 class Turntable < Array
 
-  @@version = 'v0.6.3'
+  @@version = 'v0.6.4'
 
-  # Creates a new Turntable database.
+  @@insert_proc = %q(
+    if args.length != @columns.length
+      raise ArgumentError, "wrong number of arguments (#{args.length} for #{@columns.length})"
+    end
+    row = Row.new
+    self.empty? ? (row.id = 0) : (row.id = self.last.id + 1)
+    @columns.each { |column| eval "row.#{column} = args.shift" }
+    self.push row
+    self.save_to @filename
+    row)
+
+  # Creates a new Turntable database. Accepts either a list of column names, or
+  # an array of symbols which contain the same thing.
   def initialize filename, *columns
     if columns.empty?
       raise ArgumentError, 'Turntable.new() requires a filename and at least one column name'
     end
 
+    if columns.length == 1 and columns[0].is_a?(Array) and columns[0][0].is_a?(Symbol)
+      @columns = columns[0]
+    else
+      @columns = columns
+    end
+
     @filename = File.expand_path(filename)
-    @columns  = columns
 
     self.save_to @filename
   end
@@ -61,20 +78,17 @@ class Turntable < Array
   end
 
   # Pushes self with a new row. Returns the newly created row. Requires the
-  # number of arguments passed in equal the number of columns in the database.
+  # number of arguments passed in equal the number of columns in the database,
+  # or it can accept calls from Turntable#insert_array.
   def insert *args
-    if args.length != @columns.length
-      raise ArgumentError, "wrong number of arguments (#{args.length} for #{@columns.length})"
-    end
-
-    row = Row.new
-    self.empty? ? (row.id = 0) : (row.id = self.last.id + 1)
-
-    @columns.each { |column| eval "row.#{column} = args.shift" }
-
-    self.push row
-    self.save_to @filename
-    row
+    eval @@insert_proc
+  end
+  
+  # Same as Turntable#insert, except this method accepts only one array filled
+  # with values.
+  def insert_array args
+    raise ArgumentError, 'Turntable#insert_array requires an array' unless args.is_a?(Array)
+    eval @@insert_proc
   end
 
   # Custom inspect() makes working in IRB much cleaner.
@@ -104,7 +118,7 @@ class Turntable < Array
       f.close
     end
 
-    # Users can't reassign @filename themselves.
+    # Users shouldn't reassign @filename themselves.
     attr_writer :filename
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: christiank-turntable
 version: !ruby/object:Gem::Version 
-  version: 0.6.3
+  version: 0.6.4
 platform: ruby
 authors: 
 - Christian Koch
@@ -9,14 +9,14 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-06-17 00:00:00 -07:00
+date: 2009-06-18 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
 description: 
 email: ckoch002@student.ucr.edu
-executables: []
-
+executables: 
+- sqlite3-to-turntable
 extensions: []
 
 extra_rdoc_files: []