Skip to content

Commit

Permalink
Automatically flatten Twitter::Arguments
Browse files Browse the repository at this point in the history
  • Loading branch information
sferik committed Dec 18, 2013
1 parent ca21c5b commit a556028
Show file tree
Hide file tree
Showing 9 changed files with 19 additions and 18 deletions.
2 changes: 1 addition & 1 deletion lib/twitter/arguments.rb
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ class Arguments < Array
# @return [Twitter::Arguments]
def initialize(args)
@options = args.last.is_a?(::Hash) ? args.pop : {}
super(args)
super(args.flatten)
end
end
end
2 changes: 1 addition & 1 deletion lib/twitter/rest/api/direct_messages.rb
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ def direct_messages(*args)
if arguments.empty?
direct_messages_received(arguments.options)
else
Twitter::Utils.parallel_map(arguments.flatten) do |id|
Twitter::Utils.parallel_map(arguments) do |id|
direct_message(id, arguments.options)
end
end
Expand Down
4 changes: 2 additions & 2 deletions lib/twitter/rest/api/favorites.rb
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def unfavorite(*args)
# @param options [Hash] A customizable set of options.
def favorite(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
begin
object_from_response(Twitter::Tweet, :post, '/1.1/favorites/create.json', arguments.options.merge(:id => id))
Expand Down Expand Up @@ -96,7 +96,7 @@ def favorite(*args)
# @param options [Hash] A customizable set of options.
def favorite!(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
begin
object_from_response(Twitter::Tweet, :post, '/1.1/favorites/create.json', arguments.options.merge(:id => id))
Expand Down
2 changes: 1 addition & 1 deletion lib/twitter/rest/api/friends_and_followers.rb
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,7 @@ def follow(*args)
# @option options [Boolean] :follow (false) Enable notifications for the target user.
def follow!(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |user|
Twitter::Utils.parallel_map(arguments) do |user|
object_from_response(Twitter::User, :post, '/1.1/friendships/create.json', merge_user(arguments.options, user))
end.compact
end
Expand Down
9 changes: 5 additions & 4 deletions lib/twitter/rest/api/lists.rb
Original file line number Diff line number Diff line change
Expand Up @@ -441,12 +441,13 @@ def list_from_response_with_user(request_method, path, args)
end

def list_from_response_with_users(request_method, path, args)
arguments = Twitter::Arguments.new(args)
arguments = args.dup
options = arguments.last.is_a?(::Hash) ? arguments.pop : {}
members = arguments.pop
merge_list!(arguments.options, arguments.pop)
merge_owner!(arguments.options, arguments.pop)
merge_list!(options, arguments.pop)
merge_owner!(options, arguments.pop)
Twitter::Utils.parallel_map(members.flatten.each_slice(MAX_USERS_PER_REQUEST)) do |users|
object_from_response(Twitter::List, request_method, path, merge_users(arguments.options, users))
object_from_response(Twitter::List, request_method, path, merge_users(options, users))
end.last
end

Expand Down
4 changes: 2 additions & 2 deletions lib/twitter/rest/api/saved_searches.rb
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def saved_searches(*args)
if arguments.empty?
objects_from_response(Twitter::SavedSearch, :get, '/1.1/saved_searches/list.json', arguments.options)
else
Twitter::Utils.parallel_map(arguments.flatten) do |id|
Twitter::Utils.parallel_map(arguments) do |id|
saved_search(id, arguments.options)
end
end
Expand Down Expand Up @@ -82,7 +82,7 @@ def create_saved_search(query, options = {})
# @param options [Hash] A customizable set of options.
def destroy_saved_search(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |id|
Twitter::Utils.parallel_map(arguments) do |id|
object_from_response(Twitter::SavedSearch, :post, "/1.1/saved_searches/destroy/#{id}.json", arguments.options)
end
end
Expand Down
8 changes: 4 additions & 4 deletions lib/twitter/rest/api/tweets.rb
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ def update(status, options = {})
# @option options [Boolean, String, Integer] :trim_user Each tweet returned in a timeline will include a user object with only the author's numerical ID when set to true, 't' or 1.
def retweet(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
begin
post_retweet(id, arguments.options)
Expand All @@ -169,7 +169,7 @@ def retweet(*args)
# @option options [Boolean, String, Integer] :trim_user Each tweet returned in a timeline will include a user object with only the author's numerical ID when set to true, 't' or 1.
def retweet!(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
begin
post_retweet(id, arguments.options)
Expand Down Expand Up @@ -250,7 +250,7 @@ def oembed(tweet, options = {})
# @option options [String] :lang Language code for the rendered embed. This will affect the text and localization of the rendered HTML.
def oembeds(*args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
oembed(id, arguments.options)
end
Expand Down Expand Up @@ -282,7 +282,7 @@ def retweeters_ids(*args)
# @return [Array<Twitter::Tweet>]
def threaded_tweets_from_response(request_method, path, args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |tweet|
Twitter::Utils.parallel_map(arguments) do |tweet|
id = extract_id(tweet)
object_from_response(Twitter::Tweet, request_method, path + "/#{id}.json", arguments.options)
end
Expand Down
2 changes: 1 addition & 1 deletion lib/twitter/rest/api/users.rb
Original file line number Diff line number Diff line change
Expand Up @@ -227,7 +227,7 @@ def unblock(*args)
def users(*args)
arguments = Twitter::Arguments.new(args)
method = arguments.options.delete(:method) || :post
Twitter::Utils.parallel_map(arguments.flatten.each_slice(MAX_USERS_PER_REQUEST)) do |users|
Twitter::Utils.parallel_map(arguments.each_slice(MAX_USERS_PER_REQUEST)) do |users|
objects_from_response(Twitter::User, method, '/1.1/users/lookup.json', merge_users(arguments.options, users))
end.flatten
end
Expand Down
4 changes: 2 additions & 2 deletions lib/twitter/rest/api/utils.rb
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def extract_id(object)
# @return [Array<Twitter::User>]
def threaded_user_objects_from_response(request_method, path, args)
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |user|
Twitter::Utils.parallel_map(arguments) do |user|
object_from_response(Twitter::User, request_method, path, merge_user(arguments.options, user))
end
end
Expand Down Expand Up @@ -103,7 +103,7 @@ def objects_from_array(klass, array)
# @return [Array]
def threaded_objects_from_response(klass, request_method, path, args) # rubocop:disable ParameterLists
arguments = Twitter::Arguments.new(args)
Twitter::Utils.parallel_map(arguments.flatten) do |object|
Twitter::Utils.parallel_map(arguments) do |object|
id = extract_id(object)
object_from_response(klass, request_method, path, arguments.options.merge(:id => id))
end
Expand Down

0 comments on commit a556028

Please sign in to comment.