def testCorrectConversionToElementTree(self):
test_feed = gdata.GDataFeedFromString(test_data.GBASE_FEED)
self.assert_(test_feed.total_results is not None)
element_tree = test_feed._ToElementTree()
feed = element_tree.find('{http://www.w3.org/2005/Atom}feed')
self.assert_(element_tree.find(
'{http://a9.com/-/spec/opensearchrss/1.0/}totalResults') is not None)
def testGeneratorShouldBeCleaned(self):
feed = gdata.GDataFeedFromString(test_data.GBASE_FEED)
element_tree = ElementTree.fromstring(test_data.GBASE_FEED)
self.assert_(
element_tree.findall('{http://www.w3.org/2005/Atom}generator')
[0].text != feed.generator.text)
self.assert_(feed.generator.text == 'GoogleBase')
def reload(self, id):
node = self._get_row(model.Album, id)
event_node = node.event
self._check_access(event_node)
album = gdata.GDataFeedFromString(node.settings)
node.settings = urlopen(album.GetSelfLink().href).read()
meta.Session.commit()
flash(u'Альбом успешно обновлен')
if self.back_page():
return redirect_to(**self.back_page())
return redirect_to(event_node.url())
def photos_by_album(album, photos_list=[], limit=None):
photos = gdata.GDataFeedFromString(album.settings)
photos = photos.entry[0:limit]
result = []
if photos_list:
for photo in photos:
if photo.gphoto_id.text in photos_list:
result.append(photo_prepare(photo))
else:
for photo in photos:
group = photo.FindExtensions('group')
keywords = group[0].FindChildren('keywords')[0].text
if keywords:
keywords = keywords.split(', ')
else:
keywords = ()
if u'hide' not in keywords:
result.append(photo_prepare(photo))
return result
def Put(self,
data,
uri,
extra_headers=None,
url_params=None,
escape_params=True,
redirects_remaining=3,
media_source=None,
converter=None):
"""Updates an entry at the given URI.
Args:
data: string, ElementTree._Element, or xml_wrapper.ElementWrapper The
XML containing the updated data.
uri: string A URI indicating entry to which the update will be applied.
Example: '/base/feeds/items/ITEM-ID'
extra_headers: dict (optional) HTTP headers which are to be included.
The client automatically sets the Content-Type,
Authorization, and Content-Length headers.
url_params: dict (optional) Additional URL parameters to be included
in the URI. These are translated into query arguments
in the form '&dict_key=value&...'.
Example: {'max-results': '250'} becomes &max-results=250
escape_params: boolean (optional) If false, the calling code has already
ensured that the query will form a valid URL (all
reserved characters have been escaped). If true, this
method will escape the query and any URL parameters
provided.
converter: func (optional) A function which will be executed on the
server's response. Often this is a function like
GDataEntryFromString which will parse the body of the server's
response and return a GDataEntry.
Returns:
If the put succeeded, this method will return a GDataFeed, GDataEntry,
or the results of running converter on the server's result body (if
converter was specified).
"""
if extra_headers is None:
extra_headers = {}
# Add the authentication header to the Get request
if self.__auth_token:
extra_headers['Authorization'] = self.__auth_token
if self.__gsessionid is not None:
if uri.find('gsessionid=') < 0:
if uri.find('?') > -1:
uri += '&gsessionid=%s' % (self.__gsessionid, )
else:
uri += '?gsessionid=%s' % (self.__gsessionid, )
if media_source and data:
if ElementTree.iselement(data):
data_str = ElementTree.tostring(data)
else:
data_str = str(data)
multipart = []
multipart.append('Media multipart posting\r\n--END_OF_PART\r\n' + \
'Content-Type: application/atom+xml\r\n\r\n')
multipart.append('\r\n--END_OF_PART\r\nContent-Type: ' + \
media_source.content_type+'\r\n\r\n')
multipart.append('\r\n--END_OF_PART--\r\n')
extra_headers[
'Content-Type'] = 'multipart/related; boundary=END_OF_PART'
extra_headers['MIME-version'] = '1.0'
extra_headers['Content-Length'] = str(
len(multipart[0]) + len(multipart[1]) + len(multipart[2]) +
len(data_str) + media_source.content_length)
insert_connection = atom.service.AtomService._CreateConnection(
self, uri, 'PUT', extra_headers, url_params, escape_params)
insert_connection.send(multipart[0])
insert_connection.send(data_str)
insert_connection.send(multipart[1])
while 1:
binarydata = media_source.file_handle.read(100000)
if (binarydata == ""): break
insert_connection.send(binarydata)
insert_connection.send(multipart[2])
server_response = insert_connection.getresponse()
result_body = server_response.read()
elif media_source:
extra_headers['Content-Type'] = media_source.content_type
extra_headers['Content-Length'] = media_source.content_length
insert_connection = atom.service.AtomService._CreateConnection(
self, uri, 'PUT', extra_headers, url_params, escape_params)
while 1:
binarydata = media_source.file_handle.read(100000)
if (binarydata == ""): break
insert_connection.send(binarydata)
server_response = insert_connection.getresponse()
result_body = server_response.read()
else:
http_data = data
content_type = 'application/atom+xml'
server_response = atom.service.AtomService.Put(
self, http_data, uri, extra_headers, url_params, escape_params,
content_type)
result_body = server_response.read()
if server_response.status == 200:
if converter:
return converter(result_body)
feed = gdata.GDataFeedFromString(result_body)
if not feed:
entry = gdata.GDataEntryFromString(result_body)
if not entry:
return result_body
return entry
return feed
elif server_response.status == 302:
if redirects_remaining > 0:
location = server_response.getheader('Location')
if location is not None:
m = re.compile('[\?\&]gsessionid=(\w*)').search(location)
if m is not None:
self.__gsessionid = m.group(1)
return self.Put(data,
location,
extra_headers,
url_params,
escape_params,
redirects_remaining - 1,
media_source=media_source,
converter=converter)
else:
raise RequestError, {
'status': server_response.status,
'reason': '302 received without Location header',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason':
'Redirect received, but redirects_remaining <= 0',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason': server_response.reason,
'body': result_body
}
def Get(self,
uri,
extra_headers=None,
redirects_remaining=4,
encoding='UTF-8',
converter=None):
"""Query the GData API with the given URI
The uri is the portion of the URI after the server value
(ex: www.google.com).
To perform a query against Google Base, set the server to
'base.google.com' and set the uri to '/base/feeds/...', where ... is
your query. For example, to find snippets for all digital cameras uri
should be set to: '/base/feeds/snippets?bq=digital+camera'
Args:
uri: string The query in the form of a URI. Example:
'/base/feeds/snippets?bq=digital+camera'.
extra_headers: dictionary (optional) Extra HTTP headers to be included
in the GET request. These headers are in addition to
those stored in the client's additional_headers property.
The client automatically sets the Content-Type and
Authorization headers.
redirects_remaining: int (optional) Tracks the number of additional
redirects this method will allow. If the service object receives
a redirect and remaining is 0, it will not follow the redirect.
This was added to avoid infinite redirect loops.
encoding: string (optional) The character encoding for the server's
response. Default is UTF-8
converter: func (optional) A function which will transform
the server's results before it is returned. Example: use
GDataFeedFromString to parse the server response as if it
were a GDataFeed.
Returns:
If there is no ResultsTransformer specified in the call, a GDataFeed
or GDataEntry depending on which is sent from the server. If the
response is niether a feed or entry and there is no ResultsTransformer,
return a string. If there is a ResultsTransformer, the returned value
will be that of the ResultsTransformer function.
"""
if extra_headers is None:
extra_headers = {}
# Add the authentication header to the Get request
if self.__auth_token:
extra_headers['Authorization'] = self.__auth_token
if self.__gsessionid is not None:
if uri.find('gsessionid=') < 0:
if uri.find('?') > -1:
uri += '&gsessionid=%s' % (self.__gsessionid, )
else:
uri += '?gsessionid=%s' % (self.__gsessionid, )
server_response = atom.service.AtomService.Get(self, uri,
extra_headers)
result_body = server_response.read()
if server_response.status == 200:
if converter:
return converter(result_body)
# There was no ResultsTransformer specified, so try to convert the
# server's response into a GDataFeed.
feed = gdata.GDataFeedFromString(result_body)
if not feed:
# If conversion to a GDataFeed failed, try to convert the server's
# response to a GDataEntry.
entry = gdata.GDataEntryFromString(result_body)
if not entry:
# The server's response wasn't a feed, or an entry, so return the
# response body as a string.
return result_body
return entry
return feed
elif server_response.status == 302:
if redirects_remaining > 0:
location = server_response.getheader('Location')
if location is not None:
m = re.compile('[\?\&]gsessionid=(\w*)').search(location)
if m is not None:
self.__gsessionid = m.group(1)
return self.Get(location,
extra_headers,
redirects_remaining - 1,
encoding=encoding,
converter=converter)
else:
raise RequestError, {
'status': server_response.status,
'reason': '302 received without Location header',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason':
'Redirect received, but redirects_remaining <= 0',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason': server_response.reason,
'body': result_body
}
def url_to_picasa(self):
photos = gdata.GDataFeedFromString(self.settings)
return photos.GetAlternateLink().href
def PostOrPut(self,
verb,
data,
uri,
extra_headers=None,
url_params=None,
escape_params=True,
redirects_remaining=4,
media_source=None,
converter=None):
"""Insert data into a GData service at the given URI.
Args:
verb: string, either 'POST' or 'PUT'
data: string, ElementTree._Element, atom.Entry, or gdata.GDataEntry The
XML to be sent to the uri.
uri: string The location (feed) to which the data should be inserted.
Example: '/base/feeds/items'.
extra_headers: dict (optional) HTTP headers which are to be included.
The client automatically sets the Content-Type,
Authorization, and Content-Length headers.
url_params: dict (optional) Additional URL parameters to be included
in the URI. These are translated into query arguments
in the form '&dict_key=value&...'.
Example: {'max-results': '250'} becomes &max-results=250
escape_params: boolean (optional) If false, the calling code has already
ensured that the query will form a valid URL (all
reserved characters have been escaped). If true, this
method will escape the query and any URL parameters
provided.
media_source: MediaSource (optional) Container for the media to be sent
along with the entry, if provided.
converter: func (optional) A function which will be executed on the
server's response. Often this is a function like
GDataEntryFromString which will parse the body of the server's
response and return a GDataEntry.
Returns:
If the post succeeded, this method will return a GDataFeed, GDataEntry,
or the results of running converter on the server's result body (if
converter was specified).
"""
if extra_headers is None:
extra_headers = {}
# Add the authentication header to the Get request
if self.__auth_token:
extra_headers['Authorization'] = self.__auth_token
if self.__gsessionid is not None:
if uri.find('gsessionid=') < 0:
if uri.find('?') > -1:
uri += '&gsessionid=%s' % (self.__gsessionid, )
else:
uri += '?gsessionid=%s' % (self.__gsessionid, )
if data and media_source:
if ElementTree.iselement(data):
data_str = ElementTree.tostring(data)
else:
data_str = str(data)
multipart = []
multipart.append('Media multipart posting\r\n--END_OF_PART\r\n' + \
'Content-Type: application/atom+xml\r\n\r\n')
multipart.append('\r\n--END_OF_PART\r\nContent-Type: ' + \
media_source.content_type+'\r\n\r\n')
multipart.append('\r\n--END_OF_PART--\r\n')
extra_headers['MIME-version'] = '1.0'
extra_headers['Content-Length'] = str(
len(multipart[0]) + len(multipart[1]) + len(multipart[2]) +
len(data_str) + media_source.content_length)
server_response = self.handler.HttpRequest(
self,
verb, [
multipart[0], data_str, multipart[1],
media_source.file_handle, multipart[2]
],
uri,
extra_headers=extra_headers,
url_params=url_params,
escape_params=escape_params,
content_type='multipart/related; boundary=END_OF_PART')
result_body = server_response.read()
elif media_source or isinstance(data, gdata.MediaSource):
if isinstance(data, gdata.MediaSource):
media_source = data
extra_headers['Content-Length'] = media_source.content_length
server_response = self.handler.HttpRequest(
self,
verb,
media_source.file_handle,
uri,
extra_headers=extra_headers,
url_params=url_params,
escape_params=escape_params,
content_type=media_source.content_type)
result_body = server_response.read()
else:
http_data = data
content_type = 'application/atom+xml'
server_response = self.handler.HttpRequest(
self,
verb,
http_data,
uri,
extra_headers=extra_headers,
url_params=url_params,
escape_params=escape_params,
content_type=content_type)
result_body = server_response.read()
# Server returns 201 for most post requests, but when performing a batch
# request the server responds with a 200 on success.
if server_response.status == 201 or server_response.status == 200:
if converter:
return converter(result_body)
feed = gdata.GDataFeedFromString(result_body)
if not feed:
entry = gdata.GDataEntryFromString(result_body)
if not entry:
return result_body
return entry
return feed
elif server_response.status == 302:
if redirects_remaining > 0:
location = server_response.getheader('Location')
if location is not None:
m = re.compile('[\?\&]gsessionid=(\w*)').search(location)
if m is not None:
self.__gsessionid = m.group(1)
return self.Post(data,
location,
extra_headers,
url_params,
escape_params,
redirects_remaining - 1,
media_source,
converter=converter)
else:
raise RequestError, {
'status': server_response.status,
'reason': '302 received without Location header',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason':
'Redirect received, but redirects_remaining <= 0',
'body': result_body
}
else:
raise RequestError, {
'status': server_response.status,
'reason': server_response.reason,
'body': result_body
}
