diff options
Diffstat (limited to 'designateclient/v2/utils.py')
| -rw-r--r-- | designateclient/v2/utils.py | 42 |
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 |
