lib/afr_load.rb in afr_load-0.1.3 vs lib/afr_load.rb in afr_load-0.1.4

@@ -17,21 +17,23 @@
             @url = url
         end
 
         def get_schedule()
             @document = self.class.fetch_schedule(@url)
-            @programs = Parser.parse(@document).flatten
-            @programs
+            get_program()
         end
 
         def self.get_schedule(url = AFR_LOAD_URL)
             afr = self.new(url)
             afr.get_schedule()
         end
 
-        def self.get_schedule_from_file(file_path)
-            handler = File.open(file_path)
-            document = Oga.parse_html(handler)
+        def get_schedule_from_file(file_handler)
+            @document = Oga.parse_html(file_handler)
+        end
+
+        def get_program
+            @programs = Parser.parse(@document).flatten
         end
 
         def self.fetch_schedule(url = AFR_LOAD_URL)
             Oga.parse_html(Enumerator.new do |yielder|
                 HTTPClient.get(url) do |chunk|