def __init__(self, conn):
self.conn = conn
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
self.main_path = None
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
def __init__(self, conn):
self.conn = conn
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
self.main_path = None
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
class PaellaImporter(object):
def __init__(self, conn):
self.conn = conn
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
self.main_path = None
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
def set_main_path(self, dirname):
self.main_path = path(dirname)
def parse_main_xml(self, filename=None):
if filename is None:
filename = self.main_path / 'database.xml'
parsed = PaellaParser(filename)
return parsed
def start_schema(self):
try:
start_schema(self.conn)
except AlreadyPresentError:
print "primary tables already present"
def import_all_families(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'families'
xmlfiles = dirname.listdir('*.xml')
self.report_total_families(len(xmlfiles))
while xmlfiles:
familyxml = xmlfiles.pop(0)
try:
self.import_family(familyxml)
except UnbornError:
xmlfiles.append(familyxml)
#print 'import families from', dirname
def import_family(self, filename):
self.family.import_family_xml(filename)
self.report_family_imported(filename.namebase)
def import_profile(self, filename):
self.profile.import_profile(filename)
def import_all_profiles(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'profiles'
dirname = path(dirname)
xmlfiles = dirname.listdir('*.xml')
self.report_total_profiles(len(xmlfiles))
for xmlfile in xmlfiles:
self.profile.import_profile(xmlfile)
self.report_profile_imported(xmlfile.namebase)
# warning, this is quick and sloppy
def import_aptkey(self, filename):
filename = path(filename).abspath()
basename = filename.basename()
name = basename.split('.')[0]
data = file(filename).read()
self.aptkeys.insert_key(name, data)
def import_all_aptkeys(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'aptkeys'
dirname = path(dirname)
files = [
f for f in dirname.listdir() if f.isfile() and f.endswith('.gpg')
]
for filename in files:
self.import_aptkey(filename)
# warning, this is quick and sloppy
def import_diskconfig(self, filename):
filename = path(filename).abspath()
basename = filename.basename()
name = basename.split('.')[0]
cursor = self.conn.cursor(statement=True)
content = file(filename).read()
data = dict(name=name, content=content)
cursor.insert(table='diskconfig', data=data)
def import_all_diskconfigs(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'diskconfig'
dirname = path(dirname)
files = [afile for afile in dirname.listdir() if afile.isfile()]
cursor = self.conn.cursor(statement=True)
for diskconfig in files:
name = diskconfig.basename()
data = dict(name=name, content=file(diskconfig).read())
cursor.insert(table='diskconfig', data=data)
# here suite is a parsed xml object (find name)
def make_suite(self, suite):
current_suites = self.suitecursor.get_suites()
if suite.name not in current_suites:
apt_ids = [e.apt_id for e in suite.aptsources]
self.suitecursor.make_suite(suite.name, apt_ids)
os = suite['os']
if not os:
os = None
self.suitecursor.set_os(os, suite=suite.name)
else:
raise RuntimeError, 'suite %s already exists' % suite
# aptsources is the PaellaParser.aptsources attribute
def import_apt_sources(self, aptsources):
self.report_total_apt_sources(len(aptsources))
for apt in aptsources:
self.import_parsed_apt_source(apt)
# here apt is an AptSourceParser object
def import_parsed_apt_source(self, apt):
self.report_importing_aptsrc(apt.apt_id)
self.aptsrc.insert_apt_source_row(apt.apt_id, apt.uri, apt.dist,
apt.sections, apt.local_path)
if not os.environ.has_key('PAELLA_DB_NOPACKAGETABLES'):
self.aptsrc.insert_packages(apt.apt_id)
self.report_aptsrc_imported(apt.apt_id)
def _import_traits(self, suite, traitlist, dirname):
self.report_total_traits(len(traitlist))
if not os.environ.has_key('PAELLA_DB_NOPACKAGETABLES'):
missing = self._find_missing_packages(suite, traitlist, dirname)
else:
missing = False
if missing:
self.report_missing_packages(suite, missing)
raise MissingPackagesError, report_missing_packages(suite, missing)
else:
while len(traitlist):
trait = traitlist.pop(0)
try:
self._import_trait(suite, dirname / trait)
self.report_trait_imported(trait, len(traitlist))
except UnbornError:
traitlist.append(trait)
# the dirname here must have the trait.xml and
# scripts and templates for the trait
def _import_trait(self, suite, dirname):
traitdb = Trait(self.conn, suite)
traitdb.insert_trait(dirname, suite)
def import_trait(self, suite, dirname):
self._import_trait(suite, dirname)
def _find_missing_packages(self, suite, traits, dirname):
missing = dict()
traitdb = Trait(self.conn, suite)
for trait in traits:
tdir = dirname / trait
traitxml = traitdb.parse_trait_xml(tdir, suite=suite)
missing_list = traitdb.find_missing_packages(traitxml)
if missing_list:
missing[trait] = missing_list
return missing
def perform_full_import(self, dirname):
self.set_main_path(dirname)
dbdata = self.parse_main_xml()
self.start_schema()
self.import_apt_sources(dbdata.aptsources)
self.report_total_suites(len(dbdata.suites))
for suite in dbdata.suites:
self.make_suite(suite)
suitedir = self.main_path / suite.name
self._import_traits(suite.name, dbdata.get_traits(suite.name),
suitedir)
self.report_suite_imported(suite.name)
self.import_all_families()
self.import_all_profiles()
self.import_all_diskconfigs()
machinedb = self.main_path / 'machine_database.xml'
if machinedb.isfile():
mh = MachineHandler(self.conn)
mh.import_machine_database(self.main_path)
default_environment_basename = 'default-environment'
filename = self.main_path / default_environment_basename
if filename.isfile():
# similar code exists in kde/environ.py
defenv = DefaultEnvironment(self.conn)
newcfg = RawConfigParser()
newcfg.read(filename)
defenv.update(newcfg)
self.import_all_aptkeys()
###################
# reporting methods
####################
def report_missing_packages(self, suite, missing):
print report_missing_packages(suite, missing)
def report_total_apt_sources(self, total):
print 'importing %d apt sources' % total
def report_importing_aptsrc(self, apt_id):
print 'importing %s' % apt_id
def report_aptsrc_imported(self, apt_id):
print 'apt source %s imported' % apt_id
def report_total_suites(self, total):
print 'importing %d suites' % total
def report_importing_suite(self, suite):
print 'importing suite %s' % suite
def report_suite_imported(self, suite):
print 'suite %s imported' % suite
def report_total_traits(self, total):
print 'importing %d traits' % total
def report_importing_trait(self, trait, numtraits):
print 'importing trait', trait
def report_trait_imported(self, trait, numtraits):
print 'trait %s imported' % trait
def report_total_families(self, total):
print 'importing %d families' % total
def report_family_imported(self, family):
print 'family %s imported' % family
def report_total_profiles(self, total):
print 'importing %d profiles' % total
def report_profile_imported(self, profile):
print 'profile %s imported' % profile
def setup_cursors(self):
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.machines = MachineHandler(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
class PaellaExporter(object):
def __init__(self, conn):
self.conn = conn
self.cursor = self.conn.cursor(statement=True)
self.suitecursor = SuiteCursor(self.conn)
self.init_db_element()
if len(self.cursor.tables()):
self.setup_cursors()
def setup_cursors(self):
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.machines = MachineHandler(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
def init_db_element(self):
self.dbelement = PaellaDatabaseElement()
def make_complete_db_element(self):
self.init_db_element()
self._append_apt_sources_to_db_element()
fields = ['suite']
for suite in self.suitecursor.get_suites():
self._append_suite_to_db_element(suite)
def _append_apt_sources_to_db_element(self):
rows = self.cursor.select(table='apt_sources', order=['apt_id'])
for row in rows:
self.dbelement.append_apt_source(row.apt_id, row.uri, row.dist,
row.sections, row.local_path)
def _append_suite_to_db_element(self, suite):
self.dbelement.append_suite(suite)
element = self.dbelement.suites[suite]
os = self.suitecursor.get_os(suite=suite)
if os is None:
os = ''
element.set_os(os)
rows = self.cursor.select(table='suite_apt_sources',
order=['ord'],
clause=Eq('suite', suite))
for row in rows:
self.dbelement.append_suite_apt_source(row.suite, row.apt_id,
str(row.ord))
rows = self.cursor.select(fields=['trait'],
table='%s_traits' % suite,
order=['trait'])
traits = [row.trait for row in rows]
self.dbelement.append_suite_traits(suite, traits=traits)
def set_db_export_path(self, dirname):
self.db_export_path = path(dirname)
def export_db_element(self, dirname=None, filename='database.xml'):
if dirname is None:
dirname = self.db_export_path
filename = path(dirname) / filename
dbfile = filename.open('w')
self.dbelement.writexml(dbfile, indent='\t', newl='\n', addindent='\t')
dbfile.close()
def _make_suite_export_path(self, suite):
suitedir = self.db_export_path / suite
makepaths(suitedir)
return suitedir
def export_aptkey(self, name=None, dirname=None, row=None):
dirname = path(dirname)
makepaths(dirname)
if name is None and row is None:
raise RuntimeError, "need to set either row or name"
if row is None:
row = self.aptkeys.get_row(name)
basename = '%s.gpg' % row.name
filename = dirname / basename
filename.write_text(row.data)
def export_all_aptkeys(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'aptkeys'
rows = self.aptkeys.get_rows()
for row in rows:
self.export_aptkey(dirname=dirname, row=row)
def export_profile(self, profile, dirname=None):
if dirname is None:
dirname = path('.')
dirname = path(dirname)
makepaths(dirname)
# this env object should be handled
# inside the profile object instead
# of being handled here.
env = self.profile.make_environment_object()
env.set_profile(profile)
self.profile.export_profile(dirname, profile=profile, env=env)
def export_all_profiles(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'profiles'
makepaths(dirname)
profiles = self.profile.get_profile_list()
self.report_total_profiles(len(profiles))
env = self.profile.make_environment_object()
for profile in profiles:
env.set_profile(profile)
self.profile.export_profile(dirname, profile=profile, env=env)
self.report_profile_exported(profile)
def export_diskconfig(self, name=None, dirname=None, row=None):
if dirname is None:
dirname = self.db_export_path / 'diskconfig'
dirname = path(dirname)
makepaths(dirname)
if name is None and row is None:
raise RuntimeError, 'either name or row must be passed to export_diskconfig'
if name is not None:
row = self.diskconfig.get(name)
filename = row.name
content = row.content
diskconfig = dirname / filename
diskconfig.write_text(content)
def export_all_diskconfigs(self, dirname=None):
for row in self.diskconfig.cursor.select():
self.export_diskconfig(dirname=dirname, row=row)
def export_all_families(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'families'
dirname = path(dirname)
makepaths(dirname)
self.family.export_families(dirname)
def export_family(self, family, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'families'
dirname = path(dirname)
makepaths(dirname)
self.family.write_family(family, dirname)
def export_trait(self, trait, suite=None, dirname=None, traitdb=None):
if traitdb is None:
if suite is None:
RuntimeError, "you must pass a suite if you don't pass a Trait object"
print "make new traitdb"
traitdb = Trait(self.conn, suite)
traitdb.set_trait(trait)
if suite is None:
suite = traitdb.suite
if dirname is None:
dirname = self._make_suite_export_path(suite)
traitdb.export_trait(dirname)
self.report_trait_exported(trait, dirname)
def export_all_traits(self, suite, dirname=None):
self.report_start_exporting_traits()
traitdb = Trait(self.conn, suite)
traits = traitdb.get_trait_list()
self.report_total_traits(len(traits))
for trait in traits:
self.export_trait(trait, dirname=dirname, traitdb=traitdb)
#self.report_all_traits_exported()
def export_suite(self, suite, dirname=None):
self.export_all_traits(suite, dirname=dirname)
def export_all_suites(self, dirname=None):
suites = self.suitecursor.get_suites()
self.report_total_suites(len(suites))
for suite in suites:
self.report_exporting_suite(suite)
self.export_suite(suite, dirname=dirname)
self.report_suite_exported(suite)
def export_machine_database(self, dirname=None):
if dirname is None:
dirname = self.db_export_path
else:
dirname = path(dirname)
self.machines.export_machine_database(dirname)
def export_machine(self, machine, dirname=None):
if dirname is None:
dirname = self.db_export_path
dirname = path(dirname)
makepaths(dirname)
current_machine = self.machines.current_machine
self.machines.set_machine(machine)
self.machines.export_machine(dirname)
if current_machine is not None:
self.machines.set_machine(current_machine)
def _export_environment_common(self, dirname, envtype):
if dirname is None:
dirname = self.db_export_path
else:
dirname = path(dirname)
if envtype == 'default':
envclass = DefaultEnvironment
elif envtype == 'current':
envclass = CurrentEnvironment
else:
raise RuntimeError, 'bad envtype %s' % envtype
env = envclass(self.conn)
#filename = os.path.join(path, '%s-environment' % envtype)
#efile = file(filename, 'w')
#env.write(efile)
#efile.close()
filename = '%s-environment' % envtype
fullname = dirname / filename
envfile = fullname.open('w')
env.write(envfile)
envfile.close()
def export_default_environment(self, path=None):
self._export_environment_common(path, 'default')
def export_current_environment(self, path=None):
self._export_environment_common(path, 'common')
def perform_full_export(self, path):
self.make_complete_db_element()
self.set_db_export_path(path)
self.export_db_element()
self.export_all_suites()
self.export_all_profiles()
self.export_all_families()
self.export_all_diskconfigs()
self.export_machine_database()
self.export_default_environment()
self.export_all_aptkeys()
###################
# reporting methods
####################
def report_total_suites(self, total):
print 'exporting %d suites' % total
def report_exporting_suite(self, suite):
print 'exporting suite %s' % suite
def report_suite_exported(self, suite):
print 'suite %s exported' % suite
def report_total_traits(self, total):
print 'exporting %d traits' % total
def report_trait_exported(self, trait, path):
print 'trait %s exported to %s' % (trait, path)
def report_all_traits_exported(self, *args):
print 'all traits exported'
def report_start_exporting_traits(self):
print 'starting to export traits'
def report_total_profiles(self, total):
print 'exporting %d profiles' % total
def report_profile_exported(self, profile):
print 'profile %s exported' % profile
class PaellaImporter(object):
def __init__(self, conn):
self.conn = conn
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
self.main_path = None
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
def set_main_path(self, dirname):
self.main_path = path(dirname)
def parse_main_xml(self, filename=None):
if filename is None:
filename = self.main_path / 'database.xml'
parsed = PaellaParser(filename)
return parsed
def start_schema(self):
try:
start_schema(self.conn)
except AlreadyPresentError:
print "primary tables already present"
def import_all_families(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'families'
xmlfiles = dirname.listdir('*.xml')
self.report_total_families(len(xmlfiles))
while xmlfiles:
familyxml = xmlfiles.pop(0)
try:
self.import_family(familyxml)
except UnbornError:
xmlfiles.append(familyxml)
#print 'import families from', dirname
def import_family(self, filename):
self.family.import_family_xml(filename)
self.report_family_imported(filename.namebase)
def import_profile(self, filename):
self.profile.import_profile(filename)
def import_all_profiles(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'profiles'
dirname = path(dirname)
xmlfiles = dirname.listdir('*.xml')
self.report_total_profiles(len(xmlfiles))
for xmlfile in xmlfiles:
self.profile.import_profile(xmlfile)
self.report_profile_imported(xmlfile.namebase)
# warning, this is quick and sloppy
def import_aptkey(self, filename):
filename = path(filename).abspath()
basename = filename.basename()
name = basename.split('.')[0]
data = file(filename).read()
self.aptkeys.insert_key(name, data)
def import_all_aptkeys(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'aptkeys'
dirname = path(dirname)
files = [f for f in dirname.listdir() if f.isfile() and f.endswith('.gpg')]
for filename in files:
self.import_aptkey(filename)
# warning, this is quick and sloppy
def import_diskconfig(self, filename):
filename = path(filename).abspath()
basename = filename.basename()
name = basename.split('.')[0]
cursor = self.conn.cursor(statement=True)
content = file(filename).read()
data = dict(name=name, content=content)
cursor.insert(table='diskconfig', data=data)
def import_all_diskconfigs(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'diskconfig'
dirname = path(dirname)
files = [afile for afile in dirname.listdir() if afile.isfile()]
cursor = self.conn.cursor(statement=True)
for diskconfig in files:
name= diskconfig.basename()
data = dict(name=name, content=file(diskconfig).read())
cursor.insert(table='diskconfig', data=data)
# here suite is a parsed xml object (find name)
def make_suite(self, suite):
current_suites = self.suitecursor.get_suites()
if suite.name not in current_suites:
apt_ids = [e.apt_id for e in suite.aptsources]
self.suitecursor.make_suite(suite.name, apt_ids)
os = suite['os']
if not os:
os = None
self.suitecursor.set_os(os, suite=suite.name)
else:
raise RuntimeError , 'suite %s already exists' % suite
# aptsources is the PaellaParser.aptsources attribute
def import_apt_sources(self, aptsources):
self.report_total_apt_sources(len(aptsources))
for apt in aptsources:
self.import_parsed_apt_source(apt)
# here apt is an AptSourceParser object
def import_parsed_apt_source(self, apt):
self.report_importing_aptsrc(apt.apt_id)
self.aptsrc.insert_apt_source_row(apt.apt_id, apt.uri, apt.dist,
apt.sections, apt.local_path)
if not os.environ.has_key('PAELLA_DB_NOPACKAGETABLES'):
self.aptsrc.insert_packages(apt.apt_id)
self.report_aptsrc_imported(apt.apt_id)
def _import_traits(self, suite, traitlist, dirname):
self.report_total_traits(len(traitlist))
if not os.environ.has_key('PAELLA_DB_NOPACKAGETABLES'):
missing = self._find_missing_packages(suite, traitlist, dirname)
else:
missing = False
if missing:
self.report_missing_packages(suite, missing)
raise MissingPackagesError, report_missing_packages(suite, missing)
else:
while len(traitlist):
trait = traitlist.pop(0)
try:
self._import_trait(suite, dirname / trait)
self.report_trait_imported(trait, len(traitlist))
except UnbornError:
traitlist.append(trait)
# the dirname here must have the trait.xml and
# scripts and templates for the trait
def _import_trait(self, suite, dirname):
traitdb = Trait(self.conn, suite)
traitdb.insert_trait(dirname, suite)
def import_trait(self, suite, dirname):
self._import_trait(suite, dirname)
def _find_missing_packages(self, suite, traits, dirname):
missing = dict()
traitdb = Trait(self.conn, suite)
for trait in traits:
tdir = dirname / trait
traitxml = traitdb.parse_trait_xml(tdir, suite=suite)
missing_list = traitdb.find_missing_packages(traitxml)
if missing_list:
missing[trait] = missing_list
return missing
def perform_full_import(self, dirname):
self.set_main_path(dirname)
dbdata = self.parse_main_xml()
self.start_schema()
self.import_apt_sources(dbdata.aptsources)
self.report_total_suites(len(dbdata.suites))
for suite in dbdata.suites:
self.make_suite(suite)
suitedir = self.main_path / suite.name
self._import_traits(suite.name, dbdata.get_traits(suite.name), suitedir)
self.report_suite_imported(suite.name)
self.import_all_families()
self.import_all_profiles()
self.import_all_diskconfigs()
machinedb = self.main_path / 'machine_database.xml'
if machinedb.isfile():
mh = MachineHandler(self.conn)
mh.import_machine_database(self.main_path)
default_environment_basename = 'default-environment'
filename = self.main_path / default_environment_basename
if filename.isfile():
# similar code exists in kde/environ.py
defenv = DefaultEnvironment(self.conn)
newcfg = RawConfigParser()
newcfg.read(filename)
defenv.update(newcfg)
self.import_all_aptkeys()
###################
# reporting methods
####################
def report_missing_packages(self, suite, missing):
print report_missing_packages(suite, missing)
def report_total_apt_sources(self, total):
print 'importing %d apt sources' % total
def report_importing_aptsrc(self, apt_id):
print 'importing %s' % apt_id
def report_aptsrc_imported(self, apt_id):
print 'apt source %s imported' % apt_id
def report_total_suites(self, total):
print 'importing %d suites' % total
def report_importing_suite(self, suite):
print 'importing suite %s' % suite
def report_suite_imported(self, suite):
print 'suite %s imported' % suite
def report_total_traits(self, total):
print 'importing %d traits' % total
def report_importing_trait(self, trait, numtraits):
print 'importing trait', trait
def report_trait_imported(self, trait, numtraits):
print 'trait %s imported' % trait
def report_total_families(self, total):
print 'importing %d families' % total
def report_family_imported(self, family):
print 'family %s imported' % family
def report_total_profiles(self, total):
print 'importing %d profiles' % total
def report_profile_imported(self, profile):
print 'profile %s imported' % profile
def setup_cursors(self):
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.machines = MachineHandler(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
class PaellaExporter(object):
def __init__(self, conn):
self.conn = conn
self.cursor = self.conn.cursor(statement=True)
self.suitecursor = SuiteCursor(self.conn)
self.init_db_element()
if len(self.cursor.tables()):
self.setup_cursors()
def setup_cursors(self):
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.machines = MachineHandler(self.conn)
self.diskconfig = DiskConfigHandler(self.conn)
self.aptkeys = AptKeyHandler(self.conn)
def init_db_element(self):
self.dbelement = PaellaDatabaseElement()
def make_complete_db_element(self):
self.init_db_element()
self._append_apt_sources_to_db_element()
fields = ['suite']
for suite in self.suitecursor.get_suites():
self._append_suite_to_db_element(suite)
def _append_apt_sources_to_db_element(self):
rows = self.cursor.select(table='apt_sources', order=['apt_id'])
for row in rows:
self.dbelement.append_apt_source(row.apt_id, row.uri, row.dist,
row.sections, row.local_path)
def _append_suite_to_db_element(self, suite):
self.dbelement.append_suite(suite)
element = self.dbelement.suites[suite]
os = self.suitecursor.get_os(suite=suite)
if os is None:
os = ''
element.set_os(os)
rows = self.cursor.select(table='suite_apt_sources', order=['ord'],
clause=Eq('suite', suite))
for row in rows:
self.dbelement.append_suite_apt_source(row.suite, row.apt_id,
str(row.ord))
rows = self.cursor.select(fields=['trait'], table='%s_traits' % suite,
order=['trait'])
traits = [row.trait for row in rows]
self.dbelement.append_suite_traits(suite, traits=traits)
def set_db_export_path(self, dirname):
self.db_export_path = path(dirname)
def export_db_element(self, dirname=None, filename='database.xml'):
if dirname is None:
dirname = self.db_export_path
filename = path(dirname) / filename
dbfile = filename.open('w')
self.dbelement.writexml(dbfile, indent='\t', newl='\n', addindent='\t')
dbfile.close()
def _make_suite_export_path(self, suite):
suitedir = self.db_export_path / suite
makepaths(suitedir)
return suitedir
def export_aptkey(self, name=None, dirname=None, row=None):
dirname = path(dirname)
makepaths(dirname)
if name is None and row is None:
raise RuntimeError, "need to set either row or name"
if row is None:
row = self.aptkeys.get_row(name)
basename = '%s.gpg' % row.name
filename = dirname / basename
filename.write_text(row.data)
def export_all_aptkeys(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'aptkeys'
rows = self.aptkeys.get_rows()
for row in rows:
self.export_aptkey(dirname=dirname, row=row)
def export_profile(self, profile, dirname=None):
if dirname is None:
dirname = path('.')
dirname = path(dirname)
makepaths(dirname)
# this env object should be handled
# inside the profile object instead
# of being handled here.
env = self.profile.make_environment_object()
env.set_profile(profile)
self.profile.export_profile(dirname, profile=profile, env=env)
def export_all_profiles(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'profiles'
makepaths(dirname)
profiles = self.profile.get_profile_list()
self.report_total_profiles(len(profiles))
env = self.profile.make_environment_object()
for profile in profiles:
env.set_profile(profile)
self.profile.export_profile(dirname, profile=profile, env=env)
self.report_profile_exported(profile)
def export_diskconfig(self, name=None, dirname=None, row=None):
if dirname is None:
dirname = self.db_export_path / 'diskconfig'
dirname = path(dirname)
makepaths(dirname)
if name is None and row is None:
raise RuntimeError , 'either name or row must be passed to export_diskconfig'
if name is not None:
row = self.diskconfig.get(name)
filename = row.name
content = row.content
diskconfig = dirname / filename
diskconfig.write_text(content)
def export_all_diskconfigs(self, dirname=None):
for row in self.diskconfig.cursor.select():
self.export_diskconfig(dirname=dirname, row=row)
def export_all_families(self, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'families'
dirname = path(dirname)
makepaths(dirname)
self.family.export_families(dirname)
def export_family(self, family, dirname=None):
if dirname is None:
dirname = self.db_export_path / 'families'
dirname = path(dirname)
makepaths(dirname)
self.family.write_family(family, dirname)
def export_trait(self, trait, suite=None, dirname=None, traitdb=None):
if traitdb is None:
if suite is None:
RuntimeError , "you must pass a suite if you don't pass a Trait object"
print "make new traitdb"
traitdb = Trait(self.conn, suite)
traitdb.set_trait(trait)
if suite is None:
suite = traitdb.suite
if dirname is None:
dirname = self._make_suite_export_path(suite)
traitdb.export_trait(dirname)
self.report_trait_exported(trait, dirname)
def export_all_traits(self, suite, dirname=None):
self.report_start_exporting_traits()
traitdb = Trait(self.conn, suite)
traits = traitdb.get_trait_list()
self.report_total_traits(len(traits))
for trait in traits:
self.export_trait(trait, dirname=dirname, traitdb=traitdb)
#self.report_all_traits_exported()
def export_suite(self, suite, dirname=None):
self.export_all_traits(suite, dirname=dirname)
def export_all_suites(self, dirname=None):
suites = self.suitecursor.get_suites()
self.report_total_suites(len(suites))
for suite in suites:
self.report_exporting_suite(suite)
self.export_suite(suite, dirname=dirname)
self.report_suite_exported(suite)
def export_machine_database(self, dirname=None):
if dirname is None:
dirname = self.db_export_path
else:
dirname = path(dirname)
self.machines.export_machine_database(dirname)
def export_machine(self, machine, dirname=None):
if dirname is None:
dirname = self.db_export_path
dirname = path(dirname)
makepaths(dirname)
current_machine = self.machines.current_machine
self.machines.set_machine(machine)
self.machines.export_machine(dirname)
if current_machine is not None:
self.machines.set_machine(current_machine)
def _export_environment_common(self, dirname, envtype):
if dirname is None:
dirname = self.db_export_path
else:
dirname = path(dirname)
if envtype == 'default':
envclass = DefaultEnvironment
elif envtype == 'current':
envclass = CurrentEnvironment
else:
raise RuntimeError , 'bad envtype %s' % envtype
env = envclass(self.conn)
#filename = os.path.join(path, '%s-environment' % envtype)
#efile = file(filename, 'w')
#env.write(efile)
#efile.close()
filename = '%s-environment' % envtype
fullname = dirname / filename
envfile = fullname.open('w')
env.write(envfile)
envfile.close()
def export_default_environment(self, path=None):
self._export_environment_common(path, 'default')
def export_current_environment(self, path=None):
self._export_environment_common(path, 'common')
def perform_full_export(self, path):
self.make_complete_db_element()
self.set_db_export_path(path)
self.export_db_element()
self.export_all_suites()
self.export_all_profiles()
self.export_all_families()
self.export_all_diskconfigs()
self.export_machine_database()
self.export_default_environment()
self.export_all_aptkeys()
###################
# reporting methods
####################
def report_total_suites(self, total):
print 'exporting %d suites' % total
def report_exporting_suite(self, suite):
print 'exporting suite %s' % suite
def report_suite_exported(self, suite):
print 'suite %s exported' % suite
def report_total_traits(self, total):
print 'exporting %d traits' % total
def report_trait_exported(self, trait, path):
print 'trait %s exported to %s' % (trait, path)
def report_all_traits_exported(self, *args):
print 'all traits exported'
def report_start_exporting_traits(self):
print 'starting to export traits'
def report_total_profiles(self, total):
print 'exporting %d profiles' % total
def report_profile_exported(self, profile):
print 'profile %s exported' % profile