23N/A

23N/Aimport os

23N/Aimport re

23N/Aimport sha

23N/Aimport shutil

23N/Aimport time

23N/Aimport urllib

23N/A

23N/Aimport pkg.fmri as fmri

36N/Aimport pkg.package as package

23N/A

34N/Aclass Catalog(object):

34N/A """A Catalog is the representation of the package FMRIs available to

37N/A

37N/A # XXX Mirroring records also need to be allowed from client

37N/A # configuration, and not just catalogs.

30N/A #

30N/A # XXX It would be nice to include available tags and package sizes,

37N/A # although this could also be calculated from the set of manifests.

37N/A #

37N/A # XXX self.pkgs should be a dictionary, accessed by fmri string (or

50N/A # package name). Current code is O(N_packages) O(M_versions), should be

50N/A # O(1) O(M_versions), and possibly O(1) O(1). Likely a similar need for

50N/A # critical_pkgs.

50N/A #

50N/A # XXX Initial estimates suggest that the Catalog could be composed of

50N/A # 1e5 - 1e7 lines. Catalogs across these magnitudes will need to be

26N/A # spread out into chunks, and may require a delta-oriented update

23N/A # interface.

23N/A

34N/A def __init__(self):

34N/A self.catalog_root = ""

50N/A

50N/A self.attrs = {}

50N/A self.attrs["Last-Modified"] = time.time()

45N/A

26N/A self.authorities = {}

50N/A self.pkgs = []

50N/A self.critical_pkgs = []

24N/A

23N/A self.relns = {}

45N/A

45N/A def add_authority(self, authority, urls):

45N/A self.authorities[authority] = urls

45N/A

45N/A def delete_authority(self, authority):

34N/A del(self.authorities[authority])

34N/A

34N/A def set_catalog_root(self, croot):

34N/A self.catalog_root = croot

36N/A

36N/A def load(self, path):

36N/A self.path = path

36N/A cfile = file(path, "r")

36N/A centries = cfile.readlines()

36N/A

36N/A for entry in centries:

36N/A # each V line is an fmri

50N/A m = re.match("^V (pkg:[^ ]*)", entry)

50N/A if m != None:

50N/A pname = m.group(1)

36N/A self.add_fmri(fmri.PkgFmri(pname, None))

50N/A

50N/A m = re.match("^S ([^:]*): (.*)", entry)

50N/A if m != None:

50N/A self.attrs[m.group(1)] = m.group(2)

50N/A

50N/A m = re.match("^C (pkg:[^ ]*)", entry)

50N/A if m != None:

50N/A pname = m.group(1)

50N/A self.critical_pkgs.append(fmri.PkgFmri(pname,

36N/A None))

36N/A

36N/A def add_fmri(self, pkgfmri):

36N/A name = pkgfmri.get_pkg_stem()

36N/A pfmri = fmri.PkgFmri(name, None)

36N/A

39N/A for pkg in self.pkgs:

36N/A if pkg.fmri.is_same_pkg(pfmri):

36N/A pkg.add_version(pkgfmri)

36N/A return

36N/A

36N/A pkg = package.Package(pfmri)

36N/A pkg.add_version(pkgfmri)

36N/A self.pkgs.append(pkg)

50N/A

36N/A def add_pkg(self, pkg, critical = False):

36N/A for opkg in self.pkgs:

37N/A if pkg.fmri == opkg.fmri:

37N/A #

37N/A # XXX This package is already in the catalog

37N/A # with some version set. Are we updating the

37N/A # version set or merging the two?

37N/A #

37N/A opkg = pkg

26N/A return

37N/A

50N/A self.pkgs.append(pkg)

50N/A if critical:

34N/A self.critical_pkgs.append(pkg)

36N/A

50N/A def get_matching_pkgs(self, pfmri, constraint):

50N/A """Iterate through the catalogs, looking for an exact fmri

36N/A

30N/A for pkg in self.pkgs:

36N/A if pkg.fmri.is_similar(pfmri):

30N/A return pkg.matching_versions(pfmri, constraint)

30N/A

50N/A raise KeyError, "FMRI '%s' not found in catalog" % pfmri

50N/A

50N/A def get_regex_matching_fmris(self, regex):

50N/A """Iterate through the catalogs, looking for a regular

50N/A

50N/A ret = []

50N/A

50N/A for p in self.pkgs:

50N/A for pv in p.pversions:

50N/A fn = "%s@%s" % (p.fmri, pv.version)

50N/A if re.search(regex, fn):

50N/A ret.append(fmri.PkgFmri(fn, None))

50N/A

50N/A if ret == []:

50N/A raise KeyError, "pattern '%s' not found in catalog" \

50N/A % regex

50N/A

50N/A return ret

30N/A

24N/A def __str__(self):

24N/A s = ""

50N/A for a in self.attrs.keys():

50N/A s = s + "S %s: %s\n" % (a, self.attrs[a])

26N/A for p in self.pkgs:

36N/A s = s + p.get_catalog_entry()

50N/A for c in self.critical_pkgs:

50N/A s = s + "C %s\n" % p

24N/A for r in self.relns:

26N/A s = s + "I %s\n" % r

24N/A return s

24N/A

45N/A def display(self):

45N/A """XXX -a to see all versions of all packages. Otherwise

45N/A

45N/A # XXX no access to image means that client use of this function

45N/A # limited to unsophisticated output...

34N/A

34N/A if len(self.pkgs) == 0:

34N/A print "pkg: catalog is empty"

34N/A return

34N/A

34N/A for p in self.pkgs:

45N/A print "%-50s" % p.fmri

45N/A for v in p.pversions:

45N/A print " %20s" % v.version

45N/A

45N/A def difference(self, catalog):

45N/A """Return a pair of lists, the first list being those package

45N/A return

45N/A

45N/Aif __name__ == "__main__":

45N/A c = Catalog()

45N/A

45N/A for f in [

45N/A fmri.PkgFmri("pkg:/test@1.0,5.11-1:20000101T120000Z", None),

45N/A fmri.PkgFmri("pkg:/test@1.0,5.11-1:20000101T120010Z", None),

45N/A fmri.PkgFmri("pkg:/test@1.0,5.11-1.1:20000101T120020Z", None),

45N/A fmri.PkgFmri("pkg:/test@1.0,5.11-1.2:20000101T120030Z", None),

45N/A fmri.PkgFmri("pkg:/test@1.0,5.11-2:20000101T120040Z", None),

45N/A fmri.PkgFmri("pkg:/test@1.1,5.11-1:20000101T120040Z", None)

45N/A ]:

45N/A c.add_fmri(f)

45N/A

45N/A print c

45N/A

45N/A tps = [

45N/A "pkg:/test@1.0,5.10-1:20070101T120000Z",

50N/A "pkg:/test@1.0,5.11-1:20061231T120000Z",

50N/A "pkg:/test@1.0,5.11-2",

50N/A "pkg:/test@1.0,5.11-3"

50N/A ]

50N/A

50N/A for tp in tps:

50N/A print "matches for %s:" % tp

50N/A

for p in c.get_matching_pkgs(fmri.PkgFmri(tp, None), None):

print " ", p

for p in c.get_regex_matching_fmris("test"):

print p

try:

l = c.get_regex_matching_fmris("flob")

except KeyError:

print "correctly determined no match for 'flob'"