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)
def __init__(self, conn, cfg=None):
object.__init__(self)
self.conn = conn
self.cfg = cfg
self.__set_cursors__()
self.main_path = None
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(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)
def __init__(self, conn, cfg=None):
object.__init__(self)
self.conn = conn
self.cfg = cfg
self.__set_cursors__()
self.main_path = None
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(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)
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_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)
def import_all_diskconfigs(self, dirname=None):
if dirname is None:
dirname = self.main_path / 'diskconfig'
dirname = path(dirname)
files = dirname.listdir()
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)
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)
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))
missing = self._find_missing_packages(suite, traitlist, dirname)
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)
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 _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)
###################
# 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
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
class PaellaProcessor(object):
def __init__(self, conn, cfg=None):
object.__init__(self)
self.conn = conn
self.cfg = cfg
self.__set_cursors__()
self.main_path = None
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
def parse_xml(self, filename):
self.dbdata = PaellaParser(filename)
self.main_path = os.path.dirname(filename)
def start_schema(self):
try:
start_schema(self.conn)
except AlreadyPresentError:
print "primary tables already present"
pass
def insert_apt_sources(self):
self._insert_aptsources()
def _sync_suites(self):
self.main.set_table("suites")
current_suites = [row.suite for row in self.main.select()]
for suite in self.dbdata.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)
else:
raise Error, "%s already exists." % suite
# self.main.update(data=suite)
def _insert_aptsources(self):
for apt in self.dbdata.aptsources:
self.aptsrc.insert_apt_source_row(apt.apt_id, apt.uri, apt.dist, apt.sections, apt.local_path)
self.aptsrc.insert_packages(apt.apt_id)
def insert_families(self):
path = os.path.join(self.main_path, "families")
print "path is in insert_families", path
self.family.import_families(path)
def insert_profiles(self):
path = os.path.join(self.main_path, "profiles")
print "path is in insert_profiles", path
xmlfiles = [os.path.join(path, x) for x in os.listdir(path) if x[-4:] == ".xml"]
profiles = PaellaProfiles(self.conn)
for xmlfile in xmlfiles:
xml = parse_file(xmlfile)
elements = xml.getElementsByTagName("profile")
if len(elements) != 1:
raise Error, "bad profile number %s" % len(elements)
element = elements[0]
parsed = ProfileParser(element)
profiles.insert_profile(parsed)
def insert_profile(self, profile):
idata = {"profile": profile.name, "suite": profile.suite}
self.main.insert(table="profiles", data=idata)
idata = {"profile": profile.name, "trait": None, "ord": 0}
for trait, ord in profile.traits:
print trait, ord
idata["trait"] = trait
idata["ord"] = ord # str(ord)
self.main.insert(table="profile_trait", data=idata)
idata = {"profile": profile.name, "trait": None, "name": None, "value": None}
for trait, name, value in profile.vars:
idata["trait"] = trait
idata["name"] = name
idata["value"] = value
self.main.insert(table="profile_variables", data=idata)
def insert_traits(self, suite):
self.__set_suite_cursors__(suite)
traits = [trait for trait in self.dbdata.get_traits(suite)]
missing = self._find_missing_packages(suite, traits)
if missing:
print "missing these packages in suite", suite
self._make_missing_packages_report(missing)
raise RuntimeError, "Missing packages declared in suite %s" % suite
else:
self._insert_traits_(traits, suite)
def _make_missing_packages_report(self, missing):
traits = missing.keys()
traits.sort()
for trait in traits:
print "%s:" % trait
for package in missing[t]:
print "\t%s" % package
def _find_missing_packages(self, suite, traits):
missing = dict()
tdb = Trait(self.conn, suite)
print suite, traits
for trait in traits:
path = self._get_trait_path(suite, trait)
traitxml = tdb.parse_trait_xml(path, suite=suite)
missing_list = tdb.find_missing_packages(traitxml)
if missing_list:
# print 'in trait,', trait, ', missing packages: ', missing_list
missing[trait] = missing_list
return missing
def _clear_traits(self, suite):
self.__set_suite_cursors__(suite)
self.traitparent.cmd.delete()
self.traitpackage.cmd.delete()
self.main.delete(table=ujoin(suite, "variables"))
self.traits.delete()
def _get_trait_path(self, suite, trait):
return os.path.join(self.main_path, suite, trait)
def _insert_traits_(self, traits, suite):
while len(traits):
trait = traits[0]
print "inserting %s" % trait, len(traits)
try:
self._insert_trait_(trait, suite)
except UnbornError:
traits.append(trait)
del traits[0]
def _insert_trait_(self, trait, suite):
traitdb = Trait(self.conn, suite)
# path = join(self.main_path, suite, trait + '.tar')
path = self._get_trait_path(suite, trait)
traitdb.insert_trait(path, suite)
def __set_cursors__(self):
self.main = StatementCursor(self.conn, "main_paella_cursor")
self.all_traits = StatementCursor(self.conn, "all_traits")
self.all_traits.set_table("traits")
def __set_suite_cursors__(self, suite):
self.traits = StatementCursor(self.conn, "traits")
self.traits.set_table(ujoin(suite, "traits"))
self.traitparent = TraitParent(self.conn, suite)
self.traitpackage = TraitPackage(self.conn, suite)
self.traittemplate = TraitTemplate(self.conn, suite)
self.family = Family(self.conn)
def create(self, filename):
self.parse_xml(filename)
self.start_schema()
self.insert_apt_sources()
self._sync_suites()
for suite in self.dbdata.suites:
self.insert_traits(suite.name)
self.insert_families()
self.insert_profiles()
# use the machine handler to insert
# machine types and such
if os.path.isfile(os.path.join(self.main_path, "machine_database.xml")):
mh = MachineHandler(self.conn)
mh.restore_machine_database(self.main_path)
class PaellaProcessor(object):
def __init__(self, conn, cfg=None):
object.__init__(self)
self.conn = conn
self.cfg = cfg
self.__set_cursors__()
self.main_path = None
self.suitecursor = SuiteCursor(self.conn)
self.aptsrc = AptSourceHandler(self.conn)
def parse_xml(self, filename):
self.dbdata = PaellaParser(filename)
self.main_path = os.path.dirname(filename)
def start_schema(self):
try:
start_schema(self.conn)
except AlreadyPresentError:
print "primary tables already present"
pass
def insert_apt_sources(self):
self._insert_aptsources()
def _sync_suites(self):
self.main.set_table('suites')
current_suites = [row.suite for row in self.main.select()]
for suite in self.dbdata.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)
else:
raise Error, '%s already exists.' % suite
#self.main.update(data=suite)
def _insert_aptsources(self):
for apt in self.dbdata.aptsources:
self.aptsrc.insert_apt_source_row(apt.apt_id, apt.uri, apt.dist,
apt.sections, apt.local_path)
self.aptsrc.insert_packages(apt.apt_id)
def insert_families(self):
path = os.path.join(self.main_path, 'families')
print 'path is in insert_families', path
self.family.import_families(path)
def insert_profiles(self):
path = os.path.join(self.main_path, 'profiles')
print 'path is in insert_profiles', path
xmlfiles = [os.path.join(path, x) for x in os.listdir(path) if x[-4:] == '.xml']
profiles = PaellaProfiles(self.conn)
for xmlfile in xmlfiles:
xml = parse_file(xmlfile)
elements = xml.getElementsByTagName('profile')
if len(elements) != 1:
raise Error, 'bad profile number %s' % len(elements)
element = elements[0]
parsed = ProfileParser(element)
profiles.insert_profile(parsed)
def insert_profile(self, profile):
idata = {'profile' : profile.name,
'suite' : profile.suite}
self.main.insert(table='profiles', data=idata)
idata = {'profile' : profile.name,
'trait' : None,
'ord' : 0}
for trait, ord in profile.traits:
print trait, ord
idata['trait'] = trait
idata['ord'] = ord #str(ord)
self.main.insert(table='profile_trait', data=idata)
idata = {'profile' : profile.name,
'trait' : None,
'name' : None,
'value': None}
for trait, name, value in profile.vars:
idata['trait'] = trait
idata['name'] = name
idata['value'] = value
self.main.insert(table='profile_variables', data=idata)
def insert_traits(self, suite):
self.__set_suite_cursors__(suite)
traits = [trait for trait in self.dbdata.get_traits(suite)]
missing = self._find_missing_packages(suite, traits)
if missing:
print 'missing these packages in suite', suite
self._make_missing_packages_report(missing)
raise RuntimeError, 'Missing packages declared in suite %s' % suite
else:
self._insert_traits_(traits, suite)
def _make_missing_packages_report(self, missing):
traits = missing.keys()
traits.sort()
for trait in traits:
print '%s:' % trait
for package in missing[t]:
print '\t%s' % package
def _find_missing_packages(self, suite, traits):
missing = dict()
tdb = Trait(self.conn, suite)
print suite, traits
for trait in traits:
path = self._get_trait_path(suite, trait)
traitxml = tdb.parse_trait_xml(path, suite=suite)
missing_list = tdb.find_missing_packages(traitxml)
if missing_list:
#print 'in trait,', trait, ', missing packages: ', missing_list
missing[trait] = missing_list
return missing
def _clear_traits(self, suite):
self.__set_suite_cursors__(suite)
self.traitparent.cmd.delete()
self.traitpackage.cmd.delete()
self.main.delete(table=ujoin(suite, 'variables'))
self.traits.delete()
def _get_trait_path(self, suite, trait):
return os.path.join(self.main_path, suite, trait)
def _insert_traits_(self, traits, suite):
while len(traits):
trait = traits[0]
print 'inserting %s' % trait, len(traits)
try:
self._insert_trait_(trait, suite)
except UnbornError:
traits.append(trait)
del traits[0]
def _insert_trait_(self, trait, suite):
traitdb = Trait(self.conn, suite)
#path = join(self.main_path, suite, trait + '.tar')
path = self._get_trait_path(suite, trait)
traitdb.insert_trait(path, suite)
def __set_cursors__(self):
self.main = StatementCursor(self.conn, 'main_paella_cursor')
self.all_traits = StatementCursor(self.conn, 'all_traits')
self.all_traits.set_table('traits')
def __set_suite_cursors__(self, suite):
self.traits = StatementCursor(self.conn, 'traits')
self.traits.set_table(ujoin(suite, 'traits'))
self.traitparent = TraitParent(self.conn, suite)
self.traitpackage = TraitPackage(self.conn, suite)
self.traittemplate = TraitTemplate(self.conn, suite)
self.family = Family(self.conn)
def create(self, filename):
self.parse_xml(filename)
self.start_schema()
self.insert_apt_sources()
self._sync_suites()
for suite in self.dbdata.suites:
self.insert_traits(suite.name)
self.insert_families()
self.insert_profiles()
# use the machine handler to insert
# machine types and such
if os.path.isfile(os.path.join(self.main_path, 'machine_database.xml')):
mh = MachineHandler(self.conn)
mh.restore_machine_database(self.main_path)
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)
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_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)
# 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)
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)
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))
missing = self._find_missing_packages(suite, traitlist, dirname)
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)
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 _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()
machinedb = self.main_path / 'machine_database.xml'
if machinedb.isfile():
mh = MachineHandler(self.conn)
mh.restore_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)
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
