mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-04-24 11:21:25 -07:00
a4eb4b2551
* Simplify permission changing code for rabbitmq_user module * Add check for multiple permission dicts for same host to rabbitmq_user module * Add docstring for _get_permission method of rabbitmq_user * Fix method that compares vhost permissions in rabbitmq_user * Add tests for rabbitmq_user module * Add helper function for simulating collections.Counter functionality
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
# Copyright (c), Sviatoslav Sydorenko <ssydoren@redhat.com> 2018
|
|
# Simplified BSD License (see licenses/simplified_bsd.txt or https://opensource.org/licenses/BSD-2-Clause)
|
|
"""Collection of low-level utility functions."""
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
__metaclass__ = type
|
|
|
|
|
|
from ansible.module_utils.six import binary_type, text_type
|
|
from ansible.module_utils.common._collections_compat import Sequence
|
|
|
|
|
|
def is_string(seq):
|
|
"""Identify whether the input has a string-like type (inclding bytes)."""
|
|
return isinstance(seq, (text_type, binary_type))
|
|
|
|
|
|
def is_iterable(seq, include_strings=False):
|
|
"""Identify whether the input is an iterable."""
|
|
if not include_strings and is_string(seq):
|
|
return False
|
|
|
|
try:
|
|
iter(seq)
|
|
return True
|
|
except TypeError:
|
|
return False
|
|
|
|
|
|
def is_sequence(seq, include_strings=False):
|
|
"""Identify whether the input is a sequence.
|
|
|
|
Strings and bytes are not sequences here,
|
|
unless ``include_string`` is ``True``.
|
|
|
|
Non-indexable things are never of a sequence type.
|
|
"""
|
|
if not include_strings and is_string(seq):
|
|
return False
|
|
|
|
return isinstance(seq, Sequence)
|
|
|
|
|
|
def count(seq):
|
|
"""Returns a dictionary with the number of appearances of each element of the iterable.
|
|
|
|
Resembles the collections.Counter class functionality. It is meant to be used when the
|
|
code is run on Python 2.6.* where collections.Counter is not available. It should be
|
|
deprecated and replaced when support for Python < 2.7 is dropped.
|
|
"""
|
|
if not is_iterable(seq):
|
|
raise Exception('Argument provided is not an iterable')
|
|
counters = dict()
|
|
for elem in seq:
|
|
counters[elem] = counters.get(elem, 0) + 1
|
|
return counters
|