squib 0.15.1 → 0.16.0

data/lib/squib/import/quantity_exploder.rb

@@ -0,0 +1,18 @@
+module Squib
+  module Import
+    module QuantityExploder
+      def explode_quantities(data, qty)
+        return data unless data.col? qty.to_s.strip
+        qtys = data[qty]
+        new_data = Squib::DataFrame.new
+        data.each do |col, arr|
+          new_data[col] = []
+          qtys.each_with_index do |qty, index|
+            qty.to_i.times { new_data[col] << arr[index] }
+          end
+        end
+        return new_data
+      end
+    end
+  end
+end

data/lib/squib/import/xlsx_importer.rb

@@ -0,0 +1,28 @@
+require 'roo'
+require_relative 'quantity_exploder'
+
+module Squib::Import
+  class XlsxImporter
+    include Squib::Import::QuantityExploder
+    def import_to_dataframe(import, &block)
+      s = Roo::Excelx.new(import.file)
+      s.default_sheet = s.sheets[import.sheet]
+      data = Squib::DataFrame.new
+      s.first_column.upto(s.last_column) do |col|
+        header = s.cell(s.first_row, col).to_s
+        header.strip! if import.strip?
+        data[header] = []
+        (s.first_row + 1).upto(s.last_row) do |row|
+          cell = s.cell(row, col)
+          # Roo hack for avoiding unnecessary .0's on whole integers (https://github.com/roo-rb/roo/issues/139)
+          cell = s.excelx_value(row, col) if s.excelx_type(row, col) == [:numeric_or_formula, 'General']
+          cell.strip! if cell.respond_to?(:strip) && import.strip?
+          cell = block.yield(header, cell) unless block.nil?
+          data[header] << cell
+        end# row
+      end# col
+      explode_quantities(data, import.explode)
+    end
+  end
+end
+

data/lib/squib/import/yaml_importer.rb

@@ -0,0 +1,30 @@
+require 'yaml'
+require_relative 'data_frame'
+require_relative 'quantity_exploder'
+
+module Squib::Import
+  class YamlImporter
+    include Squib::Import::QuantityExploder
+    def import_to_dataframe(import, &block)
+      data = import.data.nil? ? File.read(import.file) : import.data
+      yml = YAML.load(data)
+      data = Squib::DataFrame.new
+      # Get a universal list of keys to ensure everything is covered.
+      keys = yml.map { |c| c.keys}.flatten.uniq
+      keys.each { |k| data[k] = [] } #init arrays
+      yml.each do |card|
+      # nil value if key isn't set.
+      keys.each { |k| data[k] << card[k] }
+      end
+      unless block.nil?
+        data.each do |header, col|
+          col.map! do |val|
+          block.yield(header, val)
+          end
+        end
+      end
+      explode_quantities(data, import.explode)
+    end
+  end
+end
+

data/lib/squib/layout_parser.rb

