rsr_group 1.6.2 → 1.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fa4198a0176561bf852aee7ca7c1df677009644b
-  data.tar.gz: 84c44fffbc967de623172cc301bb347ae252b297
+  metadata.gz: b8e45f8036e006550b14071c2cf15167571e8803
+  data.tar.gz: 53dbdea34eac71093fe91859e06ab73b117ece41
 SHA512:
-  metadata.gz: 2a2a11c324f55c6638aaad729fd83365d4cb624ff67636c7d949efcde5b91078cce856addca1345fb54783bc6dc79254c32c7ed1d276b0ecec029052a1d9acb3
-  data.tar.gz: 689d5eea5422d3fe95b71c71c21ece566d596cd9896fb961f3e354c11970b38f92964a5447fb35995979cd90bf6fc40fdec889fcd87fab330fe5c8f2909e8c1a
+  metadata.gz: f0095e4f51ec8fdfe06425ff99c2bb73957817a1eab7014f99592080bdd83cd2171cc0a99c68c081225031ea58a8084b9a0559d4810230ac740975ee960009ed
+  data.tar.gz: b5dce8d75c8a82d4e821523596de3ac763e712cddc565682c43a911f7a6726444bb9ed1b174f50d5ed18cbe66fe7f3357092a6353763b3015190b54d3a6e1b61

data/lib/rsr_group.rb

@@ -7,6 +7,7 @@ require 'smarter_csv'
 
 require 'rsr_group/base'
 require 'rsr_group/constants'
+require 'rsr_group/chunker'
 require 'rsr_group/data_row'
 require 'rsr_group/department'
 require 'rsr_group/inventory'

data/lib/rsr_group/chunker.rb

@@ -0,0 +1,34 @@
+module RsrGroup
+  class Chunker
+
+    attr_accessor :chunk, :total_count, :current_count, :size
+
+    def initialize(size, total_count = nil)
+      @size           = size
+      @chunk          = Array.new
+      @current_count  = 0
+      @total_count    = total_count
+    end
+
+    def add(row)
+      reset if is_full?
+
+      @chunk.push(row)
+
+      @current_count += 1
+    end
+
+    def reset
+      @chunk.clear
+    end
+
+    def is_full?
+      @chunk.count == @size
+    end
+
+    def is_complete?
+      @total_count == @current_count
+    end
+
+  end
+end

data/lib/rsr_group/inventory.rb

@@ -30,11 +30,25 @@ module RsrGroup
       new(options).map_prices
     end
 
+    def self.map_prices_as_chunks(size = 15, options = {})
+      requires!(options, :username, :password)
+      new(options).map_prices_as_chunks(size) do |chunk|
+        yield(chunk)
+      end
+    end
+
     def self.quantities(options = {})
       requires!(options, :username, :password)
       new(options).quantities
     end
 
+    def self.quantities_as_chunks(size = 15, options = {})
+      requires!(options, :username, :password)
+      new(options).quantities_as_chunks(size) do |chunk|
+        yield(chunk)
+      end
+    end
+
     def all
       items = []
 
@@ -138,40 +152,42 @@ module RsrGroup
 
     def process_as_chunks(size)
       connect(@options) do |ftp|
-        chunk         = []
-        item_count    = 1
+        chunker       = RsrGroup::Chunker.new(size)
         temp_csv_file = Tempfile.new
 
+        # Is this a key dealer?
         if ftp.nlst.include?(KEYDEALER_DIR)
           ftp.chdir(KEYDEALER_DIR)
+
+          # Pull from the FTP and save as a temp file
           ftp.getbinaryfile(KEYDEALER_FILENAME, temp_csv_file.path)
         else
           ftp.chdir(INVENTORY_DIR)
+
+          # Pull from the FTP and save as a temp file
           ftp.getbinaryfile(INVENTORY_FILENAME, temp_csv_file.path)
         end
 
-        line_count = File.readlines(temp_csv_file).size
+        # total_count is the number of lines in the file
+        chunker.total_count = File.readlines(temp_csv_file).size
 
         CSV.readlines(temp_csv_file, col_sep: ';', quote_char: "\x00").to_enum.with_index(1).each do |row, current_line|
