Current File : //usr/lib/python2.7/site-packages/salt/daemons/flo/maint.py
# -*- coding: utf-8 -*-
'''
Define the behaviors used in the maintenance process
'''
# pylint: disable=3rd-party-module-not-gated
from __future__ import absolute_import, print_function, unicode_literals
# Import python libs
import multiprocessing
import os
# Import ioflo libs
import ioflo.base.deeding
# Import salt libs
import salt.daemons.masterapi
import salt.fileserver
import salt.loader
import salt.utils.minions
import salt.utils.stringutils
@ioflo.base.deeding.deedify(
salt.utils.stringutils.to_str('SaltRaetMaintFork'),
ioinits={'opts': salt.utils.stringutils.to_str('.salt.opts'),
'proc_mgr': salt.utils.stringutils.to_str('.salt.usr.proc_mgr')})
def maint_fork(self):
'''
For off the maintinence process from the master router process
FloScript:
do salt raet maint fork at enter
'''
self.proc_mgr.value.add_process(Maintenance, args=(self.opts.value,))
class Maintenance(multiprocessing.Process):
'''
Start the maintinance process within ioflo
'''
def __init__(self, opts):
super(Maintenance, self).__init__()
self.opts = opts
def run(self):
'''
Spin up a worker, do this in s multiprocess
'''
behaviors = ['salt.daemons.flo']
preloads = [(salt.utils.stringutils.to_str('.salt.opts'), dict(value=self.opts))]
console_logdir = self.opts.get('ioflo_console_logdir', '')
if console_logdir:
consolepath = os.path.join(console_logdir, 'maintenance.log')
else: # empty means log to std out
consolepath = ''
ioflo.app.run.start(
name='maintenance',
period=float(self.opts['loop_interval']),
stamp=0.0,
real=self.opts['ioflo_realtime'],
filepath=self.opts['maintenance_floscript'],
behaviors=behaviors,
username="",
password="",
mode=None,
houses=None,
metas=None,
preloads=preloads,
verbose=int(self.opts['ioflo_verbose']),
consolepath=consolepath,
)
class SaltRaetMaintSetup(ioflo.base.deeding.Deed):
'''
Init loader objects used
FloScript:
do salt raet maint setup at enter
'''
Ioinits = {'opts': salt.utils.stringutils.to_str('.salt.opts'),
'fileserver': salt.utils.stringutils.to_str('.salt.loader.fileserver'),
'runners': salt.utils.stringutils.to_str('.salt.loader.runners'),
'pillargitfs': salt.utils.stringutils.to_str('.salt.loader.pillargitfs'),
'ckminions': salt.utils.stringutils.to_str('.salt.loader.ckminions')}
def action(self):
'''
Set up the objects used in the maint process
'''
self.fileserver.value = salt.fileserver.Fileserver(self.opts.value)
self.runners.value = salt.loader.runner(self.opts.value)
self.ckminions.value = salt.utils.minions.CkMinions(self.opts.value)
self.pillargitfs.value = salt.daemons.masterapi.init_git_pillar(
self.opts.value)
class SaltRaetMaintFileserverClean(ioflo.base.deeding.Deed):
'''
Clear the fileserver backend caches
FloScript:
do salt raet maint fileserver clean at enter
'''
Ioinits = {'opts': salt.utils.stringutils.to_str('.salt.opts')}
def action(self):
'''
Clean!
'''
salt.daemons.masterapi.clean_fsbackend(self.opts.value)
class SaltRaetMaintOldJobsClear(ioflo.base.deeding.Deed):
'''
Iterate over the jobs directory and clean out the old jobs
FloScript:
do salt raet maint old jobs clear
'''
Ioinits = {'opts': salt.utils.stringutils.to_str('.salt.opts')}
def action(self):
'''
Clear out the old jobs cache
'''
salt.daemons.masterapi.clean_old_jobs(self.opts.value)
class SaltRaetMaintBackendsUpdate(ioflo.base.deeding.Deed):
'''
Update the fileserver and external pillar caches
FloScript:
do salt raet maint backends update
'''
Ioinits = {'opts': salt.utils.stringutils.to_str('.salt.opts'),
'fileserver': salt.utils.stringutils.to_str('.salt.loader.fileserver'),
'pillargitfs': salt.utils.stringutils.to_str('.salt.loader.pillargitfs')}
def action(self):
'''
Update!
'''
for pillargit in self.pillargitfs.value:
pillargit.update()
salt.daemons.masterapi.fileserver_update(self.fileserver.value)
Mini Shell