@@ -1,138 +1,155 @@
-require 'yaml'
-
-module Squib
-  # Internal class for handling layouts
-  # @api private
-  class LayoutParser
-
-    def initialize(dpi = 300)
-      @dpi = dpi
-    end
-
-    # Load the layout file(s), if exists
-    # @api private
-    def load_layout(files, initial = {})
-      layout = initial
-      Squib::logger.info { "  using layout(s): #{files}" }
-      Array(files).each do |file|
-        thefile = file
-        thefile = builtin(file) unless File.exists?(file)
-        if File.exists? thefile
-          # note: YAML.load_file returns false on empty file
-          yml = layout.merge(YAML.load_file(thefile) || {})
-          yml.each do |key, value|
-            layout[key] = recurse_extends(yml, key, {})
-          end
-        else
-          Squib::logger.error { "Layout file not found: #{file}. Skipping..." }
-        end
-      end
-      return layout
-    end
-
-    private
-
-    # Determine the file path of the built-in layout file
-    def builtin(file)
-      "#{File.dirname(__FILE__)}/builtin/layouts/#{file}"
-    end
-
-    # Process the extends recursively
-    # :nodoc:
-    # @api private
-    def recurse_extends(yml, key, visited)
-      assert_not_visited(key, visited)
-      return yml[key] unless has_extends?(yml, key)
-      return yml[key] unless parents_exist?(yml, key)
-      visited[key] = key
-      parent_keys = [yml[key]['extends']].flatten
-      h = {}
-      parent_keys.each do |parent_key|
-        from_extends = yml[key].merge(recurse_extends(yml, parent_key, visited)) do |key, child_val, parent_val|
-          handle_relative_operators(parent_val, child_val)
-        end
-        h = h.merge(from_extends) do |key, older_sibling, younger_sibling|
-          # In general, go with the younger sibling.
-          # UNLESS that younger sibling had a relative operator, in which use the
-          # (already computed) relative operator applied, which lands in older_sibling
-          # See bug 244.
-          sibling = younger_sibling
-          %w(+= -= *= /=).each do |op|
-            sibling = older_sibling if younger_sibling.to_s.strip.start_with? op
-          end
-          sibling
-        end
-      end
-      return h
-    end
-
-    def handle_relative_operators(parent_val, child_val)
-      if child_val.to_s.strip.start_with?('+=')
-        add_parent_child(parent_val, child_val)
-      elsif child_val.to_s.strip.start_with?('-=')
-        sub_parent_child(parent_val, child_val)
-      elsif child_val.to_s.strip.start_with?('*=')
-        mul_parent_child(parent_val, child_val)
-      elsif child_val.to_s.strip.start_with?('/=')
-        div_parent_child(parent_val, child_val)
-      else
-        child_val # child overrides parent when merging, no +=
-      end
-    end
-
-    def add_parent_child(parent, child)
-      parent_pixels = Args::UnitConversion.parse(parent, @dpi).to_f
-      child_pixels = Args::UnitConversion.parse(child.sub('+=', ''), @dpi).to_f
-      parent_pixels + child_pixels
-    end
-
-    def sub_parent_child(parent, child)
-      parent_pixels = Args::UnitConversion.parse(parent, @dpi).to_f
-      child_pixels = Args::UnitConversion.parse(child.sub('-=', ''), @dpi).to_f
-      parent_pixels - child_pixels
-    end
-
-    def mul_parent_child(parent, child)
-      parent_pixels = Args::UnitConversion.parse(parent, @dpi).to_f
-      child_float = child.sub('*=', '').to_f
-      parent_pixels * child_float
-    end
-
-    def div_parent_child(parent, child)
-      parent_pixels = Args::UnitConversion.parse(parent, @dpi).to_f
-      child_float = child.sub('/=', '').to_f
-      parent_pixels / child_float
-    end
-
-    # Does this layout entry have an extends field?
-    # i.e. is it a base-case or will it need recursion?
-    # :nodoc:
-    # @api private
-    def has_extends?(yml, key)
-      !!yml[key] && yml[key].key?('extends')
-    end
-
-    # Checks if we have any absentee parents
-    # @api private
-    def parents_exist?(yml, key)
-      exists = true
-      Array(yml[key]['extends']).each do |parent|
-        unless yml.key?(parent)
-          exists = false unless
-          Squib.logger.error "Processing layout: '#{key}' attempts to extend a missing '#{yml[key]['extends']}'"
-        end
-      end
-      return exists
-    end
-
-    # Safeguard against malformed circular extends
-    # :nodoc:
-    # @api private
-    def assert_not_visited(key, visited)
-      if visited.key? key
-        raise "Invalid layout: circular extends with '#{key}'"
-      end
-    end
-
-  end
-end
+require 'yaml'
+require_relative 'args/xywh_shorthands'
+
+module Squib
+  # Internal class for handling layouts
+  # @api private
+  class LayoutParser
+    include Args::XYWHShorthands
+
+    def initialize(dpi = 300, cell_px = 37.5)
+      @dpi = dpi
+      @cell_px = cell_px
+    end
+
+    # Load the layout file(s), if exists
+    # @api private
+    def load_layout(files, initial = {})
+      layout = initial
+      Squib::logger.info { "  using layout(s): #{files}" }
+      Array(files).each do |file|
+        thefile = file
+        thefile = builtin(file) unless File.exists?(file)
+        if File.exists? thefile
+          # note: YAML.load_file returns false on empty file
+          yml = layout.merge(YAML.load_file(thefile) || {})
+          yml.each do |key, value|
+            layout[key] = recurse_extends(yml, key, {})
+          end
+        else
+          Squib::logger.error { "Layout file not found: #{file}. Skipping..." }
+        end
+      end
+      return layout
+    end
+
+    private
+
+    # Determine the file path of the built-in layout file
+    def builtin(file)
+      "#{File.dirname(__FILE__)}/builtin/layouts/#{file}"
+    end
+
+    # Process the extends recursively
+    # :nodoc:
+    # @api private
+    def recurse_extends(yml, key, visited)
+      assert_not_visited(key, visited)
+      return yml[key] unless has_extends?(yml, key)
+      return yml[key] unless parents_exist?(yml, key)
+      visited[key] = key
+      parent_keys = [yml[key]['extends']].flatten
+      h = {}
+      parent_keys.each do |parent_key|
+        from_extends = yml[key].merge(recurse_extends(yml, parent_key, visited)) do |key, child_val, parent_val|
+          handle_relative_operators(parent_val, child_val)
+        end
+        h = h.merge(from_extends) do |key, older_sibling, younger_sibling|
+          # In general, go with the younger sibling.
+          # UNLESS that younger sibling had a relative operator, in which use the
+          # (already computed) relative operator applied, which lands in older_sibling
+          # See bug 244.
+          sibling = younger_sibling
+          %w(+= -= *= /=).each do |op|
+            sibling = older_sibling if younger_sibling.to_s.strip.start_with? op
+          end
+          sibling
+        end
+      end
+      return h
+    end
+
+    def handle_relative_operators(parent_val, child_val)
+      if uses_operator?(child_val) && !has_digits?(parent_val) && !has_digits?(child_val)
+        raise "Layout parse error: can't combine #{parent_val} and #{child_val}"
+      end
+      if child_val.to_s.strip.start_with?('+=')
+        add_parent_child(parent_val, child_val)
+      elsif child_val.to_s.strip.start_with?('-=')
+        sub_parent_child(parent_val, child_val)
+      elsif child_val.to_s.strip.start_with?('*=')
+        mul_parent_child(parent_val, child_val)
+      elsif child_val.to_s.strip.start_with?('/=')
+        div_parent_child(parent_val, child_val)
+      else
+        child_val # child overrides parent when merging, no +=
+      end
+    end
+
+    def add_parent_child(parent, child)
+      parent_pixels = Args::UnitConversion.parse(parent, @dpi, @cell_px).to_f
+      child_pixels = Args::UnitConversion.parse(child.sub('+=', ''), @dpi, @cell_px).to_f
+      parent_pixels + child_pixels
+    end
+
+    def sub_parent_child(parent, child)
+      parent_pixels = Args::UnitConversion.parse(parent, @dpi, @cell_px).to_f
+      child_pixels = Args::UnitConversion.parse(child.sub('-=', ''), @dpi, @cell_px).to_f
+      parent_pixels - child_pixels
+    end
+
+    def mul_parent_child(parent, child)
+      parent_pixels = Args::UnitConversion.parse(parent, @dpi, @cell_px).to_f
+      child_float = child.sub('*=', '').to_f
+      parent_pixels * child_float
+    end
+
+    def div_parent_child(parent, child)
+      parent_pixels = Args::UnitConversion.parse(parent, @dpi, @cell_px).to_f
+      child_float = child.sub('/=', '').to_f
+      parent_pixels / child_float
+    end
+
+    # For relative operators, it's difficult for us to handle
+    # some of the shorthands - so let's just freak out if you're trying to use
+    # relative operators with words, e.g. "middle += 0.5in"
+    def has_digits?(arg)
+      /.*\d.*/ === arg.to_s
+    end
+
+    def uses_operator?(arg)
+      /^[+=|\-=|\/=|*=].*/ === arg.to_s
+    end
+
+    # Does this layout entry have an extends field?
+    # i.e. is it a base-case or will it need recursion?
+    # :nodoc:
+    # @api private
+    def has_extends?(yml, key)
+      !!yml[key] && yml[key].key?('extends')
+    end
+
+    # Checks if we have any absentee parents
+    # @api private
+    def parents_exist?(yml, key)
+      exists = true
+      Array(yml[key]['extends']).each do |parent|
+        unless yml.key?(parent)
+          exists = false unless
+          Squib.logger.error "Processing layout: '#{key}' attempts to extend a missing '#{yml[key]['extends']}'"
+        end
+      end
+      return exists
+    end
+
+    # Safeguard against malformed circular extends
+    # :nodoc:
+    # @api private
+    def assert_not_visited(key, visited)
+      if visited.key? key
+        raise "Invalid layout: circular extends with '#{key}'"
+      end
+    end
+
+  end
+end

