tryphon-pige-client 0.0.3 → 1.0

data/.gitignore

@@ -1,18 +1,19 @@
 *.gem
 *.rbc
-.bundle
-.config
-.yardoc
-Gemfile.lock
-InstalledFiles
-_yardoc
-coverage
-doc/
-lib/bundler/man
-pkg
-rdoc
-spec/reports
-test/tmp
-test/version_tmp
-tmp
+/.bundle
+/.config
+/.yardoc
+/Gemfile.lock
+/InstalledFiles
+/_yardoc
+/coverage
+/doc/
+/lib/bundler/man
+/pkg
+/rdoc
+/spec/reports
+/test/tmp
+/test/version_tmp
+/tmp
 *~
+/.ruby-version

data/examples/scheduler_chunks_daily.rb

@@ -2,7 +2,7 @@
 
 require 'tryphon-pige-client'
 
-Pige::Client.box_url = "http://pigebox.local"
+Pige::Client.box_url = "http://#{ENV.fetch('PIGEBOX_HOST', 'pigebox.local')}"
 include Pige::Client
 
 ChunkScheduler.define do |schedule|
@@ -10,16 +10,16 @@ ChunkScheduler.define do |schedule|
   day_back = ARGV.first ? ARGV.first.to_i : 1
   schedule.day = Date.today - day_back
 
-  # Doesn't really Chunks, only log orders
-  schedule.dry_run = true
+# Doesn't really Chunks, only log orders
+  schedule.dry_run = ENV.fetch('DRY_RUN',"true") != "false"
 
   # Default Attributes
   # can be overridden in each create
   schedule.default_attributes[:margin] = 5.minutes
-  #schedule.default_attributes[:format] = "ogg"
+  # schedule.default_attributes[:format] = "vorbis"
 
   # Daily Chunks
-  schedule.create :title => "Flash 13h %d_%m_%Y", :begin => "13:00", :duration => 15.minutes, :margin => 2.minutes
+  schedule.create :title => "Flash 13h %d_%m_%Y", :begin => "13:00", :begin_label => "Generique Flash", :duration => 15.minutes, :end_label => "Fin Direct", :margin => 0
 
   # Only monday Chunks
   schedule.on(:monday) do
@@ -51,7 +51,7 @@ ChunkScheduler.define do |schedule|
   schedule.on(:sunday) do
     # ...
   end
-end
+end.download("/tmp")
 
 # Remove oldest Chunks
 #Pige::Client::Chunk.destroy_all(:older_than => 1.days.ago)

data/lib/pige/client/chunk.rb

@@ -4,5 +4,28 @@ module Pige::Client
     attr_accessor :begin, :completion_rate, :end, :format, :title
     time_attribute :begin, :end
 
+    def completion_rate
+      self.class.get("#{Pige::Client.box_url}/sources/1/chunks/#{id}.json")["completion_rate"].to_f
+    end
+
+    def extension
+      format == "vorbis" ? "ogg" : format
+    end
+
+    def download_url
+      "#{Pige::Client.box_url}/sources/1/chunks/#{id}.#{extension}"
+    end
+
+    def download_file
+      "#{title}.#{extension}"
+    end
+
+    def download(target)
+      File.open(File.join(target, download_file), "w") do |file|
+        open(download_url) do |url|
+          IO.copy_stream url, file
+        end
+      end
+    end
   end
 end

data/lib/pige/client/chunk_scheduler.rb

@@ -4,13 +4,50 @@ module Pige::Client
     attr_accessor :day, :default_attributes, :dry_run
     alias_method :dry_run?, :dry_run
 
+    def chunks
+      @chunks ||= []
+    end
+
+    def wait(timeout = nil)
+      timeout ||= chunks.size * 5
+      print "Wait for new chunks to be ready : "
+      deadline = Time.now + (timeout * 60)
+
+      uncompleted_chunks = chunks.dup
+
+      while Time.now < deadline
+        uncompleted_chunks.delete_if do |chunk|
+          chunk.completion_rate == 1
+        end
+        if uncompleted_chunks.empty?
+          puts "done\nChunks are ready"
+          break
+        else
+          print "."
+          sleep 10
+        end
+      end
+    end
+
+    def download(target)
+      wait
+
+      FileUtils.mkdir_p target
+      chunks.each do |chunk|
+        puts "Download '#{chunk.title}' to #{target}/#{chunk.download_file} ..."
+        chunk.download target
+      end
+    end
+
     def initialize(day = Date.today)
       self.day = day
       self.default_attributes = { :format => "wav" }
     end
 
     def self.define(&block)
