我使用HTTParty进行HTTP请求和使用RESTAPI。现在我想重新使用我通过POST调用的登录页面设置的cookie。
class SomeImporter
include HTTParty
def self.login
response = self.post('https://www.example.com/login', :query => {:user => 'myusername', :password => 'secret'})
self.default_cookies.add_cookies(response.header['set-cookie'])
self.get('https://www.example.com/protected')
end
end
Cookie未使用此代码正确设置。如何正确解析HTTParty给出的"设置cookie"头,并为下一个请求设置cookie?
通常HTTP头中的每个Set-Cookie
都有一个条目。HTTParty将它们合并为一个单独的字符串,作为逗号分隔的列表。但HTTParty在将它们添加回默认cookie时不会自行拆分。你必须自己解析它们。
可以使用以下方法解析"set cookie"。将其添加到您的课堂:
# Parse the 'set-cookie' string
# @param [String] all_cookies_string
# @return [Hash]
def self.parse_set_cookie(all_cookies_string)
cookies = Hash.new
if all_cookies_string.present?
# single cookies are devided with comma
all_cookies_string.split(',').each {
# @type [String] cookie_string
|single_cookie_string|
# parts of single cookie are seperated by semicolon; first part is key and value of this cookie
# @type [String]
cookie_part_string = single_cookie_string.strip.split(';')[0]
# remove whitespaces at beginning and end in place and split at '='
# @type [Array]
cookie_part = cookie_part_string.strip.split('=')
# @type [String]
key = cookie_part[0]
# @type [String]
value = cookie_part[1]
# add cookie to Hash
cookies[key] = value
}
end
cookies
end
通过调整此行,可以将cookie添加到HTTParty,以满足以下请求:
self.default_cookies.add_cookies(self.parse_set_cookie(response.header['set-cookie']))
在self.parse_set_cookie
中,仅提取具有名称和值的cookie。您可以对其进行扩展以获得更多详细信息,如Path
、Domain
等。更多详细信息请参阅RFC 2109(4.2.2设置Cookie语法)。
def cookies_to_hash(acookies)
vs = acookies.split(',')
ak = Array.new
# Avoiding break in expiry date comma
vs.each_cons(2) do |v, vn|
unless(vn.include?("="))
ak << "#{v}#{vn}"
else
ak << v if(v.include?("="))
end
end
# Cookies as array of hashes
all_cookies = []
ak.each do |each_cookie|
ecookie = []
each_cookie.split(';').each do |cookie_attr|
ecookie << cookie_attr.split("=")
end
all_cookies << Hash[ecookie]
end
all_cookies
end
上述方法将帮助您将响应标头中的设置cookie转换为有效的哈希数组。处理了过期时添加逗号的问题。
多亏了@dinespapu,我能够按照他的模式添加一些修改来处理过期日期。希望这个帮助:
def cookies_to_hash(set_cookie)
set_cookie = set_cookie.gsub(/Expires=[a-zA-Z]+,/) do |m|
# replace , with #####
m[0..-2] + "#####"
end
set_cookie = set_cookie.split(',')
# Cookies as array of hashes
all_cookies = []
set_cookie.each do |each_cookie|
ecookie = []
each_cookie.strip.split(';').each do |cookie_attr|
# replace ##### with , back
ecookie << cookie_attr.split("=").map{|e| e.gsub(/#####/, ',') if e}
end
all_cookies << Hash[ecookie]
end
all_cookies
end