mirror of
https://github.com/ceph/ceph
synced 2025-01-01 16:42:29 +00:00
26f00fc541
Put each yaml in the job_description on its own line so not so wide of a line. Make delete default None not False in function. Signed-off-by: Sandon Van Ness <sandon@inktank.com>
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import beanstalkc
|
|
import yaml
|
|
import logging
|
|
|
|
from .config import config
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
def beanstalk_connect(machine_type):
|
|
qhost = config.queue_host
|
|
qport = config.queue_port
|
|
if qhost is None or qport is None:
|
|
raise RuntimeError(
|
|
'Beanstalk queue information not found in {conf_path}'.format(
|
|
conf_path=config.teuthology_yaml))
|
|
log.info("Checking Beanstalk Queue...")
|
|
beanstalk = beanstalkc.Connection(host=qhost, port=qport)
|
|
beanstalk.watch(machine_type)
|
|
beanstalk.ignore('default')
|
|
return beanstalk
|
|
|
|
def walk_jobs(beanstalk, machine_type, delete=None):
|
|
curjobs = beanstalk.stats_tube(machine_type)['current-jobs-ready']
|
|
if curjobs != 0:
|
|
x=1
|
|
while x != curjobs:
|
|
x += 1
|
|
job = beanstalk.reserve(timeout=20)
|
|
if job is not None:
|
|
if job.body is not None:
|
|
job_config = yaml.safe_load(job.body)
|
|
job_name=job_config['name']
|
|
job_id = job.stats()['id']
|
|
job_description = job_config['description']
|
|
if delete:
|
|
if delete in job_name:
|
|
print 'Deleted Job: {job_id} from queue. Name: {job_name}'.format(
|
|
job_id = job_id,
|
|
job_name = job_name,
|
|
)
|
|
job.delete()
|
|
else:
|
|
print "Searching queue... Checked " + str(x) + "/" + str(curjobs)," Jobs\r",
|
|
else:
|
|
print 'Job: {x}/{curjobs} ID: {job_id} Name: {job_name}'.format(
|
|
x=x,
|
|
curjobs=curjobs,
|
|
job_id=job_id,
|
|
job_name=job_name,
|
|
)
|
|
if job_description:
|
|
for desc in job_description.split():
|
|
print '\t {desc}'.format(desc=desc)
|
|
log.info("Finished checking Beanstalk Queue.")
|
|
else:
|
|
log.info('No jobs in Beanstalk Queue')
|
|
|
|
def main(args):
|
|
machine_type = args['--machine_type']
|
|
delete = args['--delete']
|
|
beanstalk = beanstalk_connect(machine_type)
|
|
walk_jobs(beanstalk, machine_type, delete)
|
|
beanstalk.close()
|