def view_tag(self, request, **kwargs):
self.method_check(request, allowed=['get'])
self.is_authenticated(request)
self.throttle_check(request)
tag_name = kwargs['tag_name']
if not tag_name:
return self.create_response(request, {'result': "Empty tag name"})
ami_wrapper = AMIWrapper()
tag = ami_wrapper.get_ami_tag(tag_name)
return self.create_response(request, {'name': tag_name, 'body': tag})
def test_ami_wrapper(self):
from taskengine.atlas.datamgmt import AMIWrapper
try:
ami_wrapper = AMIWrapper()
except Exception as ex:
print "test_ami_wrapper, exception occurred (1): %s" % str(ex)
ami_wrapper = None
self.assertNotEqual(ami_wrapper, None)
self.assertNotEqual(ami_wrapper.ami_client, None)
ctag_name = 'r5555'
try:
ctag = ami_wrapper.get_ami_tag(ctag_name)
except Exception as ex:
print "test_ami_wrapper, exception occurred (2): %s" % str(ex)
ctag = None
self.assertNotEqual(ctag, None)
self.assertEqual(str(ctag['tag']), str(ctag_name))
def handle(self, *args, **options):
if options['worker_name'] == 'process_requests':
task_def = TaskDefinition(debug_mode=False)
task_def.process_requests(restart=False, no_wait=False)
elif options['worker_name'] == 'sync_ami_projects':
ami_wrapper = AMIWrapper()
ami_wrapper.sync_ami_projects()
elif options['worker_name'] == 'sync_ami_types':
ami_wrapper = AMIWrapper()
ami_wrapper.sync_ami_types()
elif options['worker_name'] == 'sync_ami_phys_containers':
ami_wrapper = AMIWrapper()
ami_wrapper.sync_ami_phys_containers()
elif options['worker_name'] == 'sync_ami_tags':
ami_wrapper = AMIWrapper()
ami_wrapper.sync_ami_tags()
elif options['worker_name'] == 'create_task_input':
from debug import create_task_input
create_task_input()
elif options['worker_name'] == 'define_task_simulation_type':
from debug import define_task_simulation_type
define_task_simulation_type()
elif options['worker_name'] == 'search_task_duplicates':
from debug import search_task_duplicates
search_task_duplicates()
