class ClientManager(CommandBoxWindow):
def __init__(self, conn, name='ClientManager'):
CommandBoxWindow.__init__(self)
self.set_title('Client Manager')
self.conn = conn
self.cfg = PaellaConfig()
client_cmds = ['import', 'export', 'remove']
self.add_menu(client_cmds, 'client', self.menu_command)
self.client_view = ScrollCList()
self.vbox.add(self.client_view)
self.client_path = self.cfg.get('management_gui', 'client_path')
self.client_cfg = Configuration(
files=[os.path.join(self.client_path, 'config')])
self.clients = self.client_cfg.sections()
self.client_view.set_rows(self.clients, ['client'])
self.dialogs = {}.fromkeys(client_cmds)
def menu_command(self, menuitem, name):
client = self.client_view.get_selected_data()[0][0]
if name == 'export':
self.export_client(client)
elif name == 'import':
self.import_client(client)
elif name == 'remove':
self.remove_client(client)
else:
dialogs.Message('%s %s' % (name, client))
def _cpaths_(self, client):
cpath = os.path.join(self.client_path, client)
ppath = os.path.join(cpath, 'profiles')
fpath = os.path.join(cpath, 'families')
tpath = os.path.join(cpath, 'traits')
return [cpath, ppath, fpath, tpath]
def _client_schema(self, client):
profiles = self.client_cfg.get_list('profiles', client)
families = self.client_cfg.get_list('families', client)
traits = self.client_cfg.get_list('traits', client)
return [profiles, families, traits]
def _client_mdata(self, client):
disks = self.client_cfg.get_list('disks', client)
mtypes = self.client_cfg.get_list('machine_types', client)
machines = self.client_cfg.get_list('machines', client)
return [disks, mtypes, machines]
def export_client(self, client):
cpath, ppath, fpath, tpath = self._cpaths_(client)
makepaths(cpath)
profiles, families, traits = self._client_schema(client)
disks, mtypes, machines = self._client_mdata(client)
if not disks:
disks = None
if not mtypes:
mtypes = None
if not machines:
machines = None
element = ClientMachineDatabaseElement(self.conn, disks, mtypes,
machines)
mdbpath = join(cpath, 'machine_database.xml')
mdfile = file(mdbpath, 'w')
mdfile.write(element.toprettyxml())
mdfile.close()
if profiles:
makepaths(ppath)
pp = PaellaProfiles(self.conn)
for profile in profiles:
pp.write_profile(profile, ppath)
if families:
makepaths(fpath)
f = Family(self.conn)
for family in families:
f.write_family(family, fpath)
if traits:
makepaths(tpath)
def import_client(self, client):
cpath, ppath, fpath, tpath = self._cpaths_(client)
profiles, families, traits = self._client_schema(client)
mdbpath = join(cpath, 'machine_database.xml')
if families:
f = Family(self.conn)
f.import_families(fpath)
if profiles:
pp = PaellaProcessor(self.conn)
pp.main_path = cpath
pp.insert_profiles()
mh = MachineHandler(self.conn)
md = mh.parse_xmlfile(mdbpath)
mh.insert_parsed_element(md)
def remove_client(self, client):
profiles, families, traits = self._client_schema(client)
disks, mtypes, machines = self._client_mdata(client)
cursor = StatementCursor(self.conn)
if machines:
cursor.delete(table='machines', clause=In('machine', machines))
for mtype in mtypes:
cursor.execute("select * from delete_mtype('%s')" % mtype)
for disk in disks:
cursor.execute("select * from delete_disk('%s')" % disk)
for profile in profiles:
cursor.execute("select * from delete_profile('%s')" % profile)
for family in families:
cursor.execute("select * from delete_family('%s')" % family)
class ClientManager(CommandBoxWindow):
def __init__(self, conn, name='ClientManager'):
CommandBoxWindow.__init__(self)
self.set_title('Client Manager')
self.conn = conn
self.cfg = PaellaConfig()
client_cmds = ['import', 'export', 'remove']
self.add_menu(client_cmds, 'client', self.menu_command)
self.client_view = ScrollCList()
self.vbox.add(self.client_view)
self.client_path = self.cfg.get('management_gui', 'client_path')
self.client_cfg = Configuration(files=[os.path.join(self.client_path, 'config')])
self.clients = self.client_cfg.sections()
self.client_view.set_rows(self.clients, ['client'])
self.dialogs = {}.fromkeys(client_cmds)
def menu_command(self, menuitem, name):
client = self.client_view.get_selected_data()[0][0]
if name == 'export':
self.export_client(client)
elif name == 'import':
self.import_client(client)
elif name == 'remove':
self.remove_client(client)
else:
dialogs.Message('%s %s' % (name, client))
def _cpaths_(self, client):
cpath = os.path.join(self.client_path, client)
ppath = os.path.join(cpath, 'profiles')
fpath = os.path.join(cpath, 'families')
tpath = os.path.join(cpath, 'traits')
return [cpath, ppath, fpath, tpath]
def _client_schema(self, client):
profiles = self.client_cfg.get_list('profiles', client)
families = self.client_cfg.get_list('families', client)
traits = self.client_cfg.get_list('traits', client)
return [profiles, families, traits]
def _client_mdata(self, client):
disks = self.client_cfg.get_list('disks', client)
mtypes = self.client_cfg.get_list('machine_types', client)
machines = self.client_cfg.get_list('machines', client)
return [disks, mtypes, machines]
def export_client(self, client):
cpath, ppath, fpath, tpath = self._cpaths_(client)
makepaths(cpath)
profiles, families, traits = self._client_schema(client)
disks, mtypes, machines = self._client_mdata(client)
if not disks:
disks = None
if not mtypes:
mtypes = None
if not machines:
machines = None
element = ClientMachineDatabaseElement(self.conn, disks, mtypes, machines)
mdbpath = join(cpath, 'machine_database.xml')
mdfile = file(mdbpath, 'w')
mdfile.write(element.toprettyxml())
mdfile.close()
if profiles:
makepaths(ppath)
pp = PaellaProfiles(self.conn)
for profile in profiles:
pp.write_profile(profile, ppath)
if families:
makepaths(fpath)
f = Family(self.conn)
for family in families:
f.write_family(family, fpath)
if traits:
makepaths(tpath)
def import_client(self, client):
cpath, ppath, fpath, tpath = self._cpaths_(client)
profiles, families, traits = self._client_schema(client)
mdbpath = join(cpath, 'machine_database.xml')
if families:
f = Family(self.conn)
f.import_families(fpath)
if profiles:
pp = PaellaProcessor(self.conn)
pp.main_path = cpath
pp.insert_profiles()
mh = MachineHandler(self.conn)
md = mh.parse_xmlfile(mdbpath)
mh.insert_parsed_element(md)
def remove_client(self, client):
profiles, families, traits = self._client_schema(client)
disks, mtypes, machines = self._client_mdata(client)
cursor = StatementCursor(self.conn)
if machines:
cursor.delete(table='machines', clause=In('machine', machines))
for mtype in mtypes:
cursor.execute("select * from delete_mtype('%s')" % mtype)
for disk in disks:
cursor.execute("select * from delete_disk('%s')" % disk)
for profile in profiles:
cursor.execute("select * from delete_profile('%s')" % profile)
for family in families:
cursor.execute("select * from delete_family('%s')" % family)
