# (c) Copyright 2014-2016 Hewlett-Packard Development Company, L.P. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import logging import pprint from cliff import command from cliff import lister from cliff import show from freezerclient import exceptions from freezerclient import utils logging = logging.getLogger(__name__) class JobShow(show.ShowOne): """Show a single job""" def get_parser(self, prog_name): parser = super(JobShow, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') return parser def take_action(self, parsed_args): job = self.app.client.jobs.get(parsed_args.job_id) if not job: raise exceptions.ApiClientException('Job not found') column = ( 'Job ID', 'Client ID', 'User ID', 'Session ID', 'Description', 'Actions', 'Start Date', 'End Date', 'Interval', 'Status', 'Result', 'Current pid', 'Event', ) data = ( job.get('job_id'), job.get('client_id'), job.get('user_id'), job.get('session_id', ''), job.get('description'), pprint.pformat(job.get('job_actions')), job.get('job_schedule', {}).get('schedule_start_date', ''), job.get('job_schedule', {}).get('schedule_end_date', ''), job.get('job_schedule', {}).get('schedule_interval', ''), job.get('job_schedule', {}).get('status', ''), job.get('job_schedule', {}).get('result', ''), job.get('job_schedule', {}).get('current_pid', ''), job.get('job_schedule', {}).get('event', ''), ) return column, data class JobList(lister.Lister): """List all the jobs for your user""" def get_parser(self, prog_name): parser = super(JobList, self).get_parser(prog_name) parser.add_argument( '--limit', dest='limit', default=100, help='Specify a limit for search query', ) parser.add_argument( '--offset', dest='offset', default=0, help='', ) parser.add_argument( '--search', dest='search', default={}, help='Define a filter for the query', ) parser.add_argument( '--client', '-C', dest='client_id', default='', help='Get jobs for a specific client', ) return parser def take_action(self, parsed_args): search = utils.prepare_search(parsed_args.search) if parsed_args.client_id: jobs = self.app.client.jobs.list( limit=parsed_args.limit, offset=parsed_args.offset, search=search, client_id=parsed_args.client_id ) else: jobs = self.app.client.jobs.list_all( limit=parsed_args.limit, offset=parsed_args.offset, search=search ) columns = ('Job ID', 'Description', '# Actions', 'Result', 'Status', 'Event', 'Session ID') # Print empty table if no jobs found if not jobs: jobs = [{}] data = ((job.get('job_id', ''), job.get('description', ''), job.get('job_actions', ''), job.get('job_schedule', {}).get('result', ''), job.get('job_schedule', {}).get('status', ''), job.get('job_schedule', {}).get('event', ''), job.get('session_id', '') ) for job in jobs) else: data = ((job.get('job_id'), job.get('description'), len(job.get('job_actions', [])), job.get('job_schedule', {}).get('result', ''), job.get('job_schedule', {}).get('status', ''), job.get('job_schedule', {}).get('event', ''), job.get('session_id', '') ) for job in jobs) return columns, data class JobGet(command.Command): """Download a job as a json file""" def get_parser(self, prog_name): parser = super(JobGet, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') parser.add_argument('--no-format', dest='no_format', default=False, action='store_true', help='Return a job in json without pretty print') return parser def take_action(self, parsed_args): job = self.app.client.jobs.get(parsed_args.job_id) if not job: raise exceptions.ApiClientException('Job not found') if parsed_args.no_format: print(job) else: pprint.pprint(job) class JobDelete(command.Command): """Delete a job from the api""" def get_parser(self, prog_name): parser = super(JobDelete, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') return parser def take_action(self, parsed_args): self.app.client.jobs.delete(parsed_args.job_id) logging.info('Job {0} deleted'.format(parsed_args.job_id)) class JobCreate(command.Command): """Create a new job from a file""" def get_parser(self, prog_name): parser = super(JobCreate, self).get_parser(prog_name) parser.add_argument( '--file', dest='file', required=True, help='Path to json file with the job') parser.add_argument( '--client', '-C', dest='client_id', required=True, help='Select a client for this job', ) return parser def take_action(self, parsed_args): job = utils.doc_from_json_file(parsed_args.file) job['client_id'] = parsed_args.client_id job_id = self.app.client.jobs.create(job) logging.info('Job {0} created'.format(job_id)) class JobStart(command.Command): """Send a start signal for a job""" def get_parser(self, prog_name): parser = super(JobStart, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') return parser def take_action(self, parsed_args): self.app.client.jobs.start_job(parsed_args.job_id) logging.info("Start request sent " "for job {0}".format(parsed_args.job_id)) class JobStop(command.Command): """Send a stop signal for a job""" def get_parser(self, prog_name): parser = super(JobStop, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') return parser def take_action(self, parsed_args): self.app.client.jobs.stop_job(parsed_args.job_id) logging.info("Stop request sent " "for job {0}".format(parsed_args.job_id)) class JobAbort(command.Command): """Abort a running job""" def get_parser(self, prog_name): parser = super(JobAbort, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') return parser def take_action(self, parsed_args): self.app.client.jobs.abort_job(parsed_args.job_id) logging.info("Abort request sent " "for job {0}".format(parsed_args.job_id)) class JobUpdate(command.Command): """Update a job from a file""" def get_parser(self, prog_name): parser = super(JobUpdate, self).get_parser(prog_name) parser.add_argument(dest='job_id', help='ID of the job') parser.add_argument(dest='file', help='Path to json file with the job') return parser def take_action(self, parsed_args): job = utils.doc_from_json_file(parsed_args.file) self.app.client.jobs.update(parsed_args.job_id, job) logging.info('Job {0} updated'.format(parsed_args.job_id))