对字典列表进行排序和分组

时间:2016-10-12 02:52:14

标签: python list python-3.x dictionary

如何对此词典列表进行排序和分组为嵌套词典,我希望通过API将其作为JSON返回。

源数据(权限列表):

[{
    'can_create': True,
    'can_read': True,
    'module_name': 'ModuleOne',
    'module_id': 1,
    'role_id': 1,
    'end_point_id': 1,
    'can_update': True,
    'end_point_name': 'entity',
    'can_delete': True,
}, {
    'can_create': True,
    'can_read': True,
    'module_name': 'ModuleTwo',
    'module_id': 2,
    'role_id': 1,
    'end_point_id': 4,
    'can_update': True,
    'end_point_name': 'financial-outlay',
    'can_delete': True,
},{
    'can_create': True,
    'can_read': True,
    'module_name': 'ModuleOne',
    'module_id': 1,
    'role_id': 1,
    'end_point_id': 2,
    'can_update': True,
    'end_point_name': 'management-type',
    'can_delete': True,
}, {
    'can_create': True,
    'can_read': True,
    'module_name': 'ModuleOne',
    'module_id': 1,
    'role_id': 1,
    'end_point_id': 3,
    'can_update': True,
    'end_point_name': 'ownership-type',
    'can_delete': False,
}, {
    'can_create': True,
    'can_read': True,
    'module_name': 'ModuleTwo',
    'module_id': 2,
    'role_id': 1,
    'end_point_id': 5,
    'can_update': True,
    'end_point_name': 'exposure',
    'can_delete': True,
}]

我想将其转换为嵌套的dicitonary对象,以使用API​​作为JSON返回。这是预期的输出:

{
    "role_id": 1,
    "modules": [{
            "module_id": 1,
            "module_name": "ModuleOne",
            "permissions": [{
                "end_point_id": 1,
                "end_point_name": "entity",
                "can_create": False,
                "can_read": True,
                "can_write": True,
                "can_delete": True
            }, {
                "end_point_id": 2,
                "end_point_name": "management-type",
                "can_create": False,
                "can_read": True,
                "can_write": True,
                "can_delete": True
            }, {
                "end_point_id": 3,
                "end_point_name": "ownership-type",
                "can_create": False,
                "can_read": True,
                "can_write": True,
                "can_delete": True
            }, ]
        }, {
            "module_id": 2,
            "module_name": "ModuleTwo",
            "permissions": [{
                "end_point_id": 4,
                "end_point_name": "financial-outlay",
                "can_create": False,
                "can_read": True,
                "can_write": True,
                "can_delete": True
            }, {
                "end_point_id": 5,
                "end_point_name": "exposure",
                "can_create": False,
                "can_read": True,
                "can_write": True,
                "can_delete": True
            }, ]
        },

    ]
}

直到我花了更多的时间才想要试图绕过它,这看起来微不足道。我尝试了很多选择而没有他们工作。这是最后一次尝试。

# Get user role
user_roles = get_user_roles()  # List of roles e.g. [{'role_id':1, role_name: 'role_one'}, {'role_id':2, role_name: 'role_two'}]

for role in user_roles:
    role_id = role['role_id']
    role_name = role['role_name']

    # Fetch Role Permissions
    role_permissions = get_role_permissions(role_id)  # List of permissions as seen above
    sorted_role_permissions = sorted(role_permissions, key=itemgetter('module_id'))  # sort dictionaries in list by 'module_id'

    modules_list = []
    permissions_list = []
    previous_module_id = 0
    is_first_loop = True
    for role_permission in sorted_role_permissions:

        module_id = role_permission['module_id']
        module_name = role_permission['module_name']
        end_point_id = role_permission['end_point_id']
        end_point_name = role_permission['end_point_name']

        if is_first_loop:
            print(0)
            is_first_loop = False
            previous_module_id = module_id
            print('end_point_name 0 {}'.format(end_point_name))
            permissions = {'end_point_id': end_point_id, 'end_point_name': end_point_name,
                           'can_create': role_permission['can_create'],
                           'can_read': role_permission['can_read'],
                           'can_update': role_permission['can_update'],
                           'can_delete': role_permission['can_delete']
                           }
            permissions_list.append(permissions)
            if len(sorted_role_permissions) == 1:
                # If there is only one permission in the role, end the loop
                modules_dict = {'module_id': module_id, 'module_name': module_name,
                                'permissions': permissions_list}
                modules_list.append(modules_dict)
                break

        else:

            if module_id == previous_module_id:
                # As long as the current module_id and the previous_module_id are the same, add to the same list

                print(1)
                permissions = {'end_point_id': end_point_id, 'end_point_name': end_point_name,
                               'can_create': role_permission['can_create'],
                               'can_read': role_permission['can_read'],
                               'can_update': role_permission['can_update'],
                               'can_delete': role_permission['can_delete']
                               }
                permissions_list.append(permissions)

            else:

                print(2)
                modules_dict = {'module_id': module_id, 'module_name': module_name,
                                'permissions': permissions_list}
                modules_list.append(modules_dict)
                permissions_list = []
                permissions = {'end_point_id': end_point_id, 'end_point_name': end_point_name,
                               'can_create': role_permission['can_create'],
                               'can_read': role_permission['can_read'],
                               'can_update': role_permission['can_update'],
                               'can_delete': role_permission['can_delete']
                               }
                permissions_list.append(permissions)
                previous_module_id = module_id

    if modules_list:
        roles.append({'role_id': role_id, 'role_name': role_name, 'modules': modules_list})

