diff --git a/app/management/commands/fix_people_resource_role.py b/app/management/commands/fix_people_resource_role.py new file mode 100644 index 0000000..9da5d5a --- /dev/null +++ b/app/management/commands/fix_people_resource_role.py @@ -0,0 +1,16 @@ +from django.core.management.base import BaseCommand +from app.models import PeopleResource, Role + + +class Command(BaseCommand): + help = 'Fix roles' + + def handle(self, *args, **options): + prs = PeopleResource.objects.all() + roles = Role.objects.all() + role_map = {r.role: r.id for r in roles} + for pr in prs: + pr.p_r_role = role_map.get(pr.role, role_map.get('')) + pr.save() + print("Done %s -- %s -- %s -- %s" % (pr.id, pr.role, pr.resource, pr.person)) + print("Done all")