pollster 0.2.3 → 2.0.0

data/lib/pollster/models/poll_question_responses.rb

@@ -0,0 +1,197 @@
+require 'date'
+
+module Pollster
+
+  class PollQuestionResponses
+    # Prompt the survey house presented to the respondent. For instance: `Hillary Clinton, the Democrat`
+    attr_accessor :text
+
+    # One of `poll_question.question.responses[*].label`: a label chosen by Pollster editors. Often, two `text` entries will have the same `pollster_label`. Rarely, the `pollster_label` may be `null`.
+    attr_accessor :pollster_label
+
+    # Value reported by the survey house, out of 100. Survey houses typically weight respondents, meaning this you cannot guess the number of respondents from this value.
+    attr_accessor :value
+
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'text' => :'text',
+        :'pollster_label' => :'pollster_label',
+        :'value' => :'value'
+      }
+    end
+
+    # Attribute type mapping.
+    def self.swagger_types
+      {
+        :'text' => :'String',
+        :'pollster_label' => :'String',
+        :'value' => :'Float'
+      }
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      return unless attributes.is_a?(Hash)
+
+      # convert string to symbol for hash key
+      attributes = attributes.each_with_object({}){|(k,v), h| h[k.to_sym] = v}
+
+      if attributes.has_key?(:'text')
+        self.text = attributes[:'text']
+      end
+
+      if attributes.has_key?(:'pollster_label')
+        self.pollster_label = attributes[:'pollster_label']
+      end
+
+      if attributes.has_key?(:'value')
+        self.value = attributes[:'value']
+      end
+
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properies with the reasons
+    def list_invalid_properties
+      invalid_properties = Array.new
+      return invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      return true
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          text == o.text &&
+          pollster_label == o.pollster_label &&
+          value == o.value
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Fixnum] Hash code
+    def hash
+      [text, pollster_label, value].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      self.class.swagger_types.each_pair do |key, type|
+        if type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the the attribute
+          # is documented as an array but the input is not
+          if attributes[self.class.attribute_map[key]].is_a?(Array)
+            self.send("#{key}=", attributes[self.class.attribute_map[key]].map{ |v| _deserialize($1, v) } )
+          end
+        elsif !attributes[self.class.attribute_map[key]].nil?
+          self.send("#{key}=", _deserialize(type, attributes[self.class.attribute_map[key]]))
+        end # or else data not found in attributes(hash), not an issue as the data can be optional
+      end
+
+      self
+    end
+
+    # Deserializes the data based on type
+    # @param string type Data type
+    # @param string value Value to be deserialized
+    # @return [Object] Deserialized data
+    def _deserialize(type, value)
+      case type.to_sym
+      when :DateTime
+        DateTime.parse(value)
+      when :Date
+        Date.parse(value)
+      when :String
+        value.to_s
+      when :Integer
+        value.to_i
+      when :Float
+        value.to_f
+      when :BOOLEAN
+        if value.to_s =~ /\A(true|t|yes|y|1)\z/i
+          true
+        else
+          false
+        end
+      when :Object
+        # generic object (usually a Hash), return directly
+        value
+      when /\AArray<(?<inner_type>.+)>\z/
+        inner_type = Regexp.last_match[:inner_type]
+        value.map { |v| _deserialize(inner_type, v) }
+      when /\AHash<(?<k_type>.+?), (?<v_type>.+)>\z/
+        k_type = Regexp.last_match[:k_type]
+        v_type = Regexp.last_match[:v_type]
+        {}.tap do |hash|
+          value.each do |k, v|
+            hash[_deserialize(k_type, k)] = _deserialize(v_type, v)
+          end
+        end
+      else # model
+        temp_model = Pollster.const_get(type).new
+        temp_model.build_from_hash(value)
+      end
+    end
+
+    # Returns the string representation of the object
+    # @return [String] String presentation of the object
+    def to_s
+      to_hash.to_s
+    end
+
+    # to_body is an alias to to_hash (backward compatibility)
+    # @return [Hash] Returns the object in the form of hash
+    def to_body
+      to_hash
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        next if value.nil?
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+    # Outputs non-array value in the form of hash
+    # For object, use to_hash. Otherwise, just return the value
+    # @param [Object] value Any valid value
+    # @return [Hash] Returns the value in the form of hash
+    def _to_hash(value)
+      if value.is_a?(Array)
+        value.compact.map{ |v| _to_hash(v) }
+      elsif value.is_a?(Hash)
+        {}.tap do |hash|
+          value.each { |k, v| hash[k] = _to_hash(v) }
+        end
+      elsif value.respond_to? :to_hash
+        value.to_hash
+      else
+        value
+      end
+    end
+
+  end
+
+end

