/[gli]/trunk/src/GLIPortage.py
Gentoo

Diff of /trunk/src/GLIPortage.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1181 Revision 1223
3# This source code is distributed under the terms of version 2 of the GNU 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 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. 5# of which can be found in the main directory of this project.
6Gentoo Linux Installer 6Gentoo Linux Installer
7 7
8$Id: GLIPortage.py,v 1.1 2005/12/23 21:14:08 agaffney Exp $ 8$Id: GLIPortage.py,v 1.32 2006/01/07 17:12:00 agaffney Exp $
9""" 9"""
10 10
11import re
11import os 12import os
12import GLIUtility 13import GLIUtility
13 14from GLIException import GLIException
14class MissingPackagesError(Exception):
15 pass
16
17class depgraph:
18
19 def __init__(self):
20 self.graph = {}
21
22 def add(self, node, parent=None):
23 if node not in self.graph:
24 self.graph[node] = [[], []]
25 if parent and parent not in self.graph[node][0]:
26 if parent not in self.graph:
27 self.graph[parent] = [[], []]
28 self.graph[node][0].append(parent)
29 self.graph[parent][1].append(node)
30
31 def remove(self, node):
32 for parent in self.graph[node][0]:
33 self.graph[parent][1].remove(node)
34 for child in self.graph[node][1]:
35 self.graph[child][0].remove(node)
36 return self.graph.pop(node)
37
38 def has_node(self, node):
39 return node in self.graph
40
41 def leaf_nodes(self):
42 return [node for node in self.graph if not self.graph[node][1]]
43
44 def node_count(self):
45 return len(self.graph)
46
47 def important_node(self):
48 important_node = None
49 importance = 0
50 for node in self.graph:
51 if len(self.graph[node][0]) > importance:
52 importance = len(self.graph[node][0])
53 important_node = node
54 return important_node
55 15
56class GLIPortage(object): 16class GLIPortage(object):
57 17
58 def __init__(self, chroot_dir): 18 def __init__(self, chroot_dir, grp_install, logger, debug, cc, compile_logfile):
59 os.environ['ROOT'] = chroot_dir
60 import portage, portage_dep
61 self.vdb = portage.db["/"]["vartree"].dbapi
62 self.chroot_dir = chroot_dir 19 self._chroot_dir = chroot_dir
20 self._grp_install = grp_install
21 self._logger = logger
22 self._debug = debug
23 self._cc = cc
24 self._compile_logfile = compile_logfile
63 25
64 def resolve_deps(self, dep_list): 26 def get_deps(self, pkgs):
65 if dep_list and dep_list[0] == "||": 27 pkglist = []
66 for dep in dep_list[1:]:
67 if isinstance(dep, list): 28 if isinstance(pkgs, str):
68 try: 29 pkgs = pkgs.split()
69 atoms = resolve_deps(dep) 30 for pkg in pkgs:
70 except MissingPackagesError: 31 if self._debug: self._logger.log("get_deps(): pkg is " + pkg)
71 continue 32 if not self._grp_install or not self.get_best_version_vdb(pkg):
33 if self._debug: self._logger.log("get_deps(): grabbing compile deps")
34 tmppkglist = GLIUtility.spawn("emerge -p " + pkg + r" 2>/dev/null | grep -e '^\[[a-z]' | cut -d ']' -f2 | sed -e 's:^ ::' -e 's: .\+$::'", chroot=self._chroot_dir, return_output=True)[1].strip().split("\n")
72 else: 35 else:
73 atoms = [dep] 36 if self._debug: self._logger.log("get_deps(): grabbing binary deps")
74 all_found = True 37 # The runtimedeps.py script generates a package install order that is *very* different from emerge itself
75 for atom in atoms: 38# tmppkglist = GLIUtility.spawn("python ../../runtimedeps.py " + self._chroot_dir + " " + pkg, return_output=True)[1].strip().split("\n")
76 if not atom.startswith("!") and not self.vdb.match(atom): 39 tmppkglist = []
77 all_found = False 40 for tmppkg in GLIUtility.spawn("emerge -p " + pkg + r" 2>/dev/null | grep -e '^\[[a-z]' | cut -d ']' -f2 | sed -e 's:^ ::' -e 's: .\+$::'", chroot=self._chroot_dir, return_output=True)[1].strip().split("\n"):
78 break 41 if self._debug: self._logger.log("get_deps(): looking at " + tmppkg)
79 if all_found: 42 if self.get_best_version_vdb("=" + tmppkg):
80 return atoms 43 if self._debug: self._logger.log("get_deps(): package " + tmppkg + " in host vdb...adding to tmppkglist")
81 raise MissingPackagesError(dep_list) 44 tmppkglist.append(tmppkg)
82 atoms = [] 45 if self._debug: self._logger.log("get_deps(): deplist for " + pkg + ": " + str(tmppkglist))
83 for dep in dep_list: 46 for tmppkg in tmppkglist:
84 if isinstance(dep, list): 47 if self._debug: self._logger.log("get_deps(): checking to see if " + tmppkg + " is already in pkglist")
85 atoms.extend(resolve_deps(dep)) 48 if not tmppkg in pkglist and not self.get_best_version_vdb_chroot("=" + tmppkg):
86 elif not dep.startswith("!"): 49 if self._debug: self._logger.log("get_deps(): adding " + tmppkg + " to pkglist")
87 if not self.vdb.match(dep): 50 pkglist.append(tmppkg)
88 raise MissingPackagesError([dep]) 51 if self._debug: self._logger.log("get_deps(): pkglist is " + str(pkglist))
89 atoms.append(dep) 52 return pkglist
90 return atoms
91 53
92 def calc_required_pkgs(self, atom, graph, parent=None): 54 def parse_vdb_contents(self, file):
93 pkg = portage.best(self.vdb.match(atom)) 55 entries = []
94 if not pkg: 56 try:
95 raise MissingPackagesError([atom]) 57 vdbfile = open(file, "r")
96 if pkg == parent: 58 except:
97 return 59 return entries
98 already_processed = graph.has_node(pkg) 60 for line in vdbfile.readlines():
99 graph.add(pkg, parent) 61 parts = line.strip().split(" ")
100 if already_processed: 62 if parts[0] == "obj":
101 return 63 entries.append(parts[1])
102 useflags = self.vdb.aux_get(pkg, ["USE"])[0].split() 64# elif parts[0] == "dir":
103 rdep_raw = " ".join(self.vdb.aux_get(pkg, ["RDEPEND", "PDEPEND"])) 65# entries.append(parts[1] + "/")
104 rdep_struct = portage_dep.use_reduce(portage_dep.paren_reduce(rdep_raw), uselist=useflags) 66 elif parts[0] == "sym":
105 rdep_struct = portage.dep_virtual(portage_dep.dep_opconvert(rdep_struct), portage.settings) 67 entries.append(" ".join(parts[1:4]))
106 rdep_atoms = portage.unique_array(resolve_deps(rdep_struct)) 68 entries.sort()
107 for atom in rdep_atoms: 69 return entries
108 calc_required_pkgs(atom, graph, pkg)
109 70
110 def prune_existing(self, graph): 71 def copy_pkg_to_chroot(self, package, use_root=False):
111 db = portage.db[portage.root]["vartree"].dbapi 72 symlinks = { '/bin': '/mnt/livecd/bin/', '/boot': '/mnt/livecd/boot/', '/lib': '/mnt/livecd/lib/',
112 for atom in db.cp_all(): 73 '/opt': '/mnt/livecd/opt/', '/sbin': '/mnt/livecd/sbin/', '/usr': '/mnt/livecd/usr/',
113 for pkg in db.match(atom): 74 '/etc/gconf': '/usr/livecd/gconf/' }
114 if graph.has_node(pkg):
115 graph.remove(pkg)
116 75
117 def get_deps(self, pkgs, grp_install): 76 tmpdir = "/var/tmp/portage"
118 if not grp_install: 77 image_dir = tmpdir + "/" + package.split("/")[1] + "/image"
119 del(os.environ['ROOT']) 78 root_cmd = ""
120 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") 79 tmp_chroot_dir = self._chroot_dir
121 os.environ['ROOT'] = self.chroot_dir 80 portage_tmpdir = "/var/tmp/portage"
81 if use_root:
82 root_cmd = "ROOT=" + self._chroot_dir
83 tmp_chroot_dir = ""
84 portage_tmpdir = self._chroot_dir + "/var/tmp/portage"
85
86 # Copy the vdb entry for the package from the LiveCD to the chroot
87 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): copying vdb entry for " + package)
88 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)):
89 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not copy vdb entry for " + package)
90
91 # Create the image dir in the chroot
92 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running 'mkdir -p " + self._chroot_dir + image_dir + "'")
93 if not GLIUtility.exitsuccess(GLIUtility.spawn("mkdir -p " + self._chroot_dir + image_dir)):
94 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not create image dir for " + package)
95
96 # Create list of files for tar to work with from CONTENTS file in vdb entry
97 entries = self.parse_vdb_contents("/var/db/pkg/" + package + "/CONTENTS")
98 if not entries:
99 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): no files for " + package + "...skipping tar and symlink fixup")
122 else: 100 else:
123 pkglist = [] 101 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot: files for " + package + ": " + str(entries))
124 graph = depgraph() 102 try:
125 for pkg in pkgs.split(): 103 tarfiles = open("/tmp/tarfilelist", "w")
126 if not self.vdb.match(pkg): 104 for entry in entries:
127 continue 105 parts = entry.split(" ")
128 self.calc_required_pkgs(pkg, graph) 106# # Hack for symlink crappiness
129 while graph.node_count(): 107# for symlink in symlinks:
130 leaf_nodes = graph.leaf_nodes() 108# if parts[0].startswith(symlink):
131 if not leaf_nodes: 109# parts[0] = symlinks[symlink] + parts[0][len(symlink):]
132 node = graph.important_node() 110 tarfiles.write(parts[0] + "\n")
133 pkglist.append(node) 111 tarfiles.close()
134 graph.remove(node) 112 except:
135 continue 113 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not create filelist for " + package)
136 pkglist.extend(leaf_nodes) 114
137 for node in leaf_nodes: 115 # Use tar to transfer files into IMAGE directory
138 graph.remove(node) 116 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running 'tar -cp --files-from=/tmp/tarfilelist --no-recursion 2>/dev/null | tar -C " + self._chroot_dir + image_dir + " -xp'")
139 return pkglist 117 if not GLIUtility.exitsuccess(GLIUtility.spawn("tar -cp --files-from=/tmp/tarfilelist --no-recursion 2>/dev/null | tar -C " + self._chroot_dir + image_dir + " -xp")):
118 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute tar for " + package)
119
120# # More symlink crappiness hacks
121# for symlink in symlinks:
122## if GLIUtility.is_file(self._chroot_dir + image_dir + symlinks[symlink]):
123# if os.path.islink(self._chroot_dir + image_dir + symlink):
124# if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): fixing " + symlink + " symlink ickiness stuff in " + image_dir + " for " + package)
125# GLIUtility.spawn("rm " + self._chroot_dir + image_dir + symlink)
126# if not GLIUtility.exitsuccess(GLIUtility.spawn("mv " + self._chroot_dir + image_dir + symlinks[symlink] + " " + self._chroot_dir + image_dir + symlink)):
127# raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not fix " + symlink + " symlink ickiness for " + package)
128
129 # Run pkg_setup
130 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running pkg_setup for " + package)
131 if not GLIUtility.exitsuccess(GLIUtility.spawn("env " + root_cmd + " PORTAGE_TMPDIR=" + portage_tmpdir + " ebuild /var/db/pkg/" + package + "/*.ebuild setup", chroot=tmp_chroot_dir)):
132 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute pkg_setup for " + package)
133
134 # Run pkg_preinst
135 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running preinst for " + package)
136 if not GLIUtility.exitsuccess(GLIUtility.spawn("env " + root_cmd + " PORTAGE_TMPDIR=" + portage_tmpdir + " ebuild /var/db/pkg/" + package + "/*.ebuild preinst", chroot=tmp_chroot_dir)):
137 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute preinst for " + package)
138
139 # Copy files from image_dir to chroot
140 if not entries:
141 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): no files for " + package + "...skipping copy from image dir to /")
142 else:
143 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): copying files from " + image_dir + " to / for " + package)
144 if not GLIUtility.exitsuccess(GLIUtility.spawn("cp -a " + self._chroot_dir + image_dir + "/* " + self._chroot_dir)):
145 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not copy files from " + image_dir + " to / for " + package)
146
147 # Run pkg_postinst
148 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): running postinst for " + package)
149 if not GLIUtility.exitsuccess(GLIUtility.spawn("env " + root_cmd + " PORTAGE_TMPDIR=" + portage_tmpdir + " ebuild /var/db/pkg/" + package + "/*.ebuild postinst", chroot=tmp_chroot_dir)):
150 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not execute postinst for " + package)
151
152 # Remove image_dir
153 if self._debug: self._logger.log("DEBUG: copy_pkg_to_chroot(): removing " + image_dir + " for " + package)
154 if not GLIUtility.exitsuccess(GLIUtility.spawn("rm -rf " + self._chroot_dir + image_dir)):
155 raise GLIException("CopyPackageToChrootError", 'fatal', 'copy_pkg_to_chroot', "Could not remove + " + image_dir + " for " + package)
156
157 def add_pkg_to_world(self, package):
158 if package.find("/") == -1:
159 package = self.get_best_version_vdb_chroot(package)
160 if not package: return False
161 expr = re.compile('^=?(.+?/.+?)(-\d.+)?$')
162 res = expr.match(package)
163 if res:
164 GLIUtility.spawn("echo " + res.group(1) + " >> " + self._chroot_dir + "/var/lib/portage/world")
165
166 def get_best_version_vdb(self, package):
167 return GLIUtility.spawn("portageq best_version / " + package, return_output=True)[1].strip()
168
169 def get_best_version_vdb_chroot(self, package):
170 return GLIUtility.spawn("portageq best_version / " + package, chroot=self._chroot_dir, return_output=True)[1].strip()
171
172# def get_best_version_tree(self, package):
173# return portage.best(tree.match(package))
174
175 def emerge(self, packages, add_to_world=True):
176 if isinstance(packages, str):
177 packages = packages.split()
178 self._cc.addNotification("progress", (0, "Calculating dependencies for " + " ".join(packages)))
179 pkglist = self.get_deps(packages)
180 if self._debug: self._logger.log("install_packages(): pkglist is " + str(pkglist))
181 for i, pkg in enumerate(pkglist):
182 if self._debug: self._logger.log("install_packages(): processing package " + pkg)
183 self._cc.addNotification("progress", (float(i) / len(pkglist), "Emerging " + pkg + " (" + str(i+1) + "/" + str(len(pkglist)) + ")"))
184 if not self._grp_install or not self.get_best_version_vdb("=" + pkg):
185 status = GLIUtility.spawn("emerge -1 =" + pkg, display_on_tty8=True, chroot=self._chroot_dir, logfile=self._compile_logfile, append_log=True)
186# status = self._emerge("=" + pkg)
187 if not GLIUtility.exitsuccess(status):
188 raise GLIException("EmergePackageError", "fatal", "emerge", "Could not emerge " + pkg + "!")
189 else:
190# try:
191 self.copy_pkg_to_chroot(pkg)
192# except:
193# raise GLIException("EmergePackageError", "fatal", "emerge", "Could not emerge " + pkg + "!")
194 self._cc.addNotification("progress", (float(i+1) / len(pkglist), "Done emerging " + pkg + " (" + str(i+1) + "/" + str(len(pkglist)) + ")"))
195 if add_to_world:
196 for package in packages:
197 self.add_pkg_to_world(package)

Legend:
Removed from v.1181  
changed lines
  Added in v.1223

  ViewVC Help
Powered by ViewVC 1.1.20