-      yield new
+      new.tap do |scheduler|
+        yield scheduler
+      end
     end
 
     def on(criteria, &block)
@@ -20,8 +57,7 @@ module Pige::Client
       end
     end
 
-    def create(attributes = {}, options = {})
-      options = { :delay => 1 }.merge(options)
+    def chunk_attributes(attributes = {})
       attributes = default_attributes.merge(attributes)
 
       [:begin, :end].each do |endpoint|
@@ -36,6 +72,26 @@ module Pige::Client
         end
       end
 
+      if name = attributes[:begin_label]
+        label = Pige::Client::Label.by_name(name, after: attributes[:begin])
+        if label
+          puts "Found label '#{name}' at #{label.timestamp}"
+          attributes[:begin] = label.timestamp
+        else
+          puts "Can't find' label '#{name}'"
+        end
+      end
+
+      if name = attributes[:end_label]
+        label = Pige::Client::Label.by_name(name, before: attributes[:end])
+        if label
+          puts "Found label '#{name}' at #{label.timestamp}"
+          attributes[:end] = label.timestamp
+        else
+          puts "Can't find' label '#{name}'"
+        end
+      end
+
       if margin = attributes.delete(:margin)
         attributes[:begin] -= margin
         attributes[:end] += margin
@@ -45,9 +101,17 @@ module Pige::Client
         attributes[:title] = day.strftime(attributes[:title])
       end
 
+      attributes
+    end
+
+    def create(attributes = {}, options = {})
+      options = { :delay => 1 }.merge(options)
+
+      attributes = chunk_attributes(attributes)
       puts "Create Chunk with #{attributes.inspect}"
+
       unless dry_run?
-        Pige::Client::Chunk.create attributes 
+        chunks << Pige::Client::Chunk.create(attributes)
         sleep options[:delay].to_i if options[:delay]
       end
     end

data/lib/pige/client/label.rb

@@ -7,25 +7,40 @@ module Pige::Client
       self.timestamp < date
     end
 
-    def self.all
+    def self.all(options = {})
       [].tap do |labels|
         page = 1
         empty_page = false
 
         begin
-          labels_in_page = page(page)
+          labels_in_page = page(page, options)
           unless labels_in_page.empty?
             labels.push *labels_in_page
             page += 1
           else
             empty_page = true
           end
-        end until empty_page 
+        end until empty_page
       end
     end
 
-    def self.page(number)
-      get("#{Pige::Client.box_url}/sources/1/labels.json?page=#{number}").map do |attributes|
+    def self.by_name(name, options = {})
+      options = options.merge(term: name)
+      attributes = get("#{Pige::Client.box_url}/sources/1/labels.json", query: options).first
+      Label.new attributes if attributes
+    end
+
+    def self.older_than(date)
+      all(before: date)
+    end
+
+    def self.page(number, options = {})
+      additionnal_query = options.map do |k,v|
+        "#{k}=#{URI.escape(v.to_s)}" if v
+      end.join('&')
+      additionnal_query = "&#{additionnal_query}" unless additionnal_query.empty?
+
+      get("#{Pige::Client.box_url}/sources/1/labels.json?page=#{number}#{additionnal_query}").map do |attributes|
         Label.new attributes["label"]
       end
     end

data/lib/pige/client/resource.rb

@@ -2,7 +2,7 @@ module Pige::Client
   class Resource
     include HTTParty
 
-    # debug_output $stderr
+    debug_output $stderr
     format :json
 
     def initialize(attributes = {})
@@ -21,6 +21,11 @@ module Pige::Client
     attr_accessor :created_at, :updated_at
     time_attribute :created_at, :updated_at
 
+    attr_accessor :errors
+    def errors
+      @errors ||= []
+    end
+
     attr_accessor :id, :source_id
 
     def older_than?(date)
@@ -35,17 +40,22 @@ module Pige::Client
       name.downcase.split("::").last
     end
 
+    def self.resources_name
+      "#{resource_name}s"
+    end
+
     def self.base_url
       "#{Pige::Client.box_url}/sources/1/#{resource_name}s"
     end
 
     def self.create(attributes = {})
