Contents

module DocParser
  # The Output base class.
  # All Output classes inherit from this one.
  class Output
    attr_reader :rowcount

    # Creates a new output
    # @param filename [String] Output filename
    def initialize(filename: filename)
      @rowcount = 0
      @filename = filename
      raise ArgumentError, 'Please specify a filename' if filename.empty?
      @file = open filename, 'w'
      open_file
    end

    # Stores the header
    def header=(row)
      @header = row
      header
    end

    # Adds a row
    def add_row(row)
      @rowcount += 1
      write_row row
    end

    # Closes output and IO
    def close
      footer
      @file.close unless @file.closed?
    end

    # Called after the file is opened
    def open_file
      # do nothing
    end

    # Called after header is set
    def header
      # do nothing
    end

    # Called when a row is added
    def write_row(row)
      raise 'No row writer defined'
    end

    # Called before closing the file
    def footer
    end

    # Displays information about the output
    # @return [String] containing number of rows and file size
    def summary
      "%s:\t%d rows, %9.2f KiB" % [@filename,
                                   @rowcount,
                                   File.size(@filename) / 1024.0]
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
docparser-0.0.1	lib/docparser/output.rb