Simplify XML query

This commit is contained in:
Markus Unterwaditzer 2015-01-02 21:09:49 +01:00
parent 51a8312f1e
commit c1653937d1

View file

@ -129,9 +129,8 @@ class Discover(object):
data=body, is_subpath=False) data=body, is_subpath=False)
root = _parse_xml(response.content) root = _parse_xml(response.content)
for element in root.iter('{*}current-user-principal'): for principal in root.iter('{*}current-user-principal/{*}href'):
for principal in element.iter(): # should be only one # should be only one
if principal.tag.endswith('href'):
yield principal.text yield principal.text
def find_dav(self): def find_dav(self):
@ -172,9 +171,7 @@ class Discover(object):
is_subpath=False) is_subpath=False)
root = etree.fromstring(response.content) root = etree.fromstring(response.content)
for element in root.iter(self._homeset_tag): for homeset in root.iter(self._homeset_tag + '/{*}href'):
for homeset in element.iter():
if homeset.tag.endswith('href'):
yield homeset.text yield homeset.text
def find_collections(self): def find_collections(self):