shaba/awx
1
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2024-10-29 20:55:32 +03:00
Code Releases Activity

safer migrations

Browse Source
This commit is contained in:
Chris Meyers 2020-08-03 09:50:02 -04:00 committed by Ryan Petrello
parent b7efad5640
commit 2eec1317bd
No known key found for this signature in database
GPG Key ID: F2AA5F2122351777
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
awx/main/migrations/0118_v380_inventory_plugins.py
View File

@ -5,6 +5,8 @@ import json
from django.db import migrations
from awx.main.models.inventory import InventorySourceOptions
from ._inventory_source_vars import FrozenInjectors
@ -12,8 +14,20 @@ logger = logging.getLogger('awx.main.migrations')
BACKUP_FILENAME = '/tmp/tower_migration_inventory_source_vars.json'
class InventorySourceOptionsWrapper(InventorySourceOptions):
'''
InventorySource inherits from InventorySourceOptions but that is not
"recorded" by Django's app registry model tracking. This will, effectively,
reintroduce the inheritance.
'''
def __init__(self, *args, **kw):
self.target = kw.pop('target')
super().__init__(self, *args, **kw)
def __getattr__(self, attr):
return getattr(self.target, attr)
def _get_inventory_sources(InventorySource):
# TODO: Maybe pull the list of cloud sources from code
return InventorySource.objects.filter(source__in=['ec2', 'gce', 'azure_rm', 'vmware', 'satellite6', 'openstack', 'rhv', 'tower'])
@ -22,7 +36,7 @@ def inventory_source_vars_forward(apps, schema_editor):
source_vars_backup = dict()
for inv_source_obj in _get_inventory_sources(InventorySource):
# TODO: Log error if this is false, it shouldn't be false
inv_source_obj = InventorySourceOptionsWrapper(target=inv_source_obj)
if inv_source_obj.source in FrozenInjectors:
source_vars_backup[inv_source_obj.id] = dict(inv_source_obj.source_vars_dict)
with open(BACKUP_FILENAME, 'w') as fh: