"""This module holds the logic for doing simple data conversions.""" import stringcase def node_to_dict(node): """Unpack the data from a node.""" data = {k: v for k, v in node.items()} return data def format_response(payload): """Format payload according to conventions.""" if type(payload) is list: payload = {'items': payload, 'pagination': {'total_records': len(payload)}} return to_snake(payload) def to_snake(data): """Transform the keys of the payload to snake case. Args: data (dict): the payload to format the keys. Returns: dict: a new dictionary with the formatted payload """ snake_case_dict = {} for k, v in data.items(): snake_case_dict[stringcase.snakecase(k)] = v return snake_case_dict def to_camel(data): """Transform the keys of the payload to camel case. Args: data (dict): the payload to format the keys. Returns: dict: a new dictionary with the formatted payload """ camel_case_dict = {} for k, v in data.items(): camel_case_dict[stringcase.camelcase(k)] = v return camel_case_dict def to_int(data): """Convert data to int."""