program-tv 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cdd826e88236e4397312508d2281c44dbfb4ce78
-  data.tar.gz: 4369235cebee6af25943fe94ccd1cb7b4350d387
+  metadata.gz: 11ad4f25b58b9672f4774ca2521d46f01cf81f15
+  data.tar.gz: 6fb5c11984c2096528e30e2c2347226c0df554a9
 SHA512:
-  metadata.gz: 415971534ac63f85bf06855d7b34b537f3d6d0978e210767ec63131c3eb10d740b09671a1baf61e60bfba919d8a3b1759c25e0db2e1ef3c66d65f68f4ceb86e7
-  data.tar.gz: 7bc7c87c47261aad47ab497a83c30c42015fcf818a442e8199d352bd55d585079d59072d7631fc75384725c2875cd63784e5e04733b41a5c37baa81139babcef
+  metadata.gz: 7682ef5601e90399f4a30277b720bfd56d59dcb7aef2f298f6a279323d33d41b41d433218d6bc96539919ec1052c3a728b46171ae49e7bc22996bda20a597caf
+  data.tar.gz: d6d3cb6c68020b8a1f3d8864cca328ad79e8e69e0bd2c19191cb591e5a9b384675ecc141b9d6b3cba9d06b8536da260bc5c2a3072b4c379ac23a28f766e74ced

data/bin/program-tv

@@ -5,7 +5,7 @@ $LOAD_PATH.push File.expand_path("../lib", File.dirname(__FILE__)) # add library
 require 'program-tv'
 require 'optparse'
 
-options   = {}
+options   = { :destination => '/home/hosting/tvo/epg/' }
 OptionParser.new do |op|
   op.banner = <<-eos
   #{ProgramTV::DESCRIPTION}
@@ -27,6 +27,10 @@ OptionParser.new do |op|
   op.on("-c", "--channel-list PATH", "path to yml channel list") do |path|
     options[:channel_list] = path
   end
+
+  op.on("-d", "--destination PATH", "directory with EPG XML files") do |path|
+    options[:destination] = path
+  end
 end.parse!
 
 begin

data/channels.yml

@@ -19,11 +19,11 @@ polsat-biznes : polsatbiznes
 jim-jam-polsat : polsatjimjam
 polsat-sport : polsatsport
 polsat-sport-news : polsatsportnews
-puls : puls
+tv-puls : puls
 puls-2 : puls2
 rbl.tv : rbltv
 russia-today : russiatoday
-stars-tv : starstv
+stars.tv : starstv
 tele-5 : tele5
 ttv : ttv
 tv-4 : tv4

data/lib/program-tv/parser.rb

@@ -10,7 +10,7 @@ module ProgramTV
     def run
       ProgramTV.selected_channels.map do |url_name, epg_name|
         schedule(epg_name, "http://www.cyfrowypolsat.pl/program-tv/#{url_name}")
-      end
+      end.compact
     end
 
     private
@@ -24,6 +24,10 @@ module ProgramTV
           title:   e.css('.name').text
         }
       end
+      if data.empty?
+        puts "Missing schedule for channel #{epg_name}"
+        return
+      end
       data[0..-2].each_with_index{ |element, i| element[:end] = data[i+1][:start] }
       data.pop
       data

data/lib/program-tv/writer.rb

@@ -1,24 +1,46 @@
 require 'builder'
+require 'pry'
+require_relative 'hash_helper'
 
 module ProgramTV
   class Writer
 
-    def initialize(data)
+    def initialize(data, destination)
       @data = data
-      @xml = Builder::XmlMarkup.new( :indent => 2 )
-      @xml.instruct! :xml, :encoding => "UTF-8"
+      @destination = destination
     end
 
-    # Runs dynamic-sprites command.
-    #
     def run
-      @xml.tv do |tv|
-        @data.flatten.each do |element|
+      @data.each do |schedule|
+        today_schedule = []
+        if File.exist?(xml_path(schedule))
+          File.open(xml_path(schedule), "r+") do |file|
+            today_schedule = Hash.from_xml(file.read)[:tv][:programme]
+          end
+        end
+        File.open(xml_path(schedule), "w+") do |file|
+          file.write(build_xml(today_schedule + schedule))
+        end
+      end
+    end
+
+    private
+
+    def xml_path(schedule)
+      File.join(@destination, "#{schedule.first[:channel]}.xml")
+    end
+
+    # Builds XML data from schedule Hash
+    def build_xml(schedule)
+      xml_builder = Builder::XmlMarkup.new( :indent => 2 )
+      xml_builder.instruct! :xml, :encoding => "UTF-8"
+      xml_builder.tv do |tv|
+        schedule.each do |program|
           tv.programme do |p|
-            p.title   element[:title]
-            p.channel element[:channel]
-            p.start   element[:start]
-            p.end     element[:end]
+            p.title   program[:title]
+            p.channel program[:channel]
+            p.start   program[:start]
+            p.end     program[:end]
           end
         end
       end

data/lib/program-tv.rb

@@ -2,7 +2,7 @@ module ProgramTV
 
   #----------------------------------------------------------------------------
 
-  VERSION     = "0.0.3"
+  VERSION     = "0.0.4"
   SUMMARY     = "TV Channel parser"
   DESCRIPTION = "Downloads TV channel list in XML format"
 
@@ -18,7 +18,7 @@ module ProgramTV
   def self.run!(options = {})
     @@options = options
     data = Parser.new.run
-    puts Writer.new(data).run
+    Writer.new(data, @@options[:destination]).run
   end
 
   private

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: program-tv
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Maciej Walusiak