mirror of
https://github.com/ZwareBear/awx.git
synced 2026-03-27 11:13:37 -05:00
0c1e1fa2fb
* Gut the HA middleware * Purge concept of primary and secondary. * UUID is not the primary host identifier, now it's based mostly on the username. Some work probably still left to do to make sure this is legit. Also removed unique constraint from the uuid field. This might become the cluster ident now... or it may just deprecate * No more secondary -> primary redirection * Initial revision of /api/v1/ping * Revise and gut tower-manage register_instance * Rename awx/main/socket.py to awx/main/socket_queue.py to prevent conflict with the "socket" module from python base * Revist/gut the Instance manager... not sure if this manager is really needed anymore
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
# Copyright (c) 2015 Ansible, Inc.
|
|
# All Rights Reserved.
|
|
|
|
import socket
|
|
from optparse import make_option
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.conf import settings
|
|
|
|
from awx.main.models import Project
|
|
|
|
|
|
class OptionEnforceError(Exception):
|
|
def __init__(self, value):
|
|
self.value = value
|
|
|
|
def __str__(self):
|
|
return repr(self.value)
|
|
|
|
class BaseCommandInstance(BaseCommand):
|
|
#option_list = BaseCommand.option_list
|
|
|
|
def __init__(self):
|
|
super(BaseCommandInstance, self).__init__()
|
|
self.enforce_hostname_set = False
|
|
self.enforce_unique_find = False
|
|
|
|
self.option_hostname = None
|
|
self.option_uuid = None
|
|
|
|
self.UUID = settings.SYSTEM_UUID
|
|
self.unique_fields = {}
|
|
|
|
@staticmethod
|
|
def generate_option_hostname():
|
|
return make_option('--hostname',
|
|
dest='hostname',
|
|
default=socket.gethostname(),
|
|
help='Find instance by specified hostname.')
|
|
|
|
@staticmethod
|
|
def generate_option_hostname_set():
|
|
return make_option('--hostname',
|
|
dest='hostname',
|
|
default=socket.gethostname(),
|
|
help='Hostname to assign to the new instance.')
|
|
|
|
@staticmethod
|
|
def generate_option_uuid():
|
|
#TODO: Likely deprecated, maybe uuid becomes the cluster ident?
|
|
return make_option('--uuid',
|
|
dest='uuid',
|
|
default='',
|
|
help='Find instance by specified uuid.')
|
|
|
|
def include_option_hostname_set(self):
|
|
BaseCommand.option_list += ( BaseCommandInstance.generate_option_hostname_set(), )
|
|
self.enforce_hostname_set = True
|
|
|
|
def include_option_hostname_uuid_find(self):
|
|
BaseCommand.option_list += ( BaseCommandInstance.generate_option_hostname(), BaseCommandInstance.generate_option_uuid(), )
|
|
self.enforce_unique_find = True
|
|
|
|
def get_option_hostname(self):
|
|
return self.option_hostname
|
|
|
|
def get_option_uuid(self):
|
|
return self.option_uuid
|
|
|
|
def get_UUID(self):
|
|
return self.UUID
|
|
|
|
# for the enforce_unique_find policy
|
|
def get_unique_fields(self):
|
|
return self.unique_fields
|
|
|
|
@property
|
|
def usage_error(self):
|
|
if self.enforce_hostname_set:
|
|
return CommandError('--hostname is required.')
|
|
|
|
def handle(self, *args, **options):
|
|
if self.enforce_hostname_set and self.enforce_unique_find:
|
|
raise OptionEnforceError('Can not enforce --hostname as a setter and --hostname as a getter')
|
|
|
|
if self.enforce_hostname_set:
|
|
if options['hostname']:
|
|
self.option_hostname = options['hostname']
|
|
else:
|
|
raise self.usage_error
|
|
|
|
if self.enforce_unique_find:
|
|
if options['hostname']:
|
|
self.unique_fields['hostname'] = self.option_hostname = options['hostname']
|
|
|
|
if options['uuid']:
|
|
self.unique_fields['uuid'] = self.option_uuid = options['uuid']
|
|
|
|
if len(self.unique_fields) == 0:
|
|
self.unique_fields['uuid'] = self.get_UUID()
|
|
|
|
@staticmethod
|
|
def __instance_str(instance, fields):
|
|
string = '('
|
|
for field in fields:
|
|
string += '%s="%s",' % (field, getattr(instance, field))
|
|
if len(fields) > 0:
|
|
string = string[:-1]
|
|
string += ')'
|
|
return string
|
|
|
|
@staticmethod
|
|
def instance_str(instance):
|
|
return BaseCommandInstance.__instance_str(instance, ('uuid', 'hostname'))
|