External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-11 18:09:57 -03:30
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #330 from chrismeyersfsu/fix-update_instance

Browse Source 
change update_instance to only take --primary role flag
This commit is contained in:
Chris Meyers 2015-07-31 15:39:42 -04:00
commit e98ee0b428
2 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
awx/main/management/commands/_base_instance.py
View File

@ -70,8 +70,13 @@ class BaseCommandInstance(BaseCommand):
default='',
help='Find instance by specified uuid.')
def include_option_primary_role(self):
BaseCommand.option_list += ( BaseCommandInstance.generate_option_primary(), )
self.enforce_primary_role = True
def include_options_roles(self):
BaseCommand.option_list += ( BaseCommandInstance.generate_option_primary(), BaseCommandInstance.generate_option_secondary(), )
self.include_option_primary_role()
BaseCommand.option_list += ( BaseCommandInstance.generate_option_secondary(), )
self.enforce_roles = True
def include_option_hostname_set(self):
@ -107,6 +112,8 @@ class BaseCommandInstance(BaseCommand):
return CommandError('--hostname and one of --primary or --secondary is required.')
elif self.enforce_hostname_set:
return CommandError('--hostname is required.')
elif self.enforce_primary_role:
return CommandError('--primary is required.')
elif self.enforce_roles:
return CommandError('One of --primary or --secondary is required.')
@ -120,6 +127,11 @@ class BaseCommandInstance(BaseCommand):
if self.is_option_primary() and self.is_option_secondary() or not (self.is_option_primary() or self.is_option_secondary()):
raise self.usage_error
elif self.enforce_primary_role:
if options['primary']:
self.option_primary = options['primary']
else:
raise self.usage_error
if self.enforce_hostname_set:
if options['hostname']:

2
awx/main/management/commands/update_instance.py
View File

@ -26,7 +26,7 @@ class Command(BaseCommandInstance):
def __init__(self):
super(Command, self).__init__()
self.include_options_roles()
self.include_option_primary_role()
self.include_option_hostname_uuid_find()
@transaction.atomic