/[gli]/branches/overhaul/src/GLIPortage.py
Gentoo

Contents of /branches/overhaul/src/GLIPortage.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1183 - (hide annotations) (download) (as text)
Fri Dec 23 22:39:29 2005 UTC (8 years, 3 months ago) by agaffney
Original Path: trunk/src/GLIPortage.py
File MIME type: text/x-python
File size: 10302 byte(s)
flipped around a few function args

1 agaffney 1181 """
2     # Copyright 1999-2005 Gentoo Foundation
3     # This source code is distributed under the terms of version 2 of the GNU
4     # General Public License as published by the Free Software Foundation, a copy
5     # of which can be found in the main directory of this project.
6     Gentoo Linux Installer
7    
8 agaffney 1183 $Id: GLIPortage.py,v 1.3 2005/12/23 22:39:29 agaffney Exp $
9 agaffney 1181 """
10    
11     import os
12 agaffney 1182 import re
13 agaffney 1181 import GLIUtility
14    
15     class MissingPackagesError(Exception):
16     pass
17    
18     class depgraph:
19    
20     def __init__(self):
21     self.graph = {}
22    
23     def add(self, node, parent=None):
24     if node not in self.graph:
25     self.graph[node] = [[], []]
26     if parent and parent not in self.graph[node][0]:
27     if parent not in self.graph:
28     self.graph[parent] = [[], []]
29     self.graph[node][0].append(parent)
30     self.graph[parent][1].append(node)
31    
32     def remove(self, node):
33     for parent in self.graph[node][0]:
34     self.graph[parent][1].remove(node)
35     for child in self.graph[node][1]:
36     self.graph[child][0].remove(node)
37     return self.graph.pop(node)
38    
39     def has_node(self, node):
40     return node in self.graph
41    
42     def leaf_nodes(self):
43     return [node for node in self.graph if not self.graph[node][1]]
44    
45     def node_count(self):
46     return len(self.graph)
47    
48     def important_node(self):
49     important_node = None
50     importance = 0
51     for node in self.graph:
52     if len(self.graph[node][0]) > importance:
53     importance = len(self.graph[node][0])
54     important_node = node
55     return important_node
56    
57     class GLIPortage(object):
58    
59 agaffney 1183 def __init__(self, chroot_dir, grp_install, logger, debug):
60 agaffney 1181 os.environ['ROOT'] = chroot_dir
61     import portage, portage_dep
62     self.vdb = portage.db["/"]["vartree"].dbapi
63 agaffney 1182 self.tree = portage.db[chroot_dir]["porttree"].dbapi
64     self._chroot_dir = chroot_dir
65     self._logger = logger
66     self._debug = debug
67 agaffney 1183 self._grp_install = grp_install
68 agaffney 1181
69     def resolve_deps(self, dep_list):
70     if dep_list and dep_list[0] == "||":
71     for dep in dep_list[1:]:
72     if isinstance(dep, list):
73     try:
74     atoms = resolve_deps(dep)
75     except MissingPackagesError:
76     continue
77     else:
78     atoms = [dep]
79     all_found = True
80     for atom in atoms:
81     if not atom.startswith("!") and not self.vdb.match(atom):
82     all_found = False
83     break
84     if all_found:
85     return atoms
86     raise MissingPackagesError(dep_list)
87     atoms = []
88     for dep in dep_list:
89     if isinstance(dep, list):
90     atoms.extend(resolve_deps(dep))
91     elif not dep.startswith("!"):
92     if not self.vdb.match(dep):
93     raise MissingPackagesError([dep])
94     atoms.append(dep)
95     return atoms
96    
97     def calc_required_pkgs(self, atom, graph, parent=None):
98     pkg = portage.best(self.vdb.match(atom))
99     if not pkg:
100     raise MissingPackagesError([atom])
101     if pkg == parent:
102     return
103     already_processed = graph.has_node(pkg)
104     graph.add(pkg, parent)
105     if already_processed:
106     return
107     useflags = self.vdb.aux_get(pkg, ["USE"])[0].split()
108     rdep_raw = " ".join(self.vdb.aux_get(pkg, ["RDEPEND", "PDEPEND"]))
109     rdep_struct = portage_dep.use_reduce(portage_dep.paren_reduce(rdep_raw), uselist=useflags)
110     rdep_struct = portage.dep_virtual(portage_dep.dep_opconvert(rdep_struct), portage.settings)
111     rdep_atoms = portage.unique_array(resolve_deps(rdep_struct))
112     for atom in rdep_atoms:
113     calc_required_pkgs(atom, graph, pkg)
114    
115     def prune_existing(self, graph):
116     db = portage.db[portage.root]["vartree"].dbapi
117     for atom in db.cp_all():
118     for pkg in db.match(atom):
119     if graph.has_node(pkg):
120     graph.remove(pkg)
121    
122 agaffney 1183 def get_deps(self, pkgs):
123     if not self._grp_install:
124 agaffney 1181 del(os.environ['ROOT'])
125 agaffney 1182 return GLIUtility.spawn("emerge -p " + pkgs + r" | grep -e '^\[[a-z]' | cut -d ']' -f2 | sed -e 's:^ ::' -e 's: .\+$::'", chroot=self._chroot_dir, return_output=True)[1].split("\n")
126     os.environ['ROOT'] = self._chroot_dir
127 agaffney 1181 else:
128     pkglist = []
129     graph = depgraph()
130     for pkg in pkgs.split():
131     if not self.vdb.match(pkg):
132 agaffney 1182 self._logger.log("get_deps(): " + pkg + " is not available via GRP...ignoring")
133 agaffney 1181 continue
134     self.calc_required_pkgs(pkg, graph)
135     while graph.node_count():
136     leaf_nodes = graph.leaf_nodes()
137     if not leaf_nodes:
138     node = graph.important_node()
139     pkglist.append(node)
140     graph.remove(node)
141     continue
142     pkglist.extend(leaf_nodes)
143     for node in leaf_nodes:
144     graph.remove(node)
145     return pkglist
146 agaffney 1182
147     def copy_pkg_to_chroot(self, package):
148     symlinks = { '/bin/': '/mnt/livecd/bin/', '/boot/': '/mnt/livecd/boot/', '/lib/': '/mnt/livecd/lib/',
149     '/opt/': '/mnt/livecd/opt/', '/sbin/': '/mnt/livecd/sbin/', '/usr/': '/mnt/livecd/usr/',
150     '/etc/gconf/': '/usr/livecd/gconf/' }
151    
152     # Copy the vdb entry for the package from the LiveCD to the chroot
153     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): copying vdb entry for " + package)
154     if not GLIUtility.exitsuccess(GLIUtility.spawn("mkdir -p " + self._chroot_dir + "/var/db/pkg/" + package + " && cp -a /var/db/pkg/" + package + "/* " + self._chroot_dir + "/var/db/pkg/" + package)):
155     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not copy vdb entry for " + package)
156    
157     # Create list of files for tar to work with from CONTENTS file in vdb entry
158     entries = GLIUtility.parse_vdb_contents("/var/db/pkg/" + package + "/CONTENTS")
159     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot: files for " + package + ": " + str(entries))
160     try:
161     tarfiles = open("/tmp/tarfilelist", "w")
162     for entry in entries:
163     parts = entry.split(" ")
164     # Hack for symlink crappiness
165     for symlink in symlinks:
166     if parts[0].startswith(symlink):
167     parts[0] = symlinks[symlink] + parts[0][len(symlink):]
168     tarfiles.write(parts[0] + "\n")
169     tarfiles.close()
170     except:
171     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not create filelist for " + package)
172    
173     # Use tar to transfer files into IMAGE directory
174     tmpdir = "/var/tmp/portage"
175     image_dir = tmpdir + "/" + package.split("/")[1] + "/image"
176     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running 'mkdir -p " + self._chroot_dir + image_dir + " && tar -c --files-from=/tmp/tarfilelist --no-recursion 2>/dev/null | tar -C " + self._chroot_dir + image_dir + " -x'")
177     if not GLIUtility.exitsuccess(GLIUtility.spawn("mkdir -p " + self._chroot_dir + image_dir + " && tar -c --files-from=/tmp/tarfilelist --no-recursion 2>/dev/null | tar -C " + self._chroot_dir + image_dir + " -x")):
178     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute tar for " + package)
179    
180     # More symlink crappiness hacks
181     for symlink in symlinks:
182     if GLIUtility.is_file(self._chroot_dir + image_dir + symlinks[symlink]):
183     # parts[0] = symlinks[symlink] + parts[len(symlink):]
184     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): fixing /usr/livecd/gconf/ stuff in " + image_dir + " for " + package)
185     if not GLIUtility.exitsuccess(GLIUtility.spawn("mv " + self._chroot_dir + image_dir + symlinks[symlink] + " " + self._chroot_dir + image_dir + symlink)):
186     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not fix /usr/livecd/gconf/ stuff for " + package)
187    
188     # Run pkg_setup
189     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running pkg_setup for " + package)
190     if not GLIUtility.exitsuccess(GLIUtility.spawn("env ROOT=" + self._chroot_dir + " PORTAGE_TMPDIR=" + self._chroot_dir + tmpdir + " ebuild " + self._chroot_dir + "/var/db/pkg/" + package + "/*.ebuild setup")):
191     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute pkg_setup for " + package)
192    
193     # Run qmerge
194     # if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running qmerge for " + package)
195     # if not GLIUtility.exitsuccess(GLIUtility.spawn("env ROOT=" + self._chroot_dir + " PORTAGE_TMPDIR=" + self._chroot_dir + tmpdir + " ebuild " + self._chroot_dir + "/var/db/pkg/" + package + "/*.ebuild qmerge")):
196     # raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute qmerge for " + package)
197    
198     # Run pkg_preinst
199     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running preinst for " + package)
200     if not GLIUtility.exitsuccess(GLIUtility.spawn("env ROOT=" + self._chroot_dir + " PORTAGE_TMPDIR=" + self._chroot_dir + tmpdir + " ebuild " + self._chroot_dir + "/var/db/pkg/" + package + "/*.ebuild preinst")):
201     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute preinst for " + package)
202    
203     # Copy files from image_dir to chroot
204     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): copying files from " + image_dir + " to / for " + package)
205     if not GLIUtility.exitsuccess(GLIUtility.spawn("cp -a " + self._chroot_dir + image_dir + "/* " + self._chroot_dir)):
206     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not copy files from " + image_dir + " to / for " + package)
207    
208     # Run pkg_postinst
209     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running postinst for " + package)
210     if not GLIUtility.exitsuccess(GLIUtility.spawn("env ROOT=" + self._chroot_dir + " PORTAGE_TMPDIR=" + self._chroot_dir + tmpdir + " ebuild " + "/var/db/pkg/" + package + "/*.ebuild postinst")):
211     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute postinst for " + package)
212    
213     # Remove image_dir
214     if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): removing + " + image_dir + " for " + package)
215     if not GLIUtility.exitsuccess(GLIUtility.spawn("rm -rf " + self._chroot_dir + image_dir)):
216     raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not remove + " + image_dir + " for " + package)
217    
218     def add_pkg_to_world(self, package):
219     if package.find("/") == -1:
220     package = GLIUtility.spawn("portageq best_version / " + package, chroot=self._chroot_dir, return_output=True)[1].strip()
221     if not package: return False
222     expr = re.compile('^(.+?)(-\d.+)?$')
223     res = expr.match(package)
224     if res:
225     GLIUtility.spawn("echo " + res.group(1) + " >> " + self._chroot_dir + "/var/lib/portage/world")
226    
227     def get_best_version_vdb(self, package):
228     return portage.best(vdb.match(package))
229    
230     def get_best_version_tree(self, package):
231     return portage.best(tree.match(package))

Properties

Name Value
svn:eol-style native

  ViewVC Help
Powered by ViewVC 1.1.20