-      post "#{base_url}.json", :query => { resource_name => attributes }
+      response = post("#{base_url}.json", :query => { resource_name => attributes })
+      new response
     end
 
     def self.all
-      get("#{Pige::Client.box_url}/sources/1/chunks.json").map do |attributes|
-        Chunk.new attributes["chunk"]
+      get("#{Pige::Client.box_url}/sources/1/#{resources_name}.json").map do |attributes|
+        new attributes[resource_name]
       end
     end
 
@@ -56,7 +66,7 @@ module Pige::Client
     end
 
     def self.destroy_all(conditions = {})
-      resources = 
+      resources =
         if older_than = conditions[:older_than]
           older_than(older_than)
         else

data/lib/pige/client/version.rb

@@ -1,6 +1,5 @@
 module Pige
   module Client
-    VERSION = "0.0.3"
+    VERSION = "1.0"
   end
 end
-

data/lib/tryphon-pige-client.rb

@@ -1,4 +1,5 @@
 require 'httparty'
+require 'open-uri'
 
 require "pige/client"
 require "pige/client/version"

data/spec/pige/client/chunk_scheduler_spec.rb

@@ -0,0 +1,99 @@
+require 'spec_helper'
+
+describe ChunkScheduler do
+
+  describe "#day" do
+
+    it "should use today by default" do
+      subject.day.should == Date.today
+    end
+
+  end
+
+  describe "#dry_run" do
+
+    it "should be false by default" do
+      subject.dry_run.should be_false
+    end
+
+  end
+
+  describe "#default_attributes" do
+
+    it "should define format to wav" do
+      subject.default_attributes[:format].should == "wav"
+    end
+
+  end
+
+  describe "#on" do
+
+    it "should yield block if day matchs criteria" do
+      subject.day.stub :dummy? => true
+      has_been_yield = false
+      subject.on(:dummy) { has_been_yield = true }
+    end
+
+  end
+
+  describe "#chunk_attributes" do
+
+    context "when begin_label is specified" do
+
+      let(:time) { Time.now }
+      let(:label) { double(timestamp: Time.now) }
+
+      it "should use label timestamp if found" do
+        Pige::Client::Label.should_receive(:by_name).with("dummy", after: nil).and_return(label)
+
+        attributes = subject.chunk_attributes begin_label: "dummy"
+        attributes[:begin].should == label.timestamp
+      end
+
+      it "should search label after begin time" do
+        Pige::Client::Label.should_receive(:by_name).with("dummy", after: time).and_return(label)
+
+        attributes = subject.chunk_attributes begin_label: "dummy", begin: time
+        attributes[:begin].should == label.timestamp
+      end
+
+      it "should use begin time if label is not found" do
+        Pige::Client::Label.stub by_name: nil
+
+        attributes = subject.chunk_attributes begin: time
+        attributes[:begin].should == time
+      end
+
+    end
+
+    context "when end_label is specified" do
+
+      let(:time) { Time.now }
+      let(:label) { double(timestamp: Time.now) }
+
+      it "should use label timestamp if found" do
+        Pige::Client::Label.should_receive(:by_name).with("dummy", before: nil).and_return(label)
+
+        attributes = subject.chunk_attributes end_label: "dummy"
+        attributes[:end].should == label.timestamp
+      end
+
+      it "should search label before end time" do
+        Pige::Client::Label.should_receive(:by_name).with("dummy", before: time).and_return(label)
+
+        attributes = subject.chunk_attributes end_label: "dummy", end: time
+        attributes[:end].should == label.timestamp
+      end
+
+      it "should use end time if label is not found" do
+        Pige::Client::Label.stub by_name: nil
+
+        attributes = subject.chunk_attributes end: time
+        attributes[:end].should == time
+      end
+
+    end
+
+  end
+
+end

data/spec/pige/client/chunk_spec.rb

@@ -2,6 +2,14 @@ require 'spec_helper'
 
 describe Chunk do
 
+  describe ".resources_name" do
+
+    it "should be chunks" do
+      Chunk.resources_name.should == "chunks"
+    end
+
+  end
+
   describe ".all" do
 
     before do
@@ -15,13 +23,13 @@ describe Chunk do
     it "should create Chunks with received data" do
       Chunk.all.first.title.should == "Emission Cinema lundi 03_12_2012"
     end
-    
+
   end
 
   describe ".destroy_all" do
 
     let(:chunk) { mock }
