christiank-turntable 0.4.2 → 0.4.3

data/turntable.rb

@@ -1,19 +1,19 @@
 #
-# turntable v0.4.2
+# turntable v0.4.3
 # Christian Koch < ckoch002@student.ucr.edu >
 #
 # This is free software. You are permitted to modify, implement, and otherwise
 # use this source code freely (as in beer and freedom).
 #
-
+ 
 class Turntable
-
-  @@version = '0.4.2'
+ 
+  @@version = '0.4.3'
   
   # Creates a new Turntable object and database file.
   # Requires a filename and at least one column name.
   def initialize filename, *args
-    raise "Turntable.new requires at least two arguments" if args.empty?
+    raise 'Turntable.new requires at least two arguments' if args.empty?
     
     @columns = []
     @table = []
@@ -44,6 +44,13 @@ class Turntable
   # Reader methods for the object's columns and filename.
   attr_reader :columns
   attr_reader :filename
+  
+  # Pushes a new column. Fills the new fields with the empty string
+  # (because filling them with nil creates issues). Returns the the new table.
+  def add_column column_name
+    @columns.push column_name
+    @table.each { |row| row[column_name] = '' }
+  end
     
   # Returns true or false, depending on whether the result is empty.
   def contains? hash
@@ -55,7 +62,7 @@ class Turntable
     raise 'delete() requires :id => some_value' unless hash.has_key?(:id)
     
     before_table = @table.dup
-
+ 
     # Delete the table for real here:
     what_wasnt_deleted = @table.delete_if { |row| row[:id] == hash[:id] }
     self.save_to @filename
@@ -64,6 +71,12 @@ class Turntable
     return before_table - what_wasnt_deleted
   end
   
+  # Eliminates an entire column.
+  def drop_column column_name
+    @columns.delete_if { |column| column == column_name}
+    @table.each { |row| row.delete column_name }
+  end
+  
   # Inserts a new row in the database. It requires that the number of arguments
   # passed in equal the number of columns, just like any SQL INSERT.
   # Columns named :id function like AUTOINCREMENT fields.
@@ -101,13 +114,17 @@ class Turntable
   # Print a "plain text" table to STDOUT
   # TODO: create a table.each.columns.each protected method, or something
   def puts_formatted
+    warn 'Warning: puts_formatted() is deprecated, use to_stdout()'
+    to_stdout
+  end
+  
+  def to_stdout
     @table.each do |row|
       @columns.each do |column|
         print "|#{row[column]}"
       end
       puts "|"
     end
-    
     return nil
   end
   
@@ -129,7 +146,7 @@ class Turntable
   
   # Returns the entire table, an array of hashes.
   def select_all
-    return @table
+    @table
   end
   
   alias :all :select_all
@@ -150,7 +167,7 @@ class Turntable
   end
   
   protected
-    # Marshals the database object to an external file. 
+    # Marshals the database object to an external file.
     def save_to filename
       if File.exists? filename
         f = File.open filename, 'w'
@@ -162,9 +179,9 @@ class Turntable
       f.close
     end
 end
-
+ 
 class Array
-
+ 
   # Rearranges an array of hashes according to a specified column name.
   # TODO: how to check for @columns, even though it's a Turntable variable?
   def order_by column_name
@@ -172,7 +189,7 @@ class Array
     # Convert all row-hashes into arrays
     rearranged_table = self.collect { |row| row.to_a }
     
-    # Convert all column-name-symbols into strings, 
+    # Convert all column-name-symbols into strings,
     # then switch the first element of each row the desired order_by column
     # (because the first element is where sort() looks)
     rearranged_table.each do |row|
@@ -189,7 +206,7 @@ class Array
         column[0] = column[0].to_sym
         this_column_hash[ column[0] ] = column[1]
       end
-
+ 
       this_row_hash.update this_column_hash
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: christiank-turntable
 version: !ruby/object:Gem::Version 
-  version: 0.4.2
+  version: 0.4.3
 platform: ruby
 authors: 
 - Christian Koch
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-05-22 00:00:00 -07:00
+date: 2009-05-24 00:00:00 -07:00
 default_executable: 
 dependencies: []