data/lib/squib/progress.rb

@@ -1,38 +1,38 @@
-require 'ruby-progressbar'
-
-module Squib
-
-  # :nodoc:
-  # @api private
-  class DoNothing
-    def increment
-    end
-
-    def finish
-    end
-  end
-
-  # A facade that handles (or doesn't) the progress bar on the console
-  #
-  # :nodoc:
-  # @api private
-  class Progress
-    attr_accessor :enabled
-
-    def initialize(enabled)
-      @enabled = enabled
-    end
-
-    def start(title='', total=100, &block)
-      if @enabled
-        @bar = ProgressBar.create(title: title, total: total, format: '%t <%B> %p%% %a')
-        yield(@bar)
-        @bar.finish
-      else
-        yield(Squib::DoNothing.new)
-      end
-    end
-  end
-
-
-end
+require 'ruby-progressbar'
+
+module Squib
+
+  # :nodoc:
+  # @api private
+  class DoNothing
+    def increment
+    end
+
+    def finish
+    end
+  end
+
+  # A facade that handles (or doesn't) the progress bar on the console
+  #
+  # :nodoc:
+  # @api private
+  class Progress
+    attr_accessor :enabled
+
+    def initialize(enabled)
+      @enabled = enabled
+    end
+
+    def start(title='', total=100, &block)
+      if @enabled
+        @bar = ProgressBar.create(title: title, total: total, format: '%t <%B> %p%% %a')
+        yield(@bar)
+        @bar.finish
+      else
+        yield(Squib::DoNothing.new)
+      end
+    end
+  end
+
+
+end