summaryrefslogtreecommitdiff
path: root/designateclient/v2/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'designateclient/v2/utils.py')
-rw-r--r--designateclient/v2/utils.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/designateclient/v2/utils.py b/designateclient/v2/utils.py
index 5fd6590..4a69a55 100644
--- a/designateclient/v2/utils.py
+++ b/designateclient/v2/utils.py
@@ -13,7 +13,12 @@
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
+
from oslo_utils import uuidutils
+from six import iteritems
+from six import iterkeys
+from six.moves.urllib.parse import parse_qs
+from six.moves.urllib.parse import urlparse
from designateclient import exceptions
@@ -36,3 +41,40 @@ def resolve_by_name(func, name, *args):
else:
msg = "Multiple matches found for %s, please use ID instead." % name
raise exceptions.NoUniqueMatch(msg)
+
+
+def parse_query_from_url(url):
+ """
+ Helper to get key bits of data from the "next" url returned
+ from the API on collections
+ :param url:
+ :return: dict
+ """
+ values = parse_qs(urlparse(url)[4])
+ return {k: values[k][0] for k in iterkeys(values)}
+
+
+def get_all(function, criterion=None, args=None):
+ """
+
+ :param function: Function to be called to get data
+ :param criterion: dict of filters to be applied
+ :param args: arguments to be given to the function
+ :return: DesignateList()
+ """
+
+ criterion = criterion or {}
+ args = args or []
+
+ data = function(*args, criterion=criterion)
+ returned_data = data
+ while True:
+ if data.next_page:
+ for k, v in iteritems(data.next_link_criterion):
+ criterion[k] = v
+ data = function(*args, criterion=criterion)
+ returned_data.extend(data)
+ else:
+ break
+
+ return returned_data \ No newline at end of file