mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Add migration for replicating legact permissions to ObjectPermissions
This commit is contained in:
41
netbox/users/migrations/0009_replicate_permissions.py
Normal file
41
netbox/users/migrations/0009_replicate_permissions.py
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
from django.db import migrations
|
||||||
|
|
||||||
|
|
||||||
|
ACTIONS = ['view', 'add', 'change', 'delete']
|
||||||
|
|
||||||
|
|
||||||
|
def replicate_permissions(apps, schema_editor):
|
||||||
|
"""
|
||||||
|
Replicate all Permission assignments as ObjectPermissions.
|
||||||
|
"""
|
||||||
|
Permission = apps.get_model('auth', 'Permission')
|
||||||
|
ObjectPermission = apps.get_model('users', 'ObjectPermission')
|
||||||
|
|
||||||
|
# TODO: Optimize this iteration so that ObjectPermissions with identical sets of users and groups
|
||||||
|
# are combined into a single ObjectPermission instance.
|
||||||
|
for perm in Permission.objects.all():
|
||||||
|
print(f'Replicating permission {perm.codename}')
|
||||||
|
action, model_name = perm.codename.split('_')
|
||||||
|
|
||||||
|
if perm.group_set.exists() or perm.user_set.exists():
|
||||||
|
obj_perm = ObjectPermission(actions=[action])
|
||||||
|
obj_perm.save()
|
||||||
|
obj_perm.content_types.add(perm.content_type)
|
||||||
|
if perm.group_set.exists():
|
||||||
|
obj_perm.groups.add(*list(perm.group_set.all()))
|
||||||
|
if perm.user_set.exists():
|
||||||
|
obj_perm.users.add(*list(perm.user_set.all()))
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('users', '0008_objectpermission'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.RunPython(
|
||||||
|
code=replicate_permissions,
|
||||||
|
reverse_code=migrations.RunPython.noop
|
||||||
|
)
|
||||||
|
]
|
Reference in New Issue
Block a user