data/lib/pollster/models/poll_question_sample_subpopulations.rb

@@ -0,0 +1,209 @@
+require 'date'
+
+module Pollster
+
+  class PollQuestionSampleSubpopulations
+    # Number of people in this subpopulation who responded to this question, or `null` if the survey house did not report a number.
+    attr_accessor :observations
+
+    # One of: Adults, Adults - Democrat, Adults - Republican, Adults - independent, Registered Voters, Registered Voters - Democrat, Registered Voters - Republican, Registered Voters - independent, Likely Voters, Likely Voters - Democrat, Likely Voters - Republican, Likely Voters - independent. `Likely Voters` is editorially important during election seasons; sometimes Pollster editors will skip all other subpopulations.
+    attr_accessor :name
+
+    # Margin of error for this subpopulation, or `null` if the survey house did not report a number.
+    attr_accessor :margin_of_error
+
+    # Options the survey house provided to respondents, and respondents' answers. These are ordered randomly.
+    attr_accessor :responses
+
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'observations' => :'observations',
+        :'name' => :'name',
+        :'margin_of_error' => :'margin_of_error',
+        :'responses' => :'responses'
+      }
+    end
+
+    # Attribute type mapping.
+    def self.swagger_types
+      {
+        :'observations' => :'Integer',
+        :'name' => :'String',
+        :'margin_of_error' => :'Float',
+        :'responses' => :'Array<PollQuestionResponses>'
+      }
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      return unless attributes.is_a?(Hash)
+
+      # convert string to symbol for hash key
+      attributes = attributes.each_with_object({}){|(k,v), h| h[k.to_sym] = v}
+
+      if attributes.has_key?(:'observations')
+        self.observations = attributes[:'observations']
+      end
+
+      if attributes.has_key?(:'name')
+        self.name = attributes[:'name']
+      end
+
+      if attributes.has_key?(:'margin_of_error')
+        self.margin_of_error = attributes[:'margin_of_error']
+      end
+
+      if attributes.has_key?(:'responses')
+        if (value = attributes[:'responses']).is_a?(Array)
+          self.responses = value
+        end
+      end
+
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properies with the reasons
+    def list_invalid_properties
+      invalid_properties = Array.new
+      return invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      return true
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          observations == o.observations &&
+          name == o.name &&
+          margin_of_error == o.margin_of_error &&
+          responses == o.responses
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Fixnum] Hash code
+    def hash
+      [observations, name, margin_of_error, responses].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      self.class.swagger_types.each_pair do |key, type|
+        if type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the the attribute
+          # is documented as an array but the input is not
+          if attributes[self.class.attribute_map[key]].is_a?(Array)
+            self.send("#{key}=", attributes[self.class.attribute_map[key]].map{ |v| _deserialize($1, v) } )
+          end
+        elsif !attributes[self.class.attribute_map[key]].nil?
+          self.send("#{key}=", _deserialize(type, attributes[self.class.attribute_map[key]]))
+        end # or else data not found in attributes(hash), not an issue as the data can be optional
+      end
+
+      self
+    end
+
+    # Deserializes the data based on type
+    # @param string type Data type
+    # @param string value Value to be deserialized
+    # @return [Object] Deserialized data
+    def _deserialize(type, value)
+      case type.to_sym
+      when :DateTime
+        DateTime.parse(value)
+      when :Date
+        Date.parse(value)
+      when :String
+        value.to_s
+      when :Integer
+        value.to_i
+      when :Float
+        value.to_f
+      when :BOOLEAN
+        if value.to_s =~ /\A(true|t|yes|y|1)\z/i
+          true
+        else
+          false
+        end
+      when :Object
+        # generic object (usually a Hash), return directly
+        value
+      when /\AArray<(?<inner_type>.+)>\z/
+        inner_type = Regexp.last_match[:inner_type]
+        value.map { |v| _deserialize(inner_type, v) }
+      when /\AHash<(?<k_type>.+?), (?<v_type>.+)>\z/
+        k_type = Regexp.last_match[:k_type]
+        v_type = Regexp.last_match[:v_type]
+        {}.tap do |hash|
+          value.each do |k, v|
+            hash[_deserialize(k_type, k)] = _deserialize(v_type, v)
+          end
+        end
+      else # model
+        temp_model = Pollster.const_get(type).new
+        temp_model.build_from_hash(value)
+      end
+    end
+
+    # Returns the string representation of the object
+    # @return [String] String presentation of the object
+    def to_s
+      to_hash.to_s
+    end
+
+    # to_body is an alias to to_hash (backward compatibility)
+    # @return [Hash] Returns the object in the form of hash
+    def to_body
+      to_hash
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        next if value.nil?
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+    # Outputs non-array value in the form of hash
+    # For object, use to_hash. Otherwise, just return the value
+    # @param [Object] value Any valid value
+    # @return [Hash] Returns the value in the form of hash
+    def _to_hash(value)
+      if value.is_a?(Array)
+        value.compact.map{ |v| _to_hash(v) }
+      elsif value.is_a?(Hash)
+        {}.tap do |hash|
+          value.each { |k, v| hash[k] = _to_hash(v) }
+        end
+      elsif value.respond_to? :to_hash
+        value.to_hash
+      else
+        value
+      end
+    end
+
+  end
+
+end

