lib/sportradar/api/nfl/hierarchy.rb in sportradar-api-0.1.38 vs lib/sportradar/api/nfl/hierarchy.rb in sportradar-api-0.9.0

@@ -6,39 +6,12 @@
       def initialize(data)
         @response = data
         @id = data["id"]
         @name = data["name"]
         @alias = data["alias"]
-        set_conferences
-        set_divisions
-        set_teams
-      end
-
-      private
-
-      def set_conferences
-        if response["conference"]
-          if response["conference"].is_a?(Array)
-            @conferences = response["conference"].map {|conference| Sportradar::Api::Nfl::Conference.new conference }
-          elsif response["conference"].is_a?(Hash)
-            @conferences = [ Sportradar::Api::Nfl::Conference.new(response["conference"]) ]
-          end
-        end
-      end
-
-      def set_divisions
-        if conferences&.all? { |conference| conference.divisions }
-          @divisions = conferences.flat_map(&:divisions)
-        elsif response["division"]
-          if response["division"].is_a?(Array)
-            @divisions = response["division"].map {|division| Sportradar::Api::Nfl::Division.new division }
-          elsif response["division"].is_a?(Hash)
-            @divisions = [ Sportradar::Api::Nfl::Division.new(response["division"]) ]
-          end
-        end
-      end
-
-      def set_teams
+        @conferences = parse_into_array(selector: data["conference"], klass: Sportradar::Api::Nfl::Conference)  if data["conference"]
+        @divisions = conferences.flat_map(&:divisions) if conferences&.all? { |conference| conference.divisions }
+        @divisions ||= parse_into_array(selector: data["division"], klass: Sportradar::Api::Nfl::Division)  if data["division"]
         @teams = @divisions.flat_map(&:teams) if divisions&.all? {|division| division.teams }
       end
 
     end
   end