External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-07 19:51:08 -03:30
Code Issues Packages Projects Releases Wiki Activity

More project migration fixes

Browse Source
This commit is contained in:
Akita Noek
2016-03-14 16:01:12 -04:00
parent 9146b29770
commit 7cf28e2f03
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
awx/main/migrations/_rbac.py
View File

@@ -1,5 +1,3 @@
from django.db import connection, transaction, reset_queries
from django.db.transaction import TransactionManagementError
from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.models import ContentType
from collections import defaultdict from collections import defaultdict
@@ -150,7 +148,6 @@ def migrate_projects(apps, schema_editor):
project.organization = first_org project.organization = first_org
project.save() project.save()
else: else:
print('Fork to %s ' % (org.name + ' - ' + original_project_name))
new_prj = Project.objects.create( new_prj = Project.objects.create(
created = project.created, created = project.created,
description = project.description, description = project.description,
@@ -172,33 +169,32 @@ def migrate_projects(apps, schema_editor):
job_templates = JobTemplate.objects.filter(inventory__organization=org).all() job_templates = JobTemplate.objects.filter(inventory__organization=org).all()
for jt in job_templates: for jt in job_templates:
jt.project = new_prj jt.project = new_prj
print('Updating jt to point to %s' % repr(new_prj))
jt.save() jt.save()
# Migrate permissions # Migrate permissions
for project in [p for p in Project.objects.all()]: for project in [p for p in Project.objects.all()]:
if project.organization is not None and project.created_by is not None: if project.organization is None and project.created_by is not None:
project.admin_role.members.add(project.created_by) project.admin_role.members.add(project.created_by)
migrations[original_project_name]['users'].add(project.created_by) migrations[project.name]['users'].add(project.created_by)
for team in project.teams.all(): for team in project.teams.all():
team.member_role.children.add(project.member_role) team.member_role.children.add(project.member_role)
migrations[original_project_name]['teams'].add(team) migrations[project.name]['teams'].add(team)
if project.organization is not None: if project.organization is not None:
for user in project.organization.deprecated_users.all(): for user in project.organization.deprecated_users.all():
project.member_role.members.add(user) project.member_role.members.add(user)
migrations[original_project_name]['users'].add(user) migrations[project.name]['users'].add(user)
for perm in Permission.objects.filter(project=project, active=True): for perm in Permission.objects.filter(project=project, active=True):
# All perms at this level just imply a user or team can read # All perms at this level just imply a user or team can read
if perm.team: if perm.team:
perm.team.member_role.children.add(project.member_role) perm.team.member_role.children.add(project.member_role)
migrations[original_project_name]['teams'].add(perm.team) migrations[project.name]['teams'].add(perm.team)
if perm.user: if perm.user:
project.member_role.members.add(perm.user) project.member_role.members.add(perm.user)
migrations[original_project_name]['users'].add(perm.user) migrations[project.name]['users'].add(perm.user)
return migrations return migrations