2 个答案:

答案 0 :(得分:2)

TADA!

from itertools import groupby

def group_by_remove(permissions, id_key, groups_key, name_key=None):
    """
    @type permissions: C{list} of C{dict} of C{str} to C{object}
    @param id_key: A string that represents the name of the id key, like "role_id" or "module_id"
    @param groups_key: A string that represents the name of the key of the groups like "modules" or "permissions"
    @param name_key: A string that represents the name of the key of names like "module_name" (can also be None for no names' key) 
    """
    result = []
    permissions_key = lambda permission: permission[id_key]
    # Must sort for groupby to work properly
    sorted_permissions = sorted(permissions, key=permissions_key)
    for key, groups in groupby(sorted_permissions, permissions_key):
        key_result = {}
        groups = list(groups)
        key_result[id_key] = key
        if name_key is not None:
            key_result[name_key] = groups[0][name_key]
        key_result[groups_key] = [{k: v for k, v in group.iteritems() if k != id_key and (name_key is None or k != name_key)} for group in groups]
        result.append(key_result)
    return result

def change_format(initial):
    """
    @type initial: C{list}
    @rtype: C{dict} of C{str} to C{list} of C{dict} of C{str} to C{object}
    """
    roles_group = group_by_remove(initial, "role_id", "modules")[0]
    roles_group["modules"] = group_by_remove(roles_group["modules"], "module_id", "permissions", "module_name")
    return roles_group

change_format(role_permissions)

享受:)

答案 1 :(得分:2)

PyFunctional非常擅长列表操作。

from pprint import pprint
from functional import seq
input = [...]  # taken from your example
output =(
    seq(input)  # convert regular python list to Sequence object

    # group by role_id
    .map(lambda e: (e.pop('role_id'), e)).group_by_key()

    # start building role dict
    .map(lambda role_modules: {
        "role_id": role_modules[0],
        "modules": seq(role_modules[1])
                   # group by (module_id, module_name)
                   .map(lambda e: ( (e.pop('module_id'), e.pop('module_name')), e) ).group_by_key()

                   # start building module/permissions dict
                   .map(lambda module_permissions: {
                       "module_id": module_permissions[0][0],
                       "module_name": module_permissions[0][1],
                       "permissions": module_permissions[1]
                   })
                   # sort by module_id, convert Seq obj to regular list
                   .sorted(key=lambda m:m['module_id']).to_list()
    })
    # sort by role_id, convert Seq obj to regular list
    .sorted(key=lambda r:r['role_id']).to_list()
)

pprint(output)

<强> RESULT

[{'modules': [{'module_id': 1,
               'module_name': 'ModuleOne',
               'permissions': [{'can_create': True,
                                'can_delete': True,
                                'can_read': True,
                                'can_update': True,
                                'end_point_id': 1,
                                'end_point_name': 'entity'},
                               {'can_create': True,
                                'can_delete': True,
                                'can_read': True,
                                'can_update': True,
                                'end_point_id': 2,
                                'end_point_name': 'management-type'},
                               {'can_create': True,
                                'can_delete': False,
                                'can_read': True,
                                'can_update': True,
                                'end_point_id': 3,
                                'end_point_name': 'ownership-type'}]},
              {'module_id': 2,
               'module_name': 'ModuleTwo',
               'permissions': [{'can_create': True,
                                'can_delete': True,
                                'can_read': True,
                                'can_update': True,
                                'end_point_id': 4,
                                'end_point_name': 'financial-outlay'},
                               {'can_create': True,
                                'can_delete': True,
                                'can_read': True,
                                'can_update': True,
                                'end_point_id': 5,
                                'end_point_name': 'exposure'}]}],
  'role_id': 1}]