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()