lib/flickr/photoset.rb in fotonauts-flickr_fu-0.3.4 vs lib/flickr/photoset.rb in fotonauts-flickr_fu-0.3.6

@@ -1,21 +1,25 @@
 class Flickr::Photosets::Photoset
-  attr_accessor :id,:num_photos,:title,:description
-  
+  attr_accessor :id,:num_photos,:title,:description,:primary_photo_id
+
   def initialize(flickr, attributes)
     @flickr = flickr
     attributes.each do |k,v|
       send("#{k}=", v)
     end
   end
-  
+
   def get_photos(options={})
     options = options.merge(:photoset_id=>id)
     rsp = @flickr.send_request('flickr.photosets.getPhotos', options)
     collect_photos(rsp)
   end
-  
+
+   def add_photo(photo_id)
+     rsp = @flickr.send_request('flickr.photosets.addPhoto', {:photo_id=>photo_id, :photoset_id => id})
+   end
+
   protected
     def collect_photos(rsp)
       photos = []
       return photos unless rsp
       if rsp.photoset.photo
@@ -24,14 +28,14 @@
           photos << Flickr::Photos::Photo.new(@flickr,attributes)
         end
       end
       return photos
     end
-    
+
     def create_attributes(photo)
       {:id => photo[:id],
-       :secret => photo[:secret], 
-       :server => photo[:server], 
+       :secret => photo[:secret],
+       :server => photo[:server],
        :farm => photo[:farm],
        :title => photo[:title]}
     end
 end