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

Diff of /trunk/src/GLIPortage.py

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

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

Legend:
Removed from v.1185  
changed lines
  Added in v.1210

  ViewVC Help
Powered by ViewVC 1.1.20