#!/usr/bin/env python
'Requeue pending scenarios'
# Import custom modules
import script_process
from np import model
from np.model import Session
# If we are running the command as a script,
if __name__ == '__main__':
# Connect
script_process.connect()
# List
for processorIP, processorWhen in Session.query(model.Processor.ip, model.Processor.when_updated).order_by(model.Processor.when_updated):
print '{}\t{}'.format(processorIP, processorWhen.strftime('%Y%m%d %H:%M'))
# Commit
Session.commit()
""" Save scenario leve inputs """
scenarioFolder = scenario.getFolder()
store.makeFolderSafely(scenarioFolder)
expandPath = lambda x: os.path.join(scenarioFolder, x)
metricConfiguration = scenario.input['metric configuration']
metric.saveMetricsConfigurationCSV(expandPath('metrics-job-input'), metricConfiguration)
store.zipFolder(scenarioFolder + '.zip', scenarioFolder)
# If the user is running the script from the command-line,
if __name__ == '__main__':
# Connect (get config and setup model from appropriate DB)
configuration = script_process.connect()
config['storage_path'] = configuration.get('app:main', 'storage_path')
# get ids from stdin into a list
ids = sys.stdin.readlines()
# Iterate through scenarios saving input files
scenarios = Session.query(model.Scenario).\
filter(and_(model.Scenario.id.in_(ids),
(model.Scenario.status == model.statusDone))).\
order_by(model.Scenario.id)
for scenario in scenarios:
print "saving metric inputs for scenario id: %s" % scenario.id
save_job_input_file(scenario)
"""
Reset database
"""
# Import custom modules
import script_process
from fp.model import meta
def run():
'Reset database'
# Connect
meta.metadata.bind = meta.engine
# Reflect
meta.metadata.reflect()
# Clear database
meta.metadata.drop_all()
meta.metadata.create_all()
# Return
return 'Reset'
# If we are running standalone,
if __name__ == '__main__':
# Parse
optionParser = script_process.buildOptionParser()
options, arguments = optionParser.parse_args()
# Initialize
script_process.connect(options)
# Run
print run()
