csv_store 0.1.0 → 0.1.1

data/lib/csv_store.rb

@@ -1,4 +1,5 @@
 require 'csv_store/version'
+require 'csv_store/constants'
 require 'csv_store/database'
 require 'csv_store/file'
 require 'fastercsv'

data/lib/csv_store/constants.rb

@@ -0,0 +1,6 @@
+module CSVStore
+  DEFAULT_OPTIONS = {
+    :col_sep => "\t",
+    :row_sep => "\n"
+  }
+end

data/lib/csv_store/database.rb

@@ -0,0 +1,55 @@
+module CSVStore
+    class Database
+
+      attr_reader :base_path
+
+      def initialize(base_path)
+        @base_path = base_path
+      end
+
+      def path_for_file(path)
+        ::File.join(base_path, path)
+      end
+
+      def path_with_existing_dir(path)
+        current_path = path_for_file(path)
+        FileUtils.mkdir_p(::File.dirname(current_path))
+        current_path
+      end
+
+      def glob(glob_string, &block)
+        current_glob = path_for_file(glob_string)
+        Dir.glob(current_glob) do |path|
+          yield File.new(path, "rb")
+        end
+      end
+
+      def each_row(path)
+        file_path = path_for_file(path)
+        ::CSVStore::File.new(file_path, "rb").each do |row|
+          yield row
+        end
+      end
+
+      def append(path, row=nil, &block)
+        file_path = path_with_existing_dir(path)
+        if block_given?
+          ::CSVStore::File.new(file_path, "ab").file(&block)
+        elsif row
+          ::CSVStore::File.new(file_path, "ab").file do |f|
+            f << row
+          end
+        end
+      end
+
+      def file(path, &block)
+        ::CSVStore::File.new(file_path, "rb").file(&block)
+      end
+
+      def truncate_file(path, &block)
+        file_path = path_with_existing_dir(path)
+        ::CSVStore::File.new(file_path, "wb").file(&block)
+      end
+
+    end
+  end

data/lib/csv_store/file.rb

@@ -0,0 +1,50 @@
+module CSVStore
+  class File
+    
+    attr_reader :path, :mode
+    
+    def initialize(path, mode)
+      @path = path
+      @mode = mode
+    end
+    
+    def file(&block)
+      if block_given?
+        FasterCSV.open(path, mode, DEFAULT_OPTIONS) do |f|
+          yield f
+        end
+      else
+        FasterCSV.open(path, mode, DEFAULT_OPTIONS)
+      end
+    end
+    
+    def each(&block)
+      file do |f|
+        f.each do |row|
+          yield row
+        end
+      end
+    end
+
+
+    def each_reverse_unique(&block)
+      each_unique(true, &block)
+    end
+
+    def each_unique(reverse = false, &block)
+      
+      read_identifiers = []
+      
+      file do |f|
+        (reverse ? f.read.reverse : f).each do |row|
+          row_identifier = row[0]
+          unless read_identifiers.include?(row_identifier)
+            yield row
+          end
+          read_identifiers << row_identifier
+        end
+      end
+    end
+
+  end
+end

data/lib/csv_store/version.rb

@@ -1,3 +1,3 @@
 module CSVStore
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Karl Ravn
@@ -41,6 +41,7 @@ extra_rdoc_files: []
 files: 
 - README.markdown
 - Rakefile
+- lib/csv_store/constants.rb
 - lib/csv_store/database.rb
 - lib/csv_store/file.rb
 - lib/csv_store/version.rb