def setup(self):
try:
new_attrs = {}
new_attrs.update(self.attrs)
new_attrs.update(self.do_get_dependencies())
self.obj = self.klass(**new_attrs)
DBSession.add(self.obj)
DBSession.flush()
return self.obj
except:
DBSession.rollback()
raise
def exec_task(self, auth, ctx):
manager = Basic.getGridManager()
groups = manager.getGroupList(auth)
for group in groups:
nodes = manager.getNodeList(auth, group.id)
for n in nodes:
try:
transaction.begin()
n=DBSession.query(ManagedNode).filter(ManagedNode.id==n.id).one()
n.refresh_environ()
n.get_running_vms()
n.socket=n.get_socket_info()
DBSession.add(n)
transaction.commit()
except Exception, e:
LOGGER.error(to_str(e))
DBSession.rollback()
def exec_task(self, auth, ctx):
manager = Basic.getGridManager()
groups = manager.getGroupList(auth)
for group in groups:
nodes = manager.getNodeList(auth, group.id)
for n in nodes:
try:
transaction.begin()
n=DBSession.query(ManagedNode).filter(ManagedNode.id==n.id).one()
n.refresh_environ()
n.get_running_vms()
n.socket=n.get_socket_info()
if n.is_up():
n.isHVM = n.is_HVM()
DBSession.add(n)
transaction.commit()
except Exception, e:
LOGGER.error(to_str(e))
DBSession.rollback()
class RefreshNodeInfoTask(Task):
def get_descriptions(self):
short_desc = m_("Refresh Task for All Nodes")
return (short_desc, (), short_desc, ())
def exec_task(self, auth, ctx):
manager = Basic.getGridManager()
groups = manager.getGroupList(auth)
for group in groups:
nodes = manager.getNodeList(auth, group.id)
for n in nodes:
try:
transaction.begin()
n=DBSession.query(ManagedNode).filter(ManagedNode.id==n.id).one()
n.refresh_environ()
n.get_running_vms()
n.socket=n.get_socket_info()
if n.is_up():
n.isHVM = n.is_HVM()
DBSession.add(n)
transaction.commit()
except Exception, e:
LOGGER.error(to_str(e))
DBSession.rollback()
ungrouped_nodes = manager.getNodeList(auth)
for n in ungrouped_nodes:
try:
transaction.begin()
n=DBSession.query(ManagedNode).filter(ManagedNode.id==n.id).one()
n.refresh_environ()
n.get_running_vms()
n.socket=n.get_socket_info()
DBSession.add(n)
transaction.commit()
except Exception, e:
LOGGER.error(to_str(e))
DBSession.rollback()
def tearDown(self):
DBSession.rollback()
