Python TaskQuery.environment Examples

Example #1

 def get_tasks_status_query(self, instance_ids):
     query = TaskQuery()
     query.owner = Identity(role=self._role)
     query.environment = self._env
     query.jobName = self._name
     query.statuses = set([ScheduleStatus.RUNNING])
     query.instanceIds = set(instance_ids)
     return query

Example #2

File: instance_watcher.py Project: sathiya-mit/incubator-aurora

 def _create_query(self, instance_ids):
   query = TaskQuery()
   query.owner = Identity(role=self._job_key.role)
   query.environment = self._job_key.environment
   query.jobName = self._job_key.name
   query.statuses = set([ScheduleStatus.RUNNING])
   query.instanceIds = instance_ids
   return query

Example #3

 def _create_query(self, instance_ids):
     query = TaskQuery()
     query.owner = Identity(role=self._job_key.role)
     query.environment = self._job_key.environment
     query.jobName = self._job_key.name
     query.statuses = set([ScheduleStatus.RUNNING])
     query.instanceIds = instance_ids
     return query

Example #4

File: test_instance_watcher.py Project: bhuvan/incubator-aurora

 def get_tasks_status_query(self, instance_ids):
   query = TaskQuery()
   query.owner = Identity(role=self._role)
   query.environment = self._env
   query.jobName = self._name
   query.statuses = set([ScheduleStatus.RUNNING])
   query.instanceIds = set(instance_ids)
   return query

Example #5

File: admin.py Project: apache/aurora

def prune_tasks(args, options):
  if len(args) == 0:
    die('Must specify at least cluster.')
  cluster = args[0]

  t = TaskQuery()
  if options.states:
    t.statuses = set(map(ScheduleStatus._NAMES_TO_VALUES.get, options.states.split(',')))
  if options.role:
    t.role = options.role
  if options.environment:
    t.environment = options.environment
  if options.limit:
    t.limit = options.limit

  api = make_admin_client_with_options(cluster)
  rsp = api.prune_tasks(t)
  if rsp.responseCode != ResponseCode.OK:
    die('Failed to prune tasks: %s' % combine_messages(rsp))
  else:
    print("Tasks pruned.")

Example #6

File: instance_watcher.py Project: dhardy92/incubator-aurora

  def _get_tasks_by_instance_id(self, instance_ids):
    log.debug('Querying instance statuses.')
    query = TaskQuery()
    query.owner = Identity(role=self._job_key.role)
    query.environment = self._job_key.environment
    query.jobName = self._job_key.name
    query.statuses = set([ScheduleStatus.RUNNING])

    query.instanceIds = instance_ids
    try:
      resp = self._scheduler.getTasksStatus(query)
    except IOError as e:
      log.error('IO Exception during scheduler call: %s' % e)
      return []

    tasks = []
    if resp.responseCode == ResponseCode.OK:
      tasks = resp.result.scheduleStatusResult.tasks

    log.debug('Response from scheduler: %s (message: %s)'
        % (ResponseCode._VALUES_TO_NAMES[resp.responseCode], resp.messageDEPRECATED))
    return tasks

Example #7

File: admin.py Project: isabella232/client-3

def prune_tasks(args, options):
    if len(args) == 0:
        die('Must specify at least cluster.')
    cluster = args[0]

    t = TaskQuery()
    if options.states:
        t.statuses = set(
            map(ScheduleStatus._NAMES_TO_VALUES.get,
                options.states.split(',')))
    if options.role:
        t.role = options.role
    if options.environment:
        t.environment = options.environment
    if options.limit:
        t.limit = options.limit

    api = make_admin_client_with_options(cluster)
    rsp = api.prune_tasks(t)
    if rsp.responseCode != ResponseCode.OK:
        die('Failed to prune tasks: %s' % combine_messages(rsp))
    else:
        print("Tasks pruned.")