# encoding: UTF-8 module Axlsx # The Worksheet class represents a worksheet in the workbook. class Worksheet include Axlsx::OptionsParser include Axlsx::SerializedAttributes # Creates a new worksheet. # @note the recommended way to manage worksheets is Workbook#add_worksheet # @see Workbook#add_worksheet # @option options [String] name The name of this worksheet. # @option options [Hash] page_margins A hash containing page margins for this worksheet. @see PageMargins # @option options [Hash] print_options A hash containing print options for this worksheet. @see PrintOptions # @option options [Hash] header_footer A hash containing header/footer options for this worksheet. @see HeaderFooter # @option options [Boolean] show_gridlines indicates if gridlines should be shown for this sheet. def initialize(wb, options={}) self.workbook = wb @sheet_protection = nil initialize_page_options(options) parse_options options @workbook.worksheets << self @sheet_id = index + 1 yield self if block_given? end serializable_attributes :sheet_id, :state # Initalizes page margin, setup and print options # @param [Hash] options Options passed in from the initializer def initialize_page_options(options) @page_margins = PageMargins.new options[:page_margins] if options[:page_margins] @page_setup = PageSetup.new options[:page_setup] if options[:page_setup] @print_options = PrintOptions.new options[:print_options] if options[:print_options] @header_footer = HeaderFooter.new options[:header_footer] if options[:header_footer] @row_breaks = RowBreaks.new @col_breaks = ColBreaks.new end # The name of the worksheet # @return [String] def name @name ||= "Sheet" + (index+1).to_s end # Specifies the visible state of this sheet. Allowed states are # :visible, :hidden or :very_hidden. The default value is :visible. # # Worksheets in the :hidden state can be shown using the sheet formatting properties in excel. # :very_hidden sheets should be inaccessible to end users. # @param [Symbol] sheet_state The visible state for this sheet. def state=(sheet_state) RestrictionValidator.validate :worksheet_state, [:visible, :hidden, :very_hidden], sheet_state @state = sheet_state end # The visibility of this sheet def state @state ||= :visible end # The sheet calculation properties # @return [SheetCalcPr] def sheet_calc_pr @sheet_calc_pr ||= SheetCalcPr.new end # The sheet protection object for this workbook # @return [SheetProtection] # @see SheetProtection def sheet_protection @sheet_protection ||= SheetProtection.new yield @sheet_protection if block_given? @sheet_protection end # The sheet view object for this worksheet # @return [SheetView] # @see [SheetView] def sheet_view @sheet_view ||= SheetView.new yield @sheet_view if block_given? @sheet_view end # The sheet format pr for this worksheet # @return [SheetFormatPr] # @see [SheetFormatPr] def sheet_format_pr @sheet_format_pr ||= SheetFormatPr.new yield @sheet_format_pr if block_given? @sheet_format_pr end # The workbook that owns this worksheet # @return [Workbook] attr_reader :workbook # The tables in this worksheet # @return [Array] of Table def tables @tables ||= Tables.new self end # The pivot tables in this worksheet # @return [Array] of Table def pivot_tables @pivot_tables ||= PivotTables.new self end # A collection of column breaks added to this worksheet # @note Please do not use this directly. Instead use # add_page_break # @see Worksheet#add_page_break def col_breaks @col_breaks ||= ColBreaks.new end # A collection of row breaks added to this worksheet # @note Please do not use this directly. Instead use # add_page_break # @see Worksheet#add_page_break def row_breaks @row_breaks ||= RowBreaks.new end # A typed collection of hyperlinks associated with this worksheet # @return [WorksheetHyperlinks] def hyperlinks @hyperlinks ||= WorksheetHyperlinks.new self end # The a shortcut to the worksheet_comments list of comments # @return [Array|SimpleTypedList] def comments worksheet_comments.comments if worksheet_comments.has_comments? end # The rows in this worksheet # @note The recommended way to manage rows is Worksheet#add_row # @return [SimpleTypedList] # @see Worksheet#add_row def rows @rows ||= SimpleTypedList.new Row end # returns the sheet data as columns # If you pass a block, it will be evaluated whenever a row does not have a # cell at a specific index. The block will be called with the row and column # index in the missing cell was found. # @example # cols { |row_index, column_index| puts "warn - row #{row_index} does not have a cell at #{column_index}" } def cols(&block) @rows.transpose(&block) end # A range that excel will apply an auto-filter to "A1:B3" # This will turn filtering on for the cells in the range. # The first row is considered the header, while subsequent rows are considered to be data. # @return String def auto_filter @auto_filter ||= AutoFilter.new self end # Indicates if the worksheet will be fit by witdh or height to a specific number of pages. # To alter the width or height for page fitting, please use page_setup.fit_to_widht or page_setup.fit_to_height. # If you want the worksheet to fit on more pages (e.g. 2x2), set {PageSetup#fit_to_width} and {PageSetup#fit_to_height} accordingly. # @return Boolean # @see #page_setup def fit_to_page? return false unless self.instance_values.keys.include?('page_setup') page_setup.fit_to_page? end # Column info for the sheet # @return [SimpleTypedList] def column_info @column_info ||= Cols.new self end # Page margins for printing the worksheet. # @example # wb = Axlsx::Package.new.workbook # # using options when creating the worksheet. # ws = wb.add_worksheet :page_margins => {:left => 1.9, :header => 0.1} # # # use the set method of the page_margins object # ws.page_margins.set(:bottom => 3, :footer => 0.7) # # # set page margins in a block # ws.page_margins do |margins| # margins.right = 6 # margins.top = 0.2 # end # @see PageMargins#initialize # @return [PageMargins] def page_margins @page_margins ||= PageMargins.new yield @page_margins if block_given? @page_margins end # Page setup settings for printing the worksheet. # @example # wb = Axlsx::Package.new.workbook # # # using options when creating the worksheet. # ws = wb.add_worksheet :page_setup => {:fit_to_width => 2, :orientation => :landscape} # # # use the set method of the page_setup object # ws.page_setup.set(:paper_width => "297mm", :paper_height => "210mm") # # # setup page in a block # ws.page_setup do |page| # page.scale = 80 # page.orientation = :portrait # end # @see PageSetup#initialize # @return [PageSetup] def page_setup @page_setup ||= PageSetup.new yield @page_setup if block_given? @page_setup end # Options for printing the worksheet. # @example # wb = Axlsx::Package.new.workbook # # using options when creating the worksheet. # ws = wb.add_worksheet :print_options => {:grid_lines => true, :horizontal_centered => true} # # # use the set method of the page_margins object # ws.print_options.set(:headings => true) # # # set page margins in a block # ws.print_options do |options| # options.horizontal_centered = true # options.vertical_centered = true # end # @see PrintOptions#initialize # @return [PrintOptions] def print_options @print_options ||= PrintOptions.new yield @print_options if block_given? @print_options end # Options for headers and footers. # @example # wb = Axlsx::Package.new.workbook # # would generate something like: "file.xlsx : sheet_name 2 of 7 date with timestamp" # header = {:different_odd_ => false, :odd_header => "&L&F : &A&C&Pof%N%R%D %T"} # ws = wb.add_worksheet :header_footer => header # # @see HeaderFooter#initialize # @return [HeaderFooter] def header_footer @header_footer ||= HeaderFooter.new yield @header_footer if block_given? @header_footer end # convenience method to access all cells in this worksheet # @return [Array] cells def cells rows.flatten end # Creates merge information for this worksheet. # Cells can be merged by calling the merge_cells method on a worksheet. # @example This would merge the three cells C1..E1 # # worksheet.merge_cells "C1:E1" # # you can also provide an array of cells to be merged # worksheet.merge_cells worksheet.rows.first.cells[(2..4)] # #alternatively you can do it from a single cell # worksheet["C1"].merge worksheet["E1"] # @param [Array, string] cells def merge_cells(cells) merged_cells.add cells end # Adds a new protected cell range to the worksheet. Note that protected ranges are only in effect when sheet protection is enabled. # @param [String|Array] cells The string reference for the cells to protect or an array of cells. # @return [ProtectedRange] # @note When using an array of cells, a contiguous range is created from the minimum top left to the maximum top bottom of the cells provided. def protect_range(cells) protected_ranges.add_range(cells) end # The dimensions of a worksheet. This is not actually a required element by the spec, # but at least a few other document readers expect this for conversion # @return [Dimension] def dimension @dimension ||= Dimension.new self end # The sheet properties for this workbook. # Currently only pageSetUpPr -> fitToPage is implemented # @return [SheetPr] def sheet_pr @sheet_pr ||= SheetPr.new self end # The name of the worksheet # The name of a worksheet must be unique in the workbook, and must not exceed 31 characters # @param [String] name def name=(name) validate_sheet_name name @name=Axlsx::coder.encode(name) end # The auto filter range for the worksheet # @param [String] v # @see auto_filter def auto_filter=(v) DataTypeValidator.validate :worksheet_auto_filter, String, v auto_filter.range = v end # Accessor for controlling whether leading and trailing spaces in cells are # preserved or ignored. The default is to preserve spaces. attr_accessor :preserve_spaces # The part name of this worksheet # @return [String] def pn "#{WORKSHEET_PN % (index+1)}" end # The relationship part name of this worksheet # @return [String] def rels_pn "#{WORKSHEET_RELS_PN % (index+1)}" end # The relationship id of this worksheet. # @return [String] # @see Relationship#Id def rId @workbook.relationships.for(self).Id end # The index of this worksheet in the owning Workbook's worksheets list. # @return [Integer] def index @workbook.worksheets.index(self) end # The drawing associated with this worksheet. # @note the recommended way to work with drawings and charts is Worksheet#add_chart # @return [Drawing] # @see Worksheet#add_chart def drawing worksheet_drawing.drawing end # Adds a row to the worksheet and updates auto fit data. # @example - put a vanilla row in your spreadsheet # ws.add_row [1, 'fish on my pl', '8'] # # @example - specify a fixed width for a column in your spreadsheet # # The first column will ignore the content of this cell when calculating column autowidth. # # The second column will include this text in calculating the columns autowidth # # The third cell will set a fixed with of 80 for the column. # # If you need to un-fix a column width, use :auto. That will recalculate the column width based on all content in the column # # ws.add_row ['I wish', 'for a fish', 'on my fish wish dish'], :widths=>[:ignore, :auto, 80] # # @example - specify a fixed height for a row # ws.add_row ['I wish', 'for a fish', 'on my fish wish dish'], :height => 40 # # @example - create and use a style for all cells in the row # blue = ws.styles.add_style :color => "#00FF00" # ws.add_row [1, 2, 3], :style=>blue # # @example - only style some cells # blue = ws.styles.add_style :color => "#00FF00" # red = ws.styles.add_style :color => "#FF0000" # big = ws.styles.add_style :sz => 40 # ws.add_row ["red fish", "blue fish", "one fish", "two fish"], :style=>[red, blue, nil, big] # the last nil is optional # # # @example - force the second cell to be a float value # ws.add_row [3, 4, 5], :types => [nil, :float] # # @example - use << alias # ws << [3, 4, 5], :types => [nil, :float] # # @example - specify whether a row should escape formulas or not # ws.add_row ['=IF(2+2=4,4,5)', 2, 3], :escape_formulas=>true # # @example - specify whether a certain cells in a row should escape formulas or not # ws.add_row ['=IF(2+2=4,4,5)', '=IF(13+13=4,4,5)'], :escape_formulas=>[true, false] # # @see Worksheet#column_widths # @return [Row] # @option options [Array] values # @option options [Array, Symbol] types # @option options [Array, Integer] style # @option options [Array] widths each member of the widths array will affect how auto_fit behavies. # @option options [Float] height the row's height (in points) # @option options [Array, Boolean] escape_formulas - Whether to treat a value starting with an equal # sign as formula (default) or as simple string. # Allowing user generated data to be interpreted as formulas can be dangerous # (see https://www.owasp.org/index.php/CSV_Injection for details). def add_row(values=[], options={}) row = Row.new(self, values, options) update_column_info row, options.delete(:widths) yield row if block_given? row end alias :<< :add_row # Add conditional formatting to this worksheet. # # @param [String] cells The range to apply the formatting to # @param [Array|Hash] rules An array of hashes (or just one) to create Conditional formatting rules from. # @example This would format column A whenever it is FALSE. # # for a longer example, see examples/example_conditional_formatting.rb (link below) # worksheet.add_conditional_formatting( "A1:A1048576", { :type => :cellIs, :operator => :equal, :formula => "FALSE", :dxfId => 1, :priority => 1 } # # @see ConditionalFormattingRule#initialize # @see file:examples/example_conditional_formatting.rb def add_conditional_formatting(cells, rules) cf = ConditionalFormatting.new( :sqref => cells ) cf.add_rules rules conditional_formattings << cf conditional_formattings end # Add data validation to this worksheet. # # @param [String] cells The cells the validation will apply to. # @param [hash] data_validation options defining the validation to apply. # @see examples/data_validation.rb for an example def add_data_validation(cells, data_validation) dv = DataValidation.new(data_validation) dv.sqref = cells data_validations << dv end # Adds a new hyperlink to the worksheet # @param [Hash] options for the hyperlink # @see WorksheetHyperlink for a list of options # @return [WorksheetHyperlink] def add_hyperlink(options={}) hyperlinks.add(options) end # Adds a chart to this worksheets drawing. This is the recommended way to create charts for your worksheet. This method wraps the complexity of dealing with ooxml drawing, anchors, markers graphic frames chart objects and all the other dirty details. # @param [Class] chart_type # @option options [Array] start_at # @option options [Array] end_at # @option options [Cell, String] title # @option options [Boolean] show_legend # @option options [Integer] style # @note each chart type also specifies additional options # @see Chart # @see Pie3DChart # @see Bar3DChart # @see Line3DChart # @see README for examples def add_chart(chart_type, options={}) chart = worksheet_drawing.add_chart(chart_type, options) yield chart if block_given? chart end # needs documentation def add_table(ref, options={}) tables << Table.new(ref, self, options) yield tables.last if block_given? tables.last end def add_pivot_table(ref, range, options={}) pivot_tables << PivotTable.new(ref, range, self, options) yield pivot_tables.last if block_given? pivot_tables.last end # Shortcut to worsksheet_comments#add_comment def add_comment(options={}) worksheet_comments.add_comment(options) end # Adds a media item to the worksheets drawing # @option [Hash] options options passed to drawing.add_image def add_image(options={}) image = worksheet_drawing.add_image(options) yield image if block_given? image end # Adds a page break (row break) to the worksheet # @param cell A Cell object or excel style string reference indicating where the break # should be added to the sheet. # @example # ws.add_page_break("A4") def add_page_break(cell) DataTypeValidator.validate :worksheet_page_break, [String, Cell], cell column_index, row_index = if cell.is_a?(String) Axlsx.name_to_indices(cell) else cell.pos end if column_index > 0 col_breaks.add_break(:id => column_index) end row_breaks.add_break(:id => row_index) end # This is a helper method that Lets you specify a fixed width for multiple columns in a worksheet in one go. # Note that you must call column_widths AFTER adding data, otherwise the width will not be set successfully. # Setting a fixed column width to nil will revert the behaviour back to calculating the width for you on the next call to add_row. # @example This would set the first and third column widhts but leave the second column in autofit state. # ws.column_widths 7.2, nil, 3 # @note For updating only a single column it is probably easier to just set the width of the ws.column_info[col_index].width directly # @param [Integer|Float|nil] widths def column_widths(*widths) widths.each_with_index do |value, index| next if value == nil Axlsx::validate_unsigned_numeric(value) unless value == nil find_or_create_column_info(index).width = value end end # Set the style for cells in a specific column # @param [Integer] index the index of the column # @param [Integer] style the cellXfs index # @param [Hash] options # @option [Integer] :row_offset only cells after this column will be updated. # @note You can also specify the style for specific columns in the call to add_row by using an array for the :styles option # @see Worksheet#add_row # @see README.md for an example def col_style(index, style, options={}) offset = options.delete(:row_offset) || 0 cells = @rows[(offset..-1)].map { |row| row[index] }.flatten.compact cells.each { |cell| cell.style = style } end # Set the style for cells in a specific row # @param [Integer] index or range of indexes in the table # @param [Integer] style the cellXfs index # @param [Hash] options the options used when applying the style # @option [Integer] :col_offset only cells after this column will be updated. # @note You can also specify the style in the add_row call # @see Worksheet#add_row # @see README.md for an example def row_style(index, style, options={}) offset = options.delete(:col_offset) || 0 cells = cols[(offset..-1)].map { |column| column[index] }.flatten.compact cells.each { |cell| cell.style = style } end # Returns a sheet node serialization for this sheet in the workbook. def to_sheet_node_xml_string(str='') add_autofilter_defined_name_to_workbook str << '') end # Serializes the worksheet object to an xml string # This intentionally does not use nokogiri for performance reasons # @return [String] def to_xml_string str='' add_autofilter_defined_name_to_workbook auto_filter.apply if auto_filter.range str << '' str << worksheet_node serializable_parts.each do |item| item.to_xml_string(str) if item end str << '' end # The worksheet relationships. This is managed automatically by the worksheet # @return [Relationships] def relationships r = Relationships.new r + [tables.relationships, worksheet_comments.relationships, hyperlinks.relationships, worksheet_drawing.relationship, pivot_tables.relationships].flatten.compact || [] r end # Returns the cell or cells defined using excel style A1:B3 references. # @param [String|Integer] cell_def the string defining the cell or range of cells, or the rownumber # @return [Cell, Array] def [](cell_def) return rows[cell_def] if cell_def.is_a?(Integer) parts = cell_def.split(':').map{ |part| name_to_cell part } if parts.size == 1 parts.first else range(*parts) end end # returns the column and row index for a named based cell # @param [String] name The cell or cell range to return. "A1" will return the first cell of the first row. # @return [Cell] def name_to_cell(name) col_index, row_index = *Axlsx::name_to_indices(name) r = rows[row_index] r[col_index] if r end # shortcut method to access styles direclty from the worksheet # This lets us do stuff like: # @example # p = Axlsx::Package.new # p.workbook.add_worksheet(:name => 'foo') do |sheet| # my_style = sheet.styles.add_style { :bg_color => "FF0000" } # sheet.add_row ['Oh No!'], :styles => my_style # end # p.serialize 'foo.xlsx' def styles @styles ||= self.workbook.styles end # shortcut level to specify the outline level for a series of rows # Oulining is what lets you add collapse and expand to a data set. # @param [Integer] start_index The zero based index of the first row of outlining. # @param [Integer] end_index The zero based index of the last row to be outlined # @param [integer] level The level of outline to apply # @param [Integer] collapsed The initial collapsed state of the outline group def outline_level_rows(start_index, end_index, level = 1, collapsed = true) outline rows, (start_index..end_index), level, collapsed end # shortcut level to specify the outline level for a series of columns # Oulining is what lets you add collapse and expand to a data set. # @param [Integer] start_index The zero based index of the first column of outlining. # @param [Integer] end_index The zero based index of the last column to be outlined # @param [integer] level The level of outline to apply # @param [Integer] collapsed The initial collapsed state of the outline group def outline_level_columns(start_index, end_index, level = 1, collapsed = true) outline column_info, (start_index..end_index), level, collapsed end private def xml_space workbook.xml_space end def outline(collection, range, level = 1, collapsed = true) range.each do |index| unless (item = collection[index]).nil? item.outline_level = level item.hidden = collapsed end sheet_view.show_outline_symbols = true end end def validate_sheet_name(name) DataTypeValidator.validate :worksheet_name, String, name # ignore first character (BOM) after encoding to utf16 because Excel does so, too. character_length = name.encode("utf-16")[1..-1].encode("utf-16").bytesize / 2 raise ArgumentError, (ERR_SHEET_NAME_TOO_LONG % name) if character_length > 31 raise ArgumentError, (ERR_SHEET_NAME_CHARACTER_FORBIDDEN % name) if '[]*/\?:'.chars.any? { |char| name.include? char } name = Axlsx::coder.encode(name) sheet_names = @workbook.worksheets.reject { |s| s == self }.map { |s| s.name } raise ArgumentError, (ERR_DUPLICATE_SHEET_NAME % name) if sheet_names.include?(name) end def serializable_parts [sheet_pr, dimension, sheet_view, sheet_format_pr, column_info, sheet_data, sheet_calc_pr, @sheet_protection, protected_ranges, auto_filter, merged_cells, conditional_formattings, data_validations, hyperlinks, print_options, page_margins, page_setup, header_footer, row_breaks, col_breaks, worksheet_drawing, worksheet_comments, tables] end def range(*cell_def) first, last = cell_def cells = [] rows[(first.row.row_index..last.row.row_index)].each do |r| r[(first.index..last.index)].each do |c| cells << c end end cells end # A collection of protected ranges in the worksheet # @note The recommended way to manage protected ranges is with Worksheet#protect_range # @see Worksheet#protect_range # @return [SimpleTypedList] The protected ranges for this worksheet def protected_ranges @protected_ranges ||= ProtectedRanges.new self # SimpleTypedList.new ProtectedRange end # conditional formattings # @return [Array] def conditional_formattings @conditional_formattings ||= ConditionalFormattings.new self end # data validations array # @return [Array] def data_validations @data_validations ||= DataValidations.new self end # merged cells array # @return [Array] def merged_cells @merged_cells ||= MergedCells.new self end # Helper method for parsingout the root node for worksheet # @return [String] def worksheet_node "" end def sheet_data @sheet_data ||= SheetData.new self end def worksheet_drawing @worksheet_drawing ||= WorksheetDrawing.new self end # The comments associated with this worksheet # @return [SimpleTypedList] def worksheet_comments @worksheet_comments ||= WorksheetComments.new self end def workbook=(v) DataTypeValidator.validate "Worksheet.workbook", Workbook, v; @workbook = v; end def update_column_info(cells, widths=nil) cells.each_with_index do |cell, index| width = widths ? widths[index] : nil col = find_or_create_column_info(index) next if width == :ignore col.update_width(cell, width, workbook.use_autowidth) end end def find_or_create_column_info(index) column_info[index] ||= Col.new(index + 1, index + 1) end def add_autofilter_defined_name_to_workbook return if !auto_filter.range workbook.add_defined_name auto_filter.defined_name, name: '_xlnm._FilterDatabase', local_sheet_id: index, hidden: 1 end end end