class Trait(object):
def __init__(self, conn, suite='sarge'):
object.__init__(self)
self.conn = conn
self.suite = suite
self._alltraits = AllTraits(self.conn)
self._traits = Traits(self.conn, self.suite)
self._parents = TraitParent(self.conn, self.suite)
self._packages = TraitPackage(self.conn, self.suite)
self._templates = TraitTemplate(self.conn, self.suite)
self._scripts = TraitScript(self.conn, self.suite)
self.current_trait = None
self.environ = {}
self._xmldata = TraitElement(self.conn, self.suite)
def set_trait(self, trait):
self.current_trait = trait
self.environ = TraitEnvironment(self.conn, self.suite, trait)
self._parents.set_trait(trait)
self._packages.set_trait(trait)
self._templates.set_trait(trait)
self._scripts.set_trait(trait)
def get_traits(self, order=None):
return self._traits.select(fields=['trait'], order=order)
def packages(self, trait=None, action=False):
if trait is None:
trait = self.current_trait
rows = self._packages.packages([trait])
if action:
return rows
else:
return [row.package for row in rows]
def get_package_rows(self):
return self._packages.packages([self.current_trait])
def set_action(self, action, packages):
self._packages.set_action(action, packages)
def insert_packages(self, packages):
self._packages.insert_packages(packages)
def add_package(self, package, action):
self._packages.insert_package(package, action)
def delete_package(self, package, action):
self._packages.delete_package(package, action)
def parents(self, trait=None):
return [x.parent for x in self._parents.parents(trait)]
def get_traitset(self):
return self._parents.get_traitset([self.current_trait])
def get_trait_list(self, order=None):
return [row.trait for row in self.get_traits(order=order)]
def insert_parents(self, traits):
self._parents.insert_parents(traits)
def delete_parents(self, parents):
self._parents.delete(parents)
def update_parents(self, parents):
self._parents.insert_new_parents_list(parents)
def templates(self, trait=None, fields=None):
rows = self._templates.templates(trait, fields)
if fields is None:
return [row.template for row in rows]
else:
return rows
def insert_template(self, template_data):
self._templates.insert_template(template_data)
def insert_template_from_tarfile(self, template_path, tarball):
self._templates.insert_template_from_tarfile(template_path, tarball)
def update_template(self, template, data=None, templatefile=None,
contents=None):
self._templates.update_template(template, data=data, templatefile=templatefile,
contents=contents)
def get_template_rows(self):
return self._templates.templates(self.current_trait)
def get_template_row(self, template):
return self._templates.get_row(template)
def edit_template(self, template):
self._templates.edit_template(template)
def get_template_contents(self, template):
return self._templates.templatedata(template)
def delete_template(self, template):
self._templates.drop_template(template)
def get_full_environment(self):
return self._parents.get_environment([self.current_trait])
def edit_script(self, name):
trait = self.current_trait
self._scripts.edit_script(name)
def insert_script(self, name, scriptfile):
self._scripts.insert_script(name, scriptfile)
def create_trait(self, trait):
insert_data = {'trait' : trait}
if trait not in self._alltraits.list():
self._alltraits.insert(data=insert_data)
if trait not in self._traits.list():
self._traits.insert(data=insert_data)
else:
raise ExistsError, '%s already exists' % trait
def delete_trait(self, trait):
environ = TraitEnvironment(self.conn, self.suite, trait)
environ.clear()
self._templates.delete_trait(trait)
self._parents.delete_trait(trait)
self._packages.delete_trait(trait)
self._scripts.delete_trait(trait)
self._traits.set_clause([('trait', trait)])
self._traits.delete()
self._traits.clear(clause=True)
def parse_trait_xml(self, path, suite=None):
trait = TraitXml(file(os.path.join(path, 'trait.xml')))
if suite is not None:
trait.suite = suite
return trait
def find_missing_packages(self, traitxml):
all_packages = []
missing = []
cursor = StatementCursor(self.conn)
suite = traitxml.suite
ptable = '%s_packages' % suite
for package, action in traitxml.packages:
if package not in all_packages:
all_packages.append(package)
for package in all_packages:
try:
row = cursor.select_row(table=ptable, clause=Eq('package', package))
except NoExistError:
missing.append(package)
return missing
def insert_trait(self, path, suite=None):
#tar = TraitTarFile(path)
#trait = tar.get_trait()
trait = self.parse_trait_xml(path, suite=suite)
all = Set([x.trait for x in self._alltraits.select()])
suite_traits = Set([x.trait for x in self._traits.select()])
parents = Set(trait.parents)
debug("parents for trait %s, %s" % (trait.name, parents))
if not parents.issubset(suite_traits):
raise UnbornError, 'Parent Unborn'
if trait.name in suite_traits:
raise Error, 'trait exists'
idata ={'trait' : trait.name}
if trait.name not in all:
self._alltraits.insert(data=idata)
description_path = os.path.join(path, 'description.txt')
if os.path.isfile(description_path):
idata['description'] = file(description_path).read()
if trait.name not in suite_traits:
self._traits.insert(data=idata)
else:
raise RuntimeError, '%s already there' % trait.name
lasttrait = self.current_trait
self._parents.set_trait(trait.name)
self._packages.set_trait(trait.name)
self._templates.set_trait(trait.name)
self._scripts.set_trait(trait.name)
self._parents.insert_parents(trait.parents)
for package, action in trait.packages:
self._packages.insert_package(package, action)
n = 0
for template, data in trait.templates:
#print template, data
#templatefile = tar.get_template(data['package'], template)
template_id = template.replace('/', '-slash-')
template_filename = os.path.join(path, 'template-%s' % template_id)
if not os.path.exists(template_filename):
print "in suite %s trait %s" % (suite, trait.name)
print "exported template %s not converted yet" % template
template_filename = os.path.join(path, 'template-%d' % n)
templatefile = file(template_filename)
idata = {'template' : template}
#print idata
idata.update(data)
#print idata
self._templates.insert_template(idata, templatefile)
n += 1
for script in trait.scripts:
#scriptfile = tar.get_script(script)
scriptfile = file(os.path.join(path, 'script-%s' % script))
self._scripts.insert_script(script, scriptfile)
environ = TraitEnvironment(self.conn, suite, trait.name)
environ.update(trait.environ)
self.set_trait(lasttrait)
def export_trait(self, suite_path):
#print "----Begin export trait", self.current_trait
#print "start xml", self.current_trait
#xmldata = TraitElement(self.conn, self.suite)
xmldata = self._xmldata
#print 'set xml data'
xmldata.set(self.current_trait)
#print 'xml data set'
bkup_path = os.path.join(suite_path, self.current_trait)
makepaths(bkup_path)
#print 'render xml'
xmlfile = file(os.path.join(bkup_path, 'trait.xml'), 'w')
xmlfile.write(xmldata.toprettyxml())
xmlfile.close()
#print 'xml rendered'
#print "end xml", self.current_trait
row = self._traits.select_row(clause=Eq('trait', self.current_trait))
if row['description'] is not None:
print 'export description', self.current_trait
descfile = file(os.path.join(bkup_path, 'description.txt'), 'w')
descfile.write(row['description'])
descfile.close()
#print "start templates,scripts", self.current_trait
self._templates.export_templates(bkup_path)
self._scripts.export_scripts(bkup_path)
#print "end templates,scripts", self.current_trait
#print 'all exported', os.listdir(bkup_path)
print 'trait', self.current_trait, 'exported in suite', self.suite
def get_description(self):
trait = self.current_trait
row = self._traits.select_row(clause=Eq('trait', trait))
# we must use row['dsc'] instead of row.dsc here
return row['description']
def set_description(self, desc):
trait = self.current_trait
data = dict(description=desc)
self._traits.update(data=data, clause=Eq('trait', trait))
class Trait(object):
def __init__(self, conn, suite='sarge'):
object.__init__(self)
self.conn = conn
self.suite = suite
self._alltraits = AllTraits(self.conn)
self._traits = Traits(self.conn, self.suite)
self._parents = TraitParent(self.conn, self.suite)
self._packages = TraitPackage(self.conn, self.suite)
self._templates = TraitTemplate(self.conn, self.suite)
self._scripts = TraitScript(self.conn, self.suite)
self.current_trait = None
self.environ = {}
def set_trait(self, trait):
self.current_trait = trait
self.environ = TraitEnvironment(self.conn, self.suite, trait)
self._parents.set_trait(trait)
self._packages.set_trait(trait)
self._templates.set_trait(trait)
self._scripts.set_trait(trait)
def parents(self, trait=None):
return [x.parent for x in self._parents.parents(trait)]
def templates(self, trait=None, fields=None):
rows = self._templates.templates(trait, fields)
if fields is None:
return [row.template for row in rows]
else:
return rows
def packages(self, trait=None, action=False):
if trait is None:
trait = self.current_trait
rows = self._packages.packages([trait])
if action:
return rows
else:
return [row.package for row in rows]
def get_package_rows(self):
return self._packages.packages([self.current_trait])
def get_traits(self):
return self._traits.select(fields=['trait'])
def get_trait_list(self):
return [row.trait for row in self.get_traits()]
def get_template_rows(self):
return self._templates.templates(self.current_trait)
def set_action(self, action, packages):
self._packages.set_action(action, packages)
def get_traitset(self):
return self._parents.get_traitset([self.current_trait])
def insert_packages(self, packages):
self._packages.insert_packages(packages)
def insert_parents(self, traits):
self._parents.insert_parents(traits)
def insert_template(self, template_data):
self._templates.insert_template(template_data)
def update_template(self, template_data):
self._templates.update_template(template_data)
def delete_parents(self, parents):
self._parents.delete(parents)
def create_trait(self, trait):
insert_data = {'trait': trait}
if trait not in self._alltraits.list():
self._alltraits.insert(data=insert_data)
if trait not in self._traits.list():
self._traits.insert(data=insert_data)
else:
raise ExistsError, '%s already exists' % trait
def delete_trait(self, trait):
environ = TraitEnvironment(self.conn, self.suite, trait)
environ.clear()
self._templates.delete_trait(trait)
self._parents.delete_trait(trait)
self._packages.delete_trait(trait)
self._scripts.delete_trait(trait)
self._traits.set_clause([('trait', trait)])
self._traits.delete()
self._traits.clear(clause=True)
def insert_trait(self, path, suite=None):
#tar = TraitTarFile(path)
#trait = tar.get_trait()
trait = TraitXml(file(join(path, 'trait.xml')))
if suite is not None:
trait.suite = suite
all = Set([x.trait for x in self._alltraits.select()])
suite_traits = Set([x.trait for x in self._traits.select()])
parents = Set(trait.parents)
debug(parents)
if not parents.issubset(suite_traits):
raise UnbornError, 'Parent Unborn'
if trait.name in suite_traits:
raise Error, 'trait exists'
idata = {'trait': trait.name}
if trait.name not in all:
self._alltraits.insert(data=idata)
if trait.name not in suite_traits:
self._traits.insert(data=idata)
lasttrait = self.current_trait
self._parents.set_trait(trait.name)
self._packages.set_trait(trait.name)
self._templates.set_trait(trait.name)
self._scripts.set_trait(trait.name)
self._parents.insert_parents(trait.parents)
for package, action in trait.packages:
self._packages.insert_package(package, action)
n = 0
for package, template, data in trait.templates:
#print template, data
#templatefile = tar.get_template(data['package'], template)
templatefile = file(join(path, 'template-%d' % n))
idata = {'template': template}
#print idata
idata.update(data)
#print idata
self._templates.insert_template(idata, templatefile)
n += 1
for script in trait.scripts:
#scriptfile = tar.get_script(script)
scriptfile = file(join(path, 'script-%s' % script))
self._scripts.insert_script(script, scriptfile)
environ = TraitEnvironment(self.conn, suite, trait.name)
environ.update(trait.environ)
self.set_trait(lasttrait)
def backup_trait(self, tball_path):
raise Error, 'this call is deprecated, use export_trait'
def export_trait(self, suite_path):
xmldata = TraitElement(self.conn, self.suite)
xmldata.set(self.current_trait)
bkup_path = join(suite_path, self.current_trait)
makepaths(bkup_path)
xmlfile = file(join(bkup_path, 'trait.xml'), 'w')
xmlfile.write(xmldata.toprettyxml())
xmlfile.close()
self._templates.export_templates(bkup_path)
self._scripts.export_scripts(bkup_path)
#print 'all exported', os.listdir(bkup_path)
print 'trait', self.current_trait, 'exported in suite', self.suite
def get_description(self):
trait = self.current_trait
row = self._traits.select_row(clause=Eq('trait', trait))
# we must use row['dsc'] instead of row.dsc here
return row['description']
def set_description(self, desc):
trait = self.current_trait
data = dict(description=desc)
self._traits.update(data=data, clause=Eq('trait', trait))
def add_package(self, package, action):
self._packages.insert_package(package, action)
def edit_script(self, name):
trait = self.current_trait
self._scripts.edit_script(name)
def edit_template(self, package, template):
self._templates.edit_template(package, template)