data/lib/pollster/models/pollster_chart_poll_questions.rb

@@ -0,0 +1,92 @@
+require 'ruby-immutable-struct'
+
+module Pollster
+  PollsterChartPollQuestion = RubyImmutableStruct.new(
+    :responses,
+    :poll_slug,
+    :survey_house,
+    :start_date,
+    :end_date,
+    :question_text,
+    :sample_subpopulation,
+    :observations,
+    :margin_of_error,
+    :mode,
+    :partisanship,
+    :partisan_affiliation
+  ) do
+    def self.from_tsv_line_with_labels(tsv_line, labels)
+      if m = PollsterChartPollQuestion::TsvLineRegex.match(tsv_line)
+        # split(..., -1) avoids omitting trailing blanks
+        values = m[:values].split(/\t/, -1).map{ |v| v.empty? ? nil : v.to_f }
+
+        if values.length != labels.length
+          raise ArgumentError.new("TSV line `#{tsv_line}` has #{values.length} values for #{labels.length} labels")
+        end
+
+        responses = {}
+        for i in 0...(labels.length)
+          responses[labels[i]] = values[i]
+        end
+
+        PollsterChartPollQuestion.new(
+          responses,
+          m[:poll_slug],
+          m[:survey_house],
+          Date.parse(m[:start_date]),
+          Date.parse(m[:end_date]),
+          m[:question_text],
+          m[:sample_subpopulation],
+          m[:observations].empty? ? nil : m[:observations].to_i,
+          m[:margin_of_error].empty? ? nil : m[:margin_of_error].to_f,
+          m[:mode],
+          m[:partisanship],
+          m[:partisan_affiliation]
+        )
+      else
+        raise ArgumentError.new("TSV line `#{tsv_line}` is not a valid PollsterChartPollQuestion line")
+      end
+    end
+  end
+  PollsterChartPollQuestion::TsvLineRegex = %r{
+    \A
+    (?<values>(?:\d+(?:\.\d+)?)?(?:\t(?:\d+(?:\.\d+)?)?)*)\t
+    (?<poll_slug>[^\t]+)\t
+    (?<survey_house>[^\t]+)\t
+    (?<start_date>\d\d\d\d-\d\d-\d\d)\t
+    (?<end_date>\d\d\d\d-\d\d-\d\d)\t
+    (?<question_text>[^\t]*)\t
+    (?<sample_subpopulation>[^\t]+)\t
+    (?<observations>\d*)\t
+    (?<margin_of_error>(?:\d+(?:\.\d+)?)?)\t
+    (?<mode>[^\t]+)\t
+    (?<partisanship>[^\t]+)\t
+    (?<partisan_affiliation>[^\t]+)\b
+    # No \Z ... Pollster's API doesn't guarantee it won't add columns
+  }x
+
+  class PollsterChartPollQuestions
+    HeaderLineRegex = /\A(?<labels>[^\t]+(?:\t[^\t]+)*)\tpoll_slug\tsurvey_house\tstart_date\tend_date\tquestion_text\tsample_subpopulation\tobservations\tmargin_of_error\tmode\tpartisanship\tpartisan_affiliation\b/
+
+    include Enumerable
+
+    def initialize(array)
+      @array = array
+    end
+
+    def each(&block)
+      @array.each(&block)
+    end
+
+    def self.from_tsv(tsv)
+      lines = tsv.split(/(?:\r?\n)+/).reject(&:empty?)
+      if m = HeaderLineRegex.match(lines[0])
+        labels = m[:labels].split(/\t/)
+        rows = lines[1..-1].map { |tsv_line| PollsterChartPollQuestion.from_tsv_line_with_labels(tsv_line, labels) }
+        PollsterChartPollQuestions.new(rows)
+      else
+        raise ArgumentError.new("First line of TSV is `#{lines[0]}`, which does not match Pollster::PollsterChartPollQuestions::HeaderLineRegex")
+      end
+    end
+  end
+end