Esempio n. 1
0
 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 = {}
Esempio n. 2
0
 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 = {}
Esempio n. 3
0
class Trait(object):
    def __init__(self, conn, suite='woody'):
        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._debconf = TraitDebconf(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._debconf.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 not None:
            trait = [traits]
        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._debconf.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()
        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._debconf.set_trait(trait.name)
        self._scripts.set_trait(trait.name)
        self._parents.insert_parents(trait.parents)
        for package, action in trait.packages.items():
            self._packages.insert_package(package, action)
        for template, data in trait.templates.items():
            print template, data
            templatefile = tar.get_template(data['package'], template)
            idata = {'template' : template}
            print idata
            idata.update(data)
            print idata
            self._templates.insert_template(idata, templatefile)
        for debconf in trait.debconf.values():
            self._debconf.insert(debconf, trait.name)
        for script in trait.scripts:
            scriptfile = tar.get_script(script)
            self._scripts.insert_script(script, scriptfile)
        environ = TraitEnvironment(self.conn, suite, trait.name)
        environ.update(trait.environ)
        self.set_trait(lasttrait)

    def get_config(self):
        return self._debconf.get_config()

    def backup_trait(self, tball_path):
        xmldata = TraitElement(self.conn, self.suite)
        xmldata.set(self.current_trait)
        bkup_path = tempfile.mkdtemp(self.suite, self.current_trait)
        makepaths(bkup_path)
        xmlfile = file(join(bkup_path, 'trait.xml'), 'w')
        xmlfile.write(xmldata.toprettyxml())
        xmlfile.close()
        self._templates.backup_templates(bkup_path)
        self._scripts.backup_scripts(bkup_path)
        make_tarball(bkup_path, tball_path, self.current_trait)
        os.system('rm %s -fr' % bkup_path)
Esempio n. 4
0
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))
Esempio n. 5
0
class Trait(object):
    def __init__(self, conn, suite='woody'):
        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._debconf = TraitDebconf(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._debconf.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 not None:
            trait = [traits]
        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._debconf.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()
        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._debconf.set_trait(trait.name)
        self._scripts.set_trait(trait.name)
        self._parents.insert_parents(trait.parents)
        for package, action in trait.packages.items():
            self._packages.insert_package(package, action)
        for template, data in trait.templates.items():
            #print template, data
            templatefile = tar.get_template(data['package'], template)
            idata = {'template' : template}
            #print idata
            idata.update(data)
            #print idata
            self._templates.insert_template(idata, templatefile)
        for debconf in trait.debconf.values():
            self._debconf.insert(debconf, trait.name)
        for script in trait.scripts:
            scriptfile = tar.get_script(script)
            self._scripts.insert_script(script, scriptfile)
        environ = TraitEnvironment(self.conn, suite, trait.name)
        environ.update(trait.environ)
        self.set_trait(lasttrait)

    def get_config(self):
        return self._debconf.get_config()

    def backup_trait(self, tball_path):
        print 'this needs to be called export_trait'
        xmldata = TraitElement(self.conn, self.suite)
        xmldata.set(self.current_trait)
        bkup_path = tempfile.mkdtemp(self.suite, 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)
        make_tarball(bkup_path, tball_path, self.current_trait)
        os.system('rm %s -fr' % bkup_path)
Esempio n. 6
0
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.items():
            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):
        print 'this needs to be called export_trait'
        xmldata = TraitElement(self.conn, self.suite)
        xmldata.set(self.current_trait)
        bkup_path = join(tball_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)
Esempio n. 7
0
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)