Source code for pydash.objects

# -*- coding: utf-8 -*-
"""Functions that operate on lists, dicts, and other objects.

.. versionadded:: 1.0.0
"""

from __future__ import absolute_import

import copy
from functools import partial
import math
import re

import pydash as pyd
from .helpers import (
    NoValue,
    callit,
    iterator,
    iteriteratee,
    getargcount,
    base_get,
    base_set
)
from ._compat import iteritems, text_type
from .utilities import PathToken, to_path, to_path_tokens


__all__ = (
    'assign',
    'assign_with',
    'callables',
    'clone',
    'clone_deep',
    'clone_deep_with',
    'clone_with',
    'defaults',
    'defaults_deep',
    'find_key',
    'find_last_key',
    'for_in',
    'for_in_right',
    'get',
    'has',
    'invert',
    'invert_by',
    'invoke',
    'keys',
    'map_keys',
    'map_values',
    'map_values_deep',
    'merge',
    'merge_with',
    'omit',
    'omit_by',
    'parse_int',
    'pick',
    'pick_by',
    'rename_keys',
    'set_',
    'set_with',
    'to_boolean',
    'to_dict',
    'to_integer',
    'to_number',
    'to_pairs',
    'to_string',
    'transform',
    'unset',
    'update',
    'update_with',
    'values',
)


