def get_requests():
'''uses credentials from the current conf to query rt for acct reset
ticks. returns list of ticks.'''
credentials = dict()
credentials['user'] = CONF['rt_username']
credentials['pass'] = CONF['rt_password']
return rt.get(CONF['rt_query'], credentials, CONF['rt_search'])
def main():
"""Load a config, load modules, get tasks from RT dispatch disable tasks
for each of them."""
with open('./config.yaml', 'r') as config_file:
config = yaml.load(config_file)
logging.basicConfig(
format='[account-disable %(levelname)s] %(asctime)s: %(message)s',
level=config['log_level'],
)
credentials = dict()
credentials['user'] = config['rt_username']
credentials['pass'] = config['rt_password']
for module in IMPLEMENTATIONS:
parent = "disableables.{0}".format(module)
fhn, filename, description = find_module(module, disableables.__path__)
vars()[module] = load_module(parent, fhn, filename, description)
# get reset requests from rt
query, uri = (config['rt_query'], config['rt_uri'])
for reset in rt.get(query, credentials, uri):
ticket, uid = reset
logging.info('Working on ticket=%s for uid=%s', ticket, uid)
tasks = dict()
for module in IMPLEMENTATIONS:
instance = vars()[module]
instance_config = dict()
for key in instance.CONFIG:
instance_config[key] = config[key]
task = instance.disable.delay(ticket, uid, config)
tasks[ticket] = (uid, task)
print "RT#{0}: {1} is {2}".format(ticket, uid, task.status)
while True:
if len(tasks) < 1: break
done = list()
for ticket in tasks.iterkeys():
uid, task = tasks[ticket]
if task.successful():
comment = "RT#{0}: Success!\n{1}".format(ticket, task.get())
#rt.comment(ticket, comment, credentials, uri)
#rt.move(ticket, 'cis-windows', credentials, uri)
done.append(ticket)
print comment
if task.failed():
comment = "RT#{0}: Failed!\n{1}".format(ticket, task.get())
# but we don't do retry yet... so...
done.append(ticket)
# can't delete dict items during iteration, causes runtime error
for ticket in done:
del tasks[ticket]