-          chunk << process_row(row)
 
-          # Return chunk if we have reached the end of the file
-          if line_count == current_line
-            yield(chunk)
-          end
+          if chunker.is_full?
+            yield(chunker.chunk)
 
-          # Return if chuck is full
-          if item_count == size
-            yield(chunk)
+            chunker.reset
+          elsif chunker.is_complete?
+            yield(chunker.chunk)
 
-            chunk      = []
-            item_count = 1
+            break
           else
-            item_count += 1
+            chunker.add(process_row(row))
           end
         end
 
         temp_csv_file.unlink
+        ftp.close
       end
     end
 
@@ -199,6 +215,46 @@ module RsrGroup
       rows
     end
 
+    def map_prices_as_chunks(size)
+      connect(@options) do |ftp|
+        chunker       = RsrGroup::Chunker.new(size)
+        temp_csv_file = Tempfile.new
+
+        # Is this a key dealer?
+        if ftp.nlst.include?(KEYDEALER_DIR)
+          ftp.chdir(KEYDEALER_DIR)
+        else
+          ftp.chdir(INVENTORY_DIR)
+        end
+
+        # Pull from the FTP and save as a temp file
+        ftp.getbinaryfile(MAP_FILENAME, temp_csv_file.path)
+
+        # total_count is hte number of lines in the file
+        chunker.total_count = File.readlines(temp_csv_file).size
+
+        CSV.readlines(temp_csv_file).each do |row|
+          if chunker.is_full?
+            yield(chunker.chunk)
+
+            chunker.reset
+          elsif chunker.is_complete?
+            yield(chunker.chunk)
+
+            break
+          else
+            chunker.add({
+              stock_number: row[0].strip,
+              map_price:    row[1]
+            })
+          end
+        end
+
+        temp_csv_file.unlink
+        ftp.close
+      end
+    end
+
     def quantities
       rows = []
 
@@ -224,6 +280,47 @@ module RsrGroup
       rows
     end
 
+    def quantities_as_chunks(size)
+      connect(@options) do |ftp|
+        chunker       = RsrGroup::Chunker.new(size)
+        temp_csv_file = Tempfile.new
+
+        # Is this a key dealer?
+        if ftp.nlst.include?(KEYDEALER_DIR)
+          ftp.chdir(KEYDEALER_DIR)
+        else
+          ftp.chdir(INVENTORY_DIR)
+        end
+
+        # Pull from the FTP and save as a temp file
+        ftp.getbinaryfile(QTY_FILENAME, temp_csv_file.path)
+
+        # total_count is hte number of lines in the file
+        chunker.total_count = File.readlines(temp_csv_file).size
+
+        CSV.readlines(temp_csv_file).each do |row|
+          if chunker.is_full?
+            yield(chunker.chunk)
+
+            chunker.reset
+          elsif chunker.is_complete?
+            yield(chunker.chunk)
+
+            break
+          else
+            chunker.add({
+              stock_number: row[0].rstrip,
+              quantity: row[1].to_i
+            })
+          end
+        end
+
+        temp_csv_file.unlink
+        ftp.close
+      end
+    end
+
+
     private
 
     def sanitize(data)

data/lib/rsr_group/version.rb

@@ -1,3 +1,3 @@
 module RsrGroup
-  VERSION = '1.6.2'
+  VERSION = '1.7.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rsr_group
 version: !ruby/object:Gem::Version
-  version: 1.6.2
+  version: 1.7.0
 platform: ruby
 authors:
 - Dale Campbell
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-03-27 00:00:00.000000000 Z
+date: 2017-03-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: smarter_csv
@@ -101,6 +101,7 @@ files:
 - config.rb.example
 - lib/rsr_group.rb
 - lib/rsr_group/base.rb
+- lib/rsr_group/chunker.rb
 - lib/rsr_group/constants.rb
 - lib/rsr_group/data_row.rb
 - lib/rsr_group/department.rb