[docs]def assign(obj, *sources): """Assigns properties of source object(s) to the destination object. Args: obj (dict): Destination object whose properties will be modified. sources (dict): Source objects to assign to `obj`. Returns: dict: Modified `obj`. Warning: `obj` is modified in place. Example: >>> obj = {} >>> obj2 = assign(obj, {'a': 1}, {'b': 2}, {'c': 3}) >>> obj == {'a': 1, 'b': 2, 'c': 3} True >>> obj is obj2 True .. versionadded:: 1.0.0 .. versionchanged:: 2.3.2 Apply :func:`clone_deep` to each `source` before assigning to `obj`. .. versionchanged:: 3.0.0 Allow iteratees to accept partial arguments. .. versionchanged:: 3.4.4 Shallow copy each `source` instead of deep copying. .. versionchanged:: 4.0.0 - Moved `iteratee` argument to :func:`assign_with`. - Removed alias ``extend``. """ return assign_with(obj, *sources)
[docs]def assign_with(obj, *sources, **kargs): """This method is like :func:`assign` except that it accepts customizer which is invoked to produce the assigned values. If customizer returns ``None``, assignment is handled by the method instead. The customizer is invoked with five arguments: ``(obj_value, src_value, key, obj, source)``. Args: obj (dict): Destination object whose properties will be modified. sources (dict): Source objects to assign to `obj`. Keyword Args: customizer (mixed, optional): Customizer applied per iteration. Returns: dict: Modified `obj`. Warning: `obj` is modified in place. Example: >>> customizer = lambda o, s: s if o is None else o >>> results = assign({'a': 1}, {'b': 2}, {'a': 3}, customizer) >>> results == {'a': 1, 'b': 2} True .. versionadded:: 4.0.0 """ sources = list(sources) customizer = kargs.get('customizer') if customizer is None and callable(sources[-1]): customizer = sources.pop() if customizer is not None: argcount = getargcount(customizer, maxargs=5) else: argcount = None for source in sources: source = source.copy() for key, value in iteritems(source): if customizer: val = callit(customizer, obj.get(key), value, key, obj, source, argcount=argcount) if val is not None: value = val obj[key] = value return obj
[docs]def callables(obj): """Creates a sorted list of keys of an object that are callable. Args: obj (list|dict): Object to inspect. Returns: list: All keys whose values are callable. Example: >>> callables({'a': 1, 'b': lambda: 2, 'c': lambda: 3}) ['b', 'c'] .. versionadded:: 1.0.0 .. versionchanged:: 2.0.0 Renamed ``functions`` to ``callables``. .. versionchanged:: 4.0.0 Removed alias ``methods``. """ return sorted(key for key, value in iterator(obj) if callable(value))
[docs]def clone(value): """Creates a clone of `value`. Args: value (list|dict): Object to clone. Example: >>> x = {'a': 1, 'b': 2, 'c': {'d': 3}} >>> y = clone(x) >>> y == y True >>> x is y False >>> x['c'] is y['c'] True Returns: list|dict: Cloned object. .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved 'iteratee' parameter to :func:`clone_with`. """ return base_clone(value)
[docs]def clone_with(value, customizer=None): """This method is like :func:`clone` except that it accepts customizer which is invoked to produce the cloned value. If customizer returns ``None``, cloning is handled by the method instead. The customizer is invoked with up to three arguments: ``(value, index|key, object)``. Args: value (list|dict): Object to clone. customizer (callable, optional): Function to customize cloning. Returns: list|dict: Cloned object. Example: >>> x = {'a': 1, 'b': 2, 'c': {'d': 3}} >>> cbk = lambda v, k: v + 2 if isinstance(v, int) and k else None >>> y = clone_with(x, cbk) >>> y == {'a': 3, 'b': 4, 'c': {'d': 3}} True """ return base_clone(value, customizer=customizer)
[docs]def clone_deep(value): """Creates a deep clone of `value`. If a iteratee is provided it will be executed to produce the cloned values. Args: value (list|dict): Object to clone. Returns: list|dict: Cloned object. Example: >>> x = {'a': 1, 'b': 2, 'c': {'d': 3}} >>> y = clone_deep(x) >>> y == y True >>> x is y False >>> x['c'] is y['c'] False .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved 'iteratee' parameter to :func:`clone_deep_with`. """ return base_clone(value, is_deep=True)
[docs]def clone_deep_with(value, customizer=None): """This method is like :func:`clone_with` except that it recursively clones `value`. Args: value (list|dict): Object to clone. customizer (callable, optional): Function to customize cloning. Returns: list|dict: Cloned object. """ return base_clone(value, is_deep=True, customizer=customizer)
[docs]def defaults(obj, *sources): """Assigns properties of source object(s) to the destination object for all destination properties that resolve to undefined. Args: obj (dict): Destination object whose properties will be modified. sources (dict): Source objects to assign to `obj`. Returns: dict: Modified `obj`. Warning: `obj` is modified in place. Example: >>> obj = {'a': 1} >>> obj2 = defaults(obj, {'b': 2}, {'c': 3}, {'a': 4}) >>> obj is obj2 True >>> obj == {'a': 1, 'b': 2, 'c': 3} True .. versionadded:: 1.0.0 """ for source in sources: for key, value in iteritems(source): obj.setdefault(key, value) return obj
[docs]def defaults_deep(obj, *sources): """This method is like :func:`defaults` except that it recursively assigns default properties. Args: obj (dict): Destination object whose properties will be modified. sources (dict): Source objects to assign to `obj`. Returns: dict: Modified `obj`. Warning: `obj` is modified in place. Example: >>> obj = {'a': {'b': 1}} >>> obj2 = defaults_deep(obj, {'a': {'b': 2, 'c': 3}}) >>> obj is obj2 True >>> obj == {'a': {'b': 1, 'c': 3}} True .. versionadded:: 3.3.0 """ def setter(obj, key, value): obj.setdefault(key, value) return merge_with(obj, *sources, _setter=setter)
[docs]def find_key(obj, predicate=None): """This method is like :func:`pydash.arrays.find_index` except that it returns the key of the first element that passes the predicate check, instead of the element itself. Args: obj (list|dict): Object to search. predicate (mixed): Predicate applied per iteration. Returns: mixed: Found key or ``None``. Example: >>> find_key({'a': 1, 'b': 2, 'c': 3}, lambda x: x == 1) 'a' >>> find_key([1, 2, 3, 4], lambda x: x == 1) 0 .. versionadded:: 1.0.0 """ for result, _, key, _ in iteriteratee(obj, predicate): if result: return key
[docs]def find_last_key(obj, predicate=None): """This method is like :func:`find_key` except that it iterates over elements of a collection in the opposite order. Args: obj (list|dict): Object to search. predicate (mixed): Predicate applied per iteration. Returns: mixed: Found key or ``None``. Example: >>> find_last_key({'a': 1, 'b': 2, 'c': 3}, lambda x: x == 1) 'a' >>> find_last_key([1, 2, 3, 1], lambda x: x == 1) 3 .. versionchanged:: 4.0.0 Made into its own function (instead of an alias of ``find_key``) with proper reverse find implementation. """ reversed_obj = reversed(list(iteriteratee(obj, predicate))) for result, _, key, _ in reversed_obj: if result: return key
[docs]def for_in(obj, iteratee=None): """Iterates over own and inherited enumerable properties of `obj`, executing `iteratee` for each property. Args: obj (list|dict): Object to process. iteratee (mixed): Iteratee applied per iteration. Returns: list|dict: `obj`. Example: >>> obj = {} >>> def cb(v, k): obj[k] = v >>> results = for_in({'a': 1, 'b': 2, 'c': 3}, cb) >>> results == {'a': 1, 'b': 2, 'c': 3} True >>> obj == {'a': 1, 'b': 2, 'c': 3} True .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Removed alias ``for_own``. """ walk = (None for ret, _, _, _ in iteriteratee(obj, iteratee) if ret is False) next(walk, None) return obj
[docs]def for_in_right(obj, iteratee=None): """This function is like :func:`for_in` except it iterates over the properties in reverse order. Args: obj (list|dict): Object to process. iteratee (mixed): Iteratee applied per iteration. Returns: list|dict: `obj`. Example: >>> data = {'product': 1} >>> def cb(v): data['product'] *= v >>> for_in_right([1, 2, 3, 4], cb) [1, 2, 3, 4] >>> data['product'] == 24 True .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Removed alias ``for_own_right``. """ walk = (None for ret, _, _, _ in iteriteratee(obj, iteratee, reverse=True) if ret is False) next(walk, None) return obj
[docs]def get(obj, path, default=None): """Get the value at any depth of a nested object based on the path described by `path`. If path doesn't exist, `default` is returned. Args: obj (list|dict): Object to process. path (str|list): List or ``.`` delimited string of path describing path. Keyword Arguments: default (mixed): Default value to return if path doesn't exist. Defaults to ``None``. Returns: mixed: Value of `obj` at path. Example: >>> get({}, 'a.b.c') is None True >>> get({'a': {'b': {'c': [1, 2, 3, 4]}}}, 'a.b.c.1') 2 >>> get({'a': {'b': [0, {'c': [1, 2]}]}}, 'a.b.1.c.1') 2 >>> get({'a': {'b': [0, {'c': [1, 2]}]}}, 'a.b.1.c.2') is None True .. versionadded:: 2.0.0 .. versionchanged:: 2.2.0 Support escaping "." delimiter in single string path key. .. versionchanged:: 3.3.0 - Added :func:`get` as main definition and :func:`get_path` as alias. - Made :func:`deep_get` an alias. .. versionchanged:: 3.4.7 Fixed bug where an iterable default was iterated over instead of being returned when an object path wasn't found. .. versionchanged:: 4.0.0 - Support attribute access on `obj` if item access fails. - Removed aliases ``get_path`` and ``deep_get``. """ if default is NoValue: # When NoValue given for default, then this method will raise if path # is not present in obj. sentinel = default else: # When a returnable default is given, use a sentinel value to detect # when base_get() returns a default value for a missing path so we can # exit early from the loop and not mistakenly iterate over the default. sentinel = object() for key in to_path(path): obj = base_get(obj, key, default=sentinel) if obj is sentinel: # Path doesn't exist so set return obj to the default. obj = default break return obj
[docs]def has(obj, path): """Checks if `path` exists as a key of `obj`. Args: obj (mixed): Object to test. path (mixed): Path to test for. Can be a list of nested keys or a ``.`` delimited string of path describing the path. Returns: bool: Whether `obj` has `path`. Example: >>> has([1, 2, 3], 1) True >>> has({'a': 1, 'b': 2}, 'b') True >>> has({'a': 1, 'b': 2}, 'c') False >>> has({'a': {'b': [0, {'c': [1, 2]}]}}, 'a.b.1.c.1') True >>> has({'a': {'b': [0, {'c': [1, 2]}]}}, 'a.b.1.c.2') False .. versionadded:: 1.0.0 .. versionchanged:: 3.0.0 Return ``False`` on ``ValueError`` when checking path. .. versionchanged:: 3.3.0 - Added :func:`deep_has` as alias. - Added :func:`has_path` as alias. .. versionchanged:: 4.0.0 Removed aliases ``deep_has`` and ``has_path``. """ try: get(obj, path, default=NoValue) exists = True except (KeyError, IndexError, TypeError, ValueError): exists = False return exists
[docs]def invert(obj): """Creates an object composed of the inverted keys and values of the given object. Args: obj (dict): Dict to invert. multivalue (bool, optional): Whether to return inverted values as lists. Defaults to ``False``. Returns: dict: Inverted dict. Example: >>> results = invert({'a': 1, 'b': 2, 'c': 3}) >>> results == {1: 'a', 2: 'b', 3: 'c'} True Note: Assumes `obj` values are hashable as ``dict`` keys. .. versionadded:: 1.0.0 .. versionchanged:: 2.0.0 Added ``multivalue`` argument. .. versionchanged:: 4.0.0 Moved ``multivalue=True`` functionality to :func:`invert_by`. """ return dict((value, key) for key, value in iterator(obj))
[docs]def invert_by(obj, iteratee=None): """This method is like :func:`invert` except that the inverted object is generated from the results of running each element of object thru iteratee. The corresponding inverted value of each inverted key is a list of keys responsible for generating the inverted value. The iteratee is invoked with one argument: ``(value)``. Args: obj (dict): Object to invert. Returns: dict: Inverted dict. Example: >>> obj = {'a': 1, 'b': 2, 'c': 1} >>> results = invert_by(obj) # {1: ['a', 'c'], 2: ['b']} >>> set(results[1]) == set(['a', 'c']) True >>> set(results[2]) == set(['b']) True >>> results2 = invert_by(obj, lambda value: 'group' + str(value)) >>> results2['group1'] == results[1] True >>> results2['group2'] == results[2] True Note: Assumes `obj` values are hashable as ``dict`` keys. .. versionadded:: 4.0.0 """ callback = pyd.iteratee(iteratee) result = {} for key, value in iterator(obj): result.setdefault(callback(value), []).append(key) return result
[docs]def invoke(obj, path, *args, **kargs): """Invokes the method at path of object. Args: obj (dict): The object to query. path (list|str): The path of the method to invoke. args (optional): Arguments to pass to method call. kargs (optional): Keyword arguments to pass to method call. Returns: mixed: Result of the invoked method. Example: >>> obj = {'a': [{'b': {'c': [1, 2, 3, 4]}}]} >>> invoke(obj, 'a[0].b.c.pop', 1) 2 >>> obj {'a': [{'b': {'c': [1, 3, 4]}}]} .. versionadded:: 1.0.0 """ return get(obj, path, default=pyd.noop)(*args, **kargs)
[docs]def keys(obj): """Creates a list composed of the keys of `obj`. Args: obj (mixed): Object to extract keys from. Returns: list: List of keys. Example: >>> keys([1, 2, 3]) [0, 1, 2] >>> set(keys({'a': 1, 'b': 2, 'c': 3})) == set(['a', 'b', 'c']) True .. versionadded:: 1.0.0 .. versionchanged:: 1.1.0 Added ``keys_in`` as alias. .. versionchanged:: 4.0.0 Removed alias ``keys_in``. """ return [key for key, _ in iterator(obj)]
[docs]def map_keys(obj, iteratee=None): """The opposite of :func:`map_values`, this method creates an object with the same values as object and keys generated by running each own enumerable string keyed property of object thru iteratee. The iteratee is invoked with three arguments: ``(value, key, object)``. Args: obj (list|dict): Object to map. iteratee (mixed): Iteratee applied per iteration. Returns: list|dict: Results of running `obj` through `iteratee`. Example: >>> callback = lambda value, key: key * 2 >>> results = map_keys({'a': 1, 'b': 2, 'c': 3}, callback) >>> results == {'aa': 1, 'bb': 2, 'cc': 3} True .. versionadded:: 3.3.0 """ return dict((result, value) for result, value, _, _ in iteriteratee(obj, iteratee))
[docs]def map_values(obj, iteratee=None): """Creates an object with the same keys as object and values generated by running each string keyed property of object thru iteratee. The iteratee is invoked with three arguments: ``(value, key, object)``. Args: obj (list|dict): Object to map. iteratee (mixed): Iteratee applied per iteration. Returns: list|dict: Results of running `obj` through `iteratee`. Example: >>> results = map_values({'a': 1, 'b': 2, 'c': 3}, lambda x: x * 2) >>> results == {'a': 2, 'b': 4, 'c': 6} True >>> results = map_values({'a': 1, 'b': {'d': 4}, 'c': 3}, {'d': 4}) >>> results == {'a': False, 'b': True, 'c': False} True .. versionadded:: 1.0.0 """ return dict((key, result) for result, _, key, _ in iteriteratee(obj, iteratee))
[docs]def map_values_deep(obj, iteratee=None, property_path=NoValue): """Map all non-object values in `obj` with return values from `iteratee`. The iteratee is invoked with two arguments: ``(obj_value, property_path)`` where ``property_path`` contains the list of path keys corresponding to the path of ``obj_value``. Args: obj (list|dict): Object to map. iteratee (function): Iteratee applied to each value. Returns: mixed: The modified object. Warning: `obj` is modified in place. Example: >>> x = {'a': 1, 'b': {'c': 2}} >>> y = map_values_deep(x, lambda val: val * 2) >>> y == {'a': 2, 'b': {'c': 4}} True >>> z = map_values_deep(x, lambda val, props: props) >>> z == {'a': ['a'], 'b': {'c': ['b', 'c']}} True .. versionadded: 2.2.0 .. versionchanged:: 3.0.0 Allow iteratees to accept partial arguments. .. versionchanged:: 4.0.0 Renamed from ``deep_map_values`` to ``map_values_deep``. """ properties = to_path(property_path) if pyd.is_object(obj): deep_iteratee = ( lambda value, key: map_values_deep(value, iteratee, pyd.flatten([properties, key]))) return assign(obj, map_values(obj, deep_iteratee)) else: return callit(iteratee, obj, properties)
[docs]def merge(obj, *sources): """Recursively merges properties of the source object(s) into the destination object. Subsequent sources will overwrite property assignments of previous sources. Args: obj (dict): Destination object to merge source(s) into. sources (dict): Source objects to merge from. subsequent sources overwrite previous ones. Returns: dict: Merged object. Warning: `obj` is modified in place. Example: >>> obj = {'a': 2} >>> obj2 = merge(obj, {'a': 1}, {'b': 2, 'c': 3}, {'d': 4}) >>> obj2 == {'a': 1, 'b': 2, 'c': 3, 'd': 4} True >>> obj is obj2 True .. versionadded:: 1.0.0 .. versionchanged:: 2.3.2 Apply :func:`clone_deep` to each `source` before assigning to `obj`. .. versionchanged:: 2.3.2 Allow `iteratee` to be passed by reference if it is the last positional argument. .. versionchanged:: 4.0.0 Moved iteratee argument to :func:`merge_with`. """ return merge_with(obj, *sources)
[docs]def merge_with(obj, *sources, **kargs): """This method is like :func:`merge` except that it accepts customizer which is invoked to produce the merged values of the destination and source properties. If customizer returns ``None``, merging is handled by this method instead. The customizer is invoked with five arguments: ``(obj_value, src_value, key, obj, source)``. Args: obj (dict): Destination object to merge source(s) into. sources (dict): Source objects to merge from. subsequent sources overwrite previous ones. Keyword Args: iteratee (function, optional): Iteratee function to handle merging (must be passed in as keyword argument). Returns: dict: Merged object. Warning: `obj` is modified in place. Example: >>> cbk = lambda obj_val, src_val: obj_val + src_val >>> obj1 = {'a': [1], 'b': [2]} >>> obj2 = {'a': [3], 'b': [4]} >>> res = merge_with(obj1, obj2, cbk) >>> obj1 == {'a': [1, 3], 'b': [2, 4]} True .. versionadded:: 4.0.0 """ sources = list(sources) _clone = kargs.get('_clone', True) iteratee = kargs.get('iteratee') setter = kargs.get('_setter', base_set) if iteratee is None and callable(sources[-1]): iteratee = sources.pop() if callable(iteratee): argcount = getargcount(iteratee, maxargs=5) cbk = partial(callit, iteratee, argcount=argcount) else: cbk = None for source in sources: # Don't re-clone if we've already cloned before. if _clone: source = copy.deepcopy(source) for key, src_value in iterator(source): obj_value = base_get(obj, key, default=None) all_sequences = all([isinstance(src_value, list), isinstance(obj_value, list)]) all_mappings = all([isinstance(src_value, dict), isinstance(obj_value, dict)]) if cbk: _result = cbk(obj_value, src_value, key, obj, source) else: _result = None if _result is not None: result = _result elif all_sequences or all_mappings: result = merge_with(obj_value, src_value, _clone=False, _setter=setter) else: result = src_value setter(obj, key, result) return obj
[docs]def omit(obj, *properties): """The opposite of :func:`pick`. This method creates an object composed of the property paths of `obj` that are not omitted. Args: obj (mixed): Object to process. properties (str): Property values to omit. Returns: dict: Results of omitting properties. Example: >>> omit({'a': 1, 'b': 2, 'c': 3}, 'b', 'c') == {'a': 1} True >>> omit({ 'a': 1, 'b': 2, 'c': 3 }, ['a', 'c']) == {'b': 2} True >>> omit([1, 2, 3, 4], 0, 3) == {1: 2, 2: 3} True .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved iteratee argument to :func:`omit_by`. """ return omit_by(obj, pyd.flatten(properties))
[docs]def omit_by(obj, iteratee=None): """The opposite of :func:`pick_by`. This method creates an object composed of the string keyed properties of object that predicate doesn't return truthy for. The predicate is invoked with two arguments: ``(value, key)``. Args: obj (mixed): Object to process. iteratee (mixed, optional): Iteratee used to determine which properties to omit. Returns: dict: Results of omitting properties. Example: >>> omit_by({'a': 1, 'b': '2', 'c': 3}, lambda v: isinstance(v, int)) {'b': '2'} .. versionadded:: 4.0.0 """ if not callable(iteratee): keys = iteratee if iteratee is not None else [] def iteratee(value, key): # pylint: disable=function-redefined return key in keys argcount = 2 else: argcount = getargcount(iteratee, maxargs=2) return dict((key, value) for key, value in iterator(obj) if not callit(iteratee, value, key, argcount=argcount))
[docs]def parse_int(value, radix=None): """Converts the given `value` into an integer of the specified `radix`. If `radix` is falsey, a radix of ``10`` is used unless the `value` is a hexadecimal, in which case a radix of 16 is used. Args: value (mixed): Value to parse. radix (int, optional): Base to convert to. Returns: mixed: Integer if parsable else ``None``. Example: >>> parse_int('5') 5 >>> parse_int('12', 8) 10 >>> parse_int('x') is None True .. versionadded:: 1.0.0 """ if not radix and pyd.is_string(value): try: # Check if value is hexadcimal and if so use base-16 conversion. int(value, 16) except ValueError: pass else: radix = 16 if not radix: radix = 10 try: # NOTE: Must convert value to string when supplying radix to int(). # Dropping radix arg when 10 is needed to allow floats to parse # correctly. args = (value,) if radix == 10 else (to_string(value), radix) parsed = int(*args) except (ValueError, TypeError): parsed = None return parsed
[docs]def pick(obj, *properties): """Creates an object composed of the picked object properties. Args: obj (list|dict): Object to pick from. properties (str): Property values to pick. Returns: dict: Dict containg picked properties. Example: >>> pick({'a': 1, 'b': 2, 'c': 3}, 'a', 'b') == {'a': 1, 'b': 2} True .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved iteratee argument to :func:`pick_by`. """ return pick_by(obj, pyd.flatten(properties))
[docs]def pick_by(obj, iteratee=None): """Creates an object composed of the object properties predicate returns truthy for. The predicate is invoked with two arguments: ``(value, key)``. Args: obj (list|dict): Object to pick from. iteratee (mixed, optional): Iteratee used to determine which properties to pick. Returns: dict: Dict containg picked properties. Example: >>> obj = {'a': 1, 'b': '2', 'c': 3 } >>> pick_by(obj, lambda v: isinstance(v, int)) == {'a': 1, 'c': 3} True .. versionadded:: 4.0.0 """ if not callable(iteratee): keys = iteratee if iteratee is not None else [] def iteratee(value, key): # pylint: disable=function-redefined return key in keys argcount = 2 else: argcount = getargcount(iteratee, maxargs=2) return dict((key, value) for key, value in iterator(obj) if callit(iteratee, value, key, argcount=argcount))
[docs]def rename_keys(obj, key_map): """Rename the keys of `obj` using `key_map` and return new object. Args: obj (dict): Object to rename. key_map (dict): Renaming map whose keys correspond to existing keys in `obj` and whose values are the new key name. Returns: dict: Renamed `obj`. Example: >>> obj = rename_keys({'a': 1, 'b': 2, 'c': 3}, {'a': 'A', 'b': 'B'}) >>> obj == {'A': 1, 'B': 2, 'c': 3} True .. versionadded:: 2.0.0 """ return dict((key_map.get(key, key), value) for key, value in iteritems(obj))
[docs]def set_(obj, path, value): """Sets the value of an object described by `path`. If any part of the object path doesn't exist, it will be created. Args: obj (list|dict): Object to modify. path (str | list): Target path to set value to. value (mixed): Value to set. Returns: mixed: Modified `obj`. Warning: `obj` is modified in place. Example: >>> set_({}, 'a.b.c', 1) {'a': {'b': {'c': 1}}} >>> set_({}, 'a.0.c', 1) {'a': {'0': {'c': 1}}} >>> set_([1, 2], '[2][0]', 1) [1, 2, [1]] >>> set_({}, 'a.b[0].c', 1) {'a': {'b': [{'c': 1}]}} .. versionadded:: 2.2.0 .. versionchanged:: 3.3.0 Added :func:`set_` as main definition and :func:`deep_set` as alias. .. versionchanged:: 4.0.0 - Modify `obj` in place. - Support creating default path values as ``list`` or ``dict`` based on whether key or index substrings are used. - Remove alias ``deep_set``. """ return set_with(obj, path, value)
[docs]def set_with(obj, path, value, customizer=None): """This method is like :func:`set_` except that it accepts customizer which is invoked to produce the objects of path. If customizer returns undefined path creation is handled by the method instead. The customizer is invoked with three arguments: ``(nested_value, key, nested_object)``. Args: obj (list|dict): Object to modify. path (str | list): Target path to set value to. value (mixed): Value to set. customizer (function, optional): The function to customize assigned values. Returns: mixed: Modified `obj`. Warning: `obj` is modified in place. Example: >>> set_with({}, '[0][1]', 'a', lambda: {}) {0: {1: 'a'}} .. versionadded:: 4.0.0 """ return update_with(obj, path, value, customizer=customizer)
[docs]def to_boolean(obj, true_values=('true', '1'), false_values=('false', '0')): """Convert `obj` to boolean. This is not like the builtin ``bool`` function. By default commonly considered strings values are converted to their boolean equivalent, i.e., ``'0'`` and ``'false'`` are converted to ``False`` while ``'1'`` and ``'true'`` are converted to ``True``. If a string value is provided that isn't recognized as having a common boolean conversion, then the returned value is ``None``. Non-string values of `obj` are converted using ``bool``. Optionally, `true_values` and `false_values` can be overridden but each value must be a string. Args: obj (mixed): Object to convert. true_values (tuple, optional): Values to consider ``True``. Each value must be a string. Comparision is case-insensitive. Defaults to ``('true', '1')``. false_values (tuple, optional): Values to consider ``False``. Each value must be a string. Comparision is case-insensitive. Defaults to ``('false', '0')``. Returns: bool: Boolean value of `obj`. Example: >>> to_boolean('true') True >>> to_boolean('1') True >>> to_boolean('false') False >>> to_boolean('0') False >>> assert to_boolean('a') is None .. versionadded:: 3.0.0 """ if pyd.is_string(obj): obj = obj.strip() def boolean_match(text, vals): if text.lower() in [val.lower() for val in vals]: return True else: return re.match('|'.join(vals), text) if true_values and boolean_match(obj, true_values): value = True elif false_values and boolean_match(obj, false_values): value = False else: value = None else: value = bool(obj) return value
[docs]def to_dict(obj): """Convert `obj` to ``dict`` by created a new ``dict`` using `obj` keys and values. Args: obj: (mixed): Object to convert. Returns: dict: Object converted to ``dict``. Example: >>> obj = {'a': 1, 'b': 2} >>> obj2 = to_dict(obj) >>> obj2 == obj True >>> obj2 is not obj True .. versionadded:: 3.0.0 .. versionchanged:: 4.0.0 Removed alias ``to_plain_object``. """ return dict(zip(pyd.keys(obj), pyd.values(obj)))
[docs]def to_integer(obj): """Converts `obj` to an integer. Args: obj (str|int|float): Object to convert. Returns: int: Converted integer or ``0`` if it can't be converted. Example: >>> to_integer(3.2) 3 >>> to_integer('3.2') 3 >>> to_integer('3.9') 3 >>> to_integer('invalid') 0 .. versionadded:: 4.0.0 """ try: # Convert to float first to handle converting floats as string since # int('1.1') would fail but this wouldn't. num = int(float(obj)) except (ValueError, TypeError): num = 0 return num
[docs]def to_number(obj, precision=0): """Convert `obj` to a number. All numbers are retuned as ``float``. If precision is negative, round `obj` to the nearest positive integer place. If `obj` can't be converted to a number, ``None`` is returned. Args: obj (str|int|float): Object to convert. precision (int, optional): Precision to round number to. Defaults to ``0``. Returns: float: Converted number or ``None`` if can't be converted. Example: >>> to_number('1234.5678') 1235.0 >>> to_number('1234.5678', 4) 1234.5678 >>> to_number(1, 2) 1.0 .. versionadded:: 3.0.0 """ try: factor = pow(10, precision) if precision < 0: # Round down since negative `precision` means we are going to # the nearest positive integer place. rounder = math.floor else: rounder = round num = rounder(float(obj) * factor) / factor except Exception: # pylint: disable=broad-except num = None return num
[docs]def to_pairs(obj): """Creates a two dimensional list of an object's key-value pairs, i.e. ``[[key1, value1], [key2, value2]]``. Args: obj (mixed): Object to process. Returns: list: Two dimensional list of object's key-value pairs. Example: >>> to_pairs([1, 2, 3, 4]) [[0, 1], [1, 2], [2, 3], [3, 4]] >>> to_pairs({'a': 1}) [['a', 1]] .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Renamed from ``pairs`` to ``to_pairs``. """ return [[key, value] for key, value in iterator(obj)]
[docs]def to_string(obj): """Converts an object to string. Args: obj (mixed): Object to convert. Returns: str: String representation of `obj`. Example: >>> to_string(1) == '1' True >>> to_string(None) == '' True >>> to_string([1, 2, 3]) == '[1, 2, 3]' True >>> to_string('a') == 'a' True .. versionadded:: 2.0.0 .. versionchanged:: 3.0.0 Convert ``None`` to empty string. """ if pyd.is_string(obj): res = obj elif obj is None: res = '' else: res = text_type(obj) return res
[docs]def transform(obj, iteratee=None, accumulator=None): """An alernative to :func:`pydash.collections.reduce`, this method transforms `obj` to a new accumulator object which is the result of running each of its properties through a iteratee, with each iteratee execution potentially mutating the accumulator object. The iteratee is invoked with four arguments: ``(accumulator, value, key, object)``. Iteratees may exit iteration early by explicitly returning ``False``. Args: obj (list|dict): Object to process. iteratee (mixed): Iteratee applied per iteration. accumulator (mixed, optional): Accumulated object. Defaults to ``list``. Returns: mixed: Accumulated object. Example: >>> transform([1, 2, 3, 4], lambda acc, v, k: acc.append((k, v))) [(0, 1), (1, 2), (2, 3), (3, 4)] .. versionadded:: 1.0.0 """ if iteratee is None: iteratee = pyd.identity if accumulator is None: accumulator = [] argcount = getargcount(iteratee, maxargs=4) walk = (None for key, item in iterator(obj) if callit(iteratee, accumulator, item, key, obj, argcount=argcount) is False) next(walk, None) return accumulator
[docs]def update(obj, path, updater): """This method is like :func:`set_` except that accepts updater to produce the value to set. Use :func:`update_with` to customize path creation. The updater is invoked with one argument: ``(value)``. Args: obj (list|dict): Object to modify. path (str|list): A string or list of keys that describe the object path to modify. updater (function): Function that returns updated value. Returns: mixed: Updated `obj`. Warning: `obj` is modified in place. Example: >>> update({}, ['a', 'b'], lambda value: value) {'a': {'b': None}} >>> update([], [0, 0], lambda value: 1) [[1]] .. versionadded:: 4.0.0 """ return update_with(obj, path, updater)
[docs]def update_with(obj, path, updater, customizer=None): """This method is like :func:`update` except that it accepts customizer which is invoked to produce the objects of path. If customizer returns ``None``, path creation is handled by the method instead. The customizer is invoked with three arguments: ``(nested_value, key, nested_object)``. Args: obj (list|dict): Object to modify. path (str|list): A string or list of keys that describe the object path to modify. updater (function): Function that returns updated value. customizer (function, optional): The function to customize assigned values. Returns: mixed: Updated `obj`. Warning: `obj` is modified in place. Example: >>> update_with({}, '[0][1]', lambda: 'a', lambda: {}) {0: {1: 'a'}} .. versionadded:: 4.0.0 """ if not callable(updater): updater = pyd.constant(updater) if customizer is not None and not callable(customizer): call_customizer = partial(callit, clone, customizer, argcount=1) elif customizer: call_customizer = partial(callit, customizer, argcount=getargcount(customizer, maxargs=3)) else: call_customizer = None default_type = dict if isinstance(obj, dict) else list tokens = to_path_tokens(path) if not pyd.is_list(tokens): # pragma: no cover tokens = [tokens] last_key = pyd.last(tokens) if isinstance(last_key, PathToken): last_key = last_key.key target = obj for idx, token in enumerate(pyd.initial(tokens)): if isinstance(token, PathToken): key = token.key default_factory = pyd.get(tokens, [idx + 1, 'default_factory'], default=default_type) else: key = token default_factory = default_type obj_val = base_get(target, key, default=None) path_obj = None if call_customizer: path_obj = call_customizer(obj_val, key, target) if path_obj is None: path_obj = default_factory() base_set(target, key, path_obj, allow_override=False) try: target = target[key] except TypeError: # pragma: no cover target = target[int(key)] value = base_get(target, last_key, default=None) base_set(target, last_key, callit(updater, value)) return obj
[docs]def unset(obj, path): """Removes the property at `path` of `obj`. Note: Only ``list``, ``dict``, or objects with a ``pop()`` method can be unset by this function. Args: obj (mixed): The object to modify. path (mixed): The path of the property to unset. Returns: bool: Whether the property was deleted. Warning: `obj` is modified in place. Example: >>> obj = {'a': [{'b': {'c': 7}}]} >>> unset(obj, 'a[0].b.c') True >>> obj {'a': [{'b': {}}]} >>> unset(obj, 'a[0].b.c') False """ tokens = to_path_tokens(path) if not pyd.is_list(tokens): # pragma: no cover tokens = [tokens] last_key = pyd.last(tokens) if isinstance(last_key, PathToken): last_key = last_key.key target = obj for idx, token in enumerate(pyd.initial(tokens)): if isinstance(token, PathToken): key = token.key else: key = token try: try: target = target[key] except TypeError: target = target[int(key)] except Exception: target = NoValue if target is NoValue: break did_unset = False if target is not NoValue: try: try: target.pop(last_key) did_unset = True except TypeError: target.pop(int(last_key)) did_unset = True except Exception: pass return did_unset
[docs]def values(obj): """Creates a list composed of the values of `obj`. Args: obj (mixed): Object to extract values from. Returns: list: List of values. Example: >>> results = values({'a': 1, 'b': 2, 'c': 3}) >>> set(results) == set([1, 2, 3]) True >>> values([2, 4, 6, 8]) [2, 4, 6, 8] .. versionadded:: 1.0.0 .. versionchanged:: 1.1.0 Added ``values_in`` as alias. .. versionchanged:: 4.0.0 Removed alias ``values_in``. """ return [value for _, value in iterator(obj)]
# # Utility methods not a part of the main API # def base_clone(value, is_deep=False, customizer=None, key=None, obj=None, _clone=True): """Base clone function that supports deep clone and customizer callback.""" clone_by = copy.deepcopy if is_deep else copy.copy result = None if callable(customizer) and _clone: argcount = getargcount(customizer, maxargs=4) cbk = partial(callit, customizer, argcount=argcount) elif not _clone: cbk = customizer else: cbk = None if cbk: result = cbk(value, key, value) if result is not None: return result if _clone: result = clone_by(value) else: result = value if cbk: for key, subvalue in iterator(value): if is_deep: val = base_clone(subvalue, is_deep, cbk, key, value, _clone=False) else: val = cbk(subvalue, key, value) if val is not None: result[key] = val return result