-    
+
     context "without conditions" do
 
       it "should destroy all Chunks returned by Chunk.all" do
@@ -41,7 +49,7 @@ describe Chunk do
         chunk.should_receive(:destroy)
         Chunk.destroy_all :older_than => date
       end
-                                          
+
     end
 
   end
@@ -60,7 +68,7 @@ describe Chunk do
       chunk.created_at = date + 1
       chunk.should_not be_older_than(date)
     end
-    
+
   end
 
   describe ".older_than" do
@@ -71,7 +79,7 @@ describe Chunk do
     before do
       Chunk.stub :all => [chunk]
     end
-    
+
     it "should select Chunks older than specified date" do
       chunk.should_receive(:older_than?).with(date).and_return(true)
       Chunk.older_than(date).should include(chunk)
@@ -94,8 +102,5 @@ describe Chunk do
     end
 
   end
-  
-end
-
-
 
+end

data/spec/pige/client/label_spec.rb

@@ -15,7 +15,7 @@ describe Label do
     it "should create Labels with received data" do
       Label.page(2).first.name.should == "Test"
     end
-    
+
   end
 
   describe ".all" do
@@ -23,9 +23,9 @@ describe Label do
     let(:label) { mock }
 
     it "should concat pages until an empty one" do
-      Label.should_receive(:page).with(1).and_return([label])
-      Label.should_receive(:page).with(2).and_return([label])
-      Label.should_receive(:page).with(3).and_return([])
+      Label.should_receive(:page).with(1, {}).and_return([label])
+      Label.should_receive(:page).with(2, {}).and_return([label])
+      Label.should_receive(:page).with(3, {}).and_return([])
 
       Label.all.should == [label, label]
     end
@@ -35,7 +35,7 @@ describe Label do
   describe ".destroy_all" do
 
     let(:label) { mock }
-    
+
     context "without conditions" do
 
       it "should destroy all Labels returned by Label.all" do
@@ -55,7 +55,7 @@ describe Label do
         label.should_receive(:destroy)
         Label.destroy_all :older_than => date
       end
-                                          
+
     end
 
   end
@@ -74,26 +74,17 @@ describe Label do
       label.timestamp = date + 1
       label.should_not be_older_than(date)
     end
-    
+
   end
 
   describe ".older_than" do
 
-    let(:label) { mock }
     let(:date) { Time.now }
 
-    before do
-      Label.stub :all => [label]
-    end
-    
     it "should select Labels older than specified date" do
-      label.should_receive(:older_than?).with(date).and_return(true)
-      Label.older_than(date).should include(label)
-    end
-
-    it "should ignore Labels not older than specified date" do
-      label.should_receive(:older_than?).with(date).and_return(false)
-      Label.older_than(date).should_not include(label)
+      FakeWeb.register_uri(:get, "http://pigebox.local/sources/1/labels.json?page=1&before=#{URI.escape(date.to_s)}", :body => fixture_content("labels.json"))
+      FakeWeb.register_uri(:get, "http://pigebox.local/sources/1/labels.json?page=2&before=#{URI.escape(date.to_s)}", :body => [].to_json)
+      Label.older_than(date).first.name.should == "Test"
     end
 
   end
@@ -108,8 +99,5 @@ describe Label do
     end
 
   end
-  
-end
-
-
 
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tryphon-pige-client
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: '1.0'
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-08 00:00:00.000000000 Z
+date: 2015-01-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fakeweb
@@ -150,6 +150,7 @@ files:
 - lib/tryphon-pige-client.rb
 - spec/fixtures/chunks.json
 - spec/fixtures/labels.json
+- spec/pige/client/chunk_scheduler_spec.rb
 - spec/pige/client/chunk_spec.rb
 - spec/pige/client/label_spec.rb
 - spec/spec_helper.rb
@@ -173,7 +174,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3105553272469934144
+      hash: -3329681472246691501
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -182,7 +183,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3105553272469934144
+      hash: -3329681472246691501
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.23
@@ -192,6 +193,7 @@ summary: Tryphon PigeBox API ruby client
 test_files:
 - spec/fixtures/chunks.json
 - spec/fixtures/labels.json
+- spec/pige/client/chunk_scheduler_spec.rb
 - spec/pige/client/chunk_spec.rb
 - spec/pige/client/label_spec.rb
 - spec/spec_helper.rb