110 lines
3.2 KiB
Python
Executable File
110 lines
3.2 KiB
Python
Executable File
#!/usr/bin/python
|
|
"""main script to run as service to update hosts installing progress."""
|
|
import logging
|
|
import signal
|
|
import sys
|
|
import time
|
|
import daemon
|
|
|
|
from compass.actions import progress_update
|
|
from compass.db import database
|
|
from compass.db.model import Cluster
|
|
from compass.tasks.client import celery
|
|
from compass.utils import flags
|
|
from compass.utils import logsetting
|
|
from compass.utils import setting_wrapper as setting
|
|
|
|
|
|
flags.add('clusterids',
|
|
help='comma seperated cluster ids',
|
|
default='')
|
|
flags.add_bool('async',
|
|
help='ryn in async mode',
|
|
default=True)
|
|
flags.add_bool('once',
|
|
help='run once or forever',
|
|
default=False)
|
|
flags.add('run_interval',
|
|
help='run interval in seconds',
|
|
default=setting.PROGRESS_UPDATE_INTERVAL)
|
|
flags.add_bool('daemonize',
|
|
help='run as daemon',
|
|
default=False)
|
|
|
|
|
|
BUSY = False
|
|
KILLED = False
|
|
|
|
def handle_term(signum, frame):
|
|
global BUSY
|
|
global KILLED
|
|
logging.info('Caught signal %s', signum)
|
|
KILLED = True
|
|
if not BUSY:
|
|
sys.exit(0)
|
|
|
|
|
|
def main(argv):
|
|
"""entry function."""
|
|
global BUSY
|
|
global KILLED
|
|
clusterids = [
|
|
int(clusterid) for clusterid in flags.OPTIONS.clusterids.split(',')
|
|
if clusterid
|
|
]
|
|
signal.signal(signal.SIGINT, handle_term)
|
|
|
|
while True:
|
|
BUSY = True
|
|
with database.session() as session:
|
|
if not clusterids:
|
|
clusters = session.query(Cluster).all()
|
|
update_clusterids = [cluster.id for cluster in clusters]
|
|
else:
|
|
update_clusterids = clusterids
|
|
|
|
logging.info('update progress for clusters: %s', update_clusterids)
|
|
for clusterid in update_clusterids:
|
|
if flags.OPTIONS.async:
|
|
celery.send_task('compass.tasks.progress_update', (clusterid,))
|
|
else:
|
|
try:
|
|
progress_update.update_progress(clusterid)
|
|
except Exception as error:
|
|
logging.error('failed to update progress for cluster %s',
|
|
clusterid)
|
|
logging.exception(error)
|
|
pass
|
|
|
|
BUSY = False
|
|
if KILLED:
|
|
logging.info('exit progress update loop')
|
|
break
|
|
|
|
if flags.OPTIONS.once:
|
|
logging.info('trigger installer finsished')
|
|
break
|
|
|
|
if flags.OPTIONS.run_interval > 0:
|
|
logging.info('will rerun the trigger install after %s',
|
|
flags.OPTIONS.run_interval)
|
|
time.sleep(flags.OPTIONS.run_interval)
|
|
else:
|
|
logging.info('rerun the trigger installer immediately')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
flags.init()
|
|
logsetting.init()
|
|
logging.info('run progress update: %s', sys.argv)
|
|
if flags.OPTIONS.daemonize:
|
|
with daemon.DaemonContext(
|
|
pidfile=lockfile.FileLock('/var/run/progress_update.pid'),
|
|
stderr=open('/tmp/progress_update_err.log', 'w+'),
|
|
stdout=open('/tmp/progress_update_out.log', 'w+')
|
|
):
|
|
logging.info('run progress update as daemon')
|
|
main(sys.argv)
|
|
else:
|
|
main(sys.argv)
|