Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 1 | """Package Install Manager for Python. |
| 2 | |
| 3 | This is currently a MacOSX-only strawman implementation. |
| 4 | Motto: "He may be shabby, but he gets you what you need" :-) |
| 5 | |
| 6 | Tools to allow easy installation of packages. The idea is that there is |
| 7 | an online XML database per (platform, python-version) containing packages |
| 8 | known to work with that combination. This module contains tools for getting |
| 9 | and parsing the database, testing whether packages are installed, computing |
| 10 | dependencies and installing packages. |
| 11 | |
| 12 | There is a minimal main program that works as a command line tool, but the |
| 13 | intention is that the end user will use this through a GUI. |
| 14 | """ |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 15 | import sys |
| 16 | import os |
| 17 | import urllib |
| 18 | import urlparse |
| 19 | import plistlib |
| 20 | import distutils.util |
Jack Jansen | e71b9f8 | 2003-02-12 16:37:00 +0000 | [diff] [blame] | 21 | import distutils.sysconfig |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 22 | import md5 |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 23 | |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 24 | __all__ = ["PimpPreferences", "PimpDatabase", "PimpPackage", "main"] |
| 25 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 26 | _scriptExc_NotInstalled = "pimp._scriptExc_NotInstalled" |
| 27 | _scriptExc_OldInstalled = "pimp._scriptExc_OldInstalled" |
| 28 | _scriptExc_BadInstalled = "pimp._scriptExc_BadInstalled" |
| 29 | |
| 30 | NO_EXECUTE=0 |
| 31 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 32 | PIMP_VERSION="0.1" |
| 33 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 34 | # Flavors: |
| 35 | # source: setup-based package |
| 36 | # binary: tar (or other) archive created with setup.py bdist. |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 37 | DEFAULT_FLAVORORDER=['source', 'binary'] |
| 38 | DEFAULT_DOWNLOADDIR='/tmp' |
| 39 | DEFAULT_BUILDDIR='/tmp' |
Jack Jansen | e71b9f8 | 2003-02-12 16:37:00 +0000 | [diff] [blame] | 40 | DEFAULT_INSTALLDIR=distutils.sysconfig.get_python_lib() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 41 | DEFAULT_PIMPDATABASE="http://www.cwi.nl/~jack/pimp/pimp-%s.plist" % distutils.util.get_platform() |
| 42 | |
| 43 | ARCHIVE_FORMATS = [ |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 44 | (".tar.Z", "zcat \"%s\" | tar -xf -"), |
| 45 | (".taz", "zcat \"%s\" | tar -xf -"), |
| 46 | (".tar.gz", "zcat \"%s\" | tar -xf -"), |
| 47 | (".tgz", "zcat \"%s\" | tar -xf -"), |
| 48 | (".tar.bz", "bzcat \"%s\" | tar -xf -"), |
Jack Jansen | a460f44 | 2003-02-17 12:21:05 +0000 | [diff] [blame] | 49 | (".zip", "unzip \"%s\""), |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 50 | ] |
| 51 | |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 52 | class MyURLopener(urllib.FancyURLopener): |
Jack Jansen | 8d326b8 | 2003-02-10 16:08:17 +0000 | [diff] [blame] | 53 | """Like FancyURLOpener, but we do want to get errors as exceptions.""" |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 54 | def http_error_default(self, url, fp, errcode, errmsg, headers): |
| 55 | urllib.URLopener.http_error_default(self, url, fp, errcode, errmsg, headers) |
| 56 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 57 | class PimpPreferences: |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 58 | """Container for per-user preferences, such as the database to use |
Jack Jansen | 8d326b8 | 2003-02-10 16:08:17 +0000 | [diff] [blame] | 59 | and where to install packages.""" |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 60 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 61 | def __init__(self, |
| 62 | flavorOrder=None, |
| 63 | downloadDir=None, |
| 64 | buildDir=None, |
| 65 | installDir=None, |
| 66 | pimpDatabase=None): |
| 67 | if not flavorOrder: |
| 68 | flavorOrder = DEFAULT_FLAVORORDER |
| 69 | if not downloadDir: |
| 70 | downloadDir = DEFAULT_DOWNLOADDIR |
| 71 | if not buildDir: |
| 72 | buildDir = DEFAULT_BUILDDIR |
| 73 | if not installDir: |
| 74 | installDir = DEFAULT_INSTALLDIR |
| 75 | if not pimpDatabase: |
| 76 | pimpDatabase = DEFAULT_PIMPDATABASE |
| 77 | self.flavorOrder = flavorOrder |
| 78 | self.downloadDir = downloadDir |
| 79 | self.buildDir = buildDir |
| 80 | self.installDir = installDir |
| 81 | self.pimpDatabase = pimpDatabase |
| 82 | |
| 83 | def check(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 84 | """Check that the preferences make sense: directories exist and are |
| 85 | writable, the install directory is on sys.path, etc.""" |
| 86 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 87 | rv = "" |
| 88 | RWX_OK = os.R_OK|os.W_OK|os.X_OK |
| 89 | if not os.path.exists(self.downloadDir): |
| 90 | rv += "Warning: Download directory \"%s\" does not exist\n" % self.downloadDir |
| 91 | elif not os.access(self.downloadDir, RWX_OK): |
| 92 | rv += "Warning: Download directory \"%s\" is not writable or not readable\n" % self.downloadDir |
| 93 | if not os.path.exists(self.buildDir): |
| 94 | rv += "Warning: Build directory \"%s\" does not exist\n" % self.buildDir |
| 95 | elif not os.access(self.buildDir, RWX_OK): |
| 96 | rv += "Warning: Build directory \"%s\" is not writable or not readable\n" % self.buildDir |
| 97 | if not os.path.exists(self.installDir): |
| 98 | rv += "Warning: Install directory \"%s\" does not exist\n" % self.installDir |
| 99 | elif not os.access(self.installDir, RWX_OK): |
| 100 | rv += "Warning: Install directory \"%s\" is not writable or not readable\n" % self.installDir |
| 101 | else: |
| 102 | installDir = os.path.realpath(self.installDir) |
| 103 | for p in sys.path: |
| 104 | try: |
| 105 | realpath = os.path.realpath(p) |
| 106 | except: |
| 107 | pass |
| 108 | if installDir == realpath: |
| 109 | break |
| 110 | else: |
| 111 | rv += "Warning: Install directory \"%s\" is not on sys.path\n" % self.installDir |
| 112 | return rv |
| 113 | |
| 114 | def compareFlavors(self, left, right): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 115 | """Compare two flavor strings. This is part of your preferences |
| 116 | because whether the user prefers installing from source or binary is.""" |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 117 | if left in self.flavorOrder: |
| 118 | if right in self.flavorOrder: |
| 119 | return cmp(self.flavorOrder.index(left), self.flavorOrder.index(right)) |
| 120 | return -1 |
| 121 | if right in self.flavorOrder: |
| 122 | return 1 |
| 123 | return cmp(left, right) |
| 124 | |
| 125 | class PimpDatabase: |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 126 | """Class representing a pimp database. It can actually contain |
| 127 | information from multiple databases through inclusion, but the |
| 128 | toplevel database is considered the master, as its maintainer is |
Jack Jansen | 8d326b8 | 2003-02-10 16:08:17 +0000 | [diff] [blame] | 129 | "responsible" for the contents.""" |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 130 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 131 | def __init__(self, prefs): |
| 132 | self._packages = [] |
| 133 | self.preferences = prefs |
| 134 | self._urllist = [] |
| 135 | self._version = "" |
| 136 | self._maintainer = "" |
| 137 | self._description = "" |
| 138 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 139 | def close(self): |
| 140 | """Clean up""" |
| 141 | self._packages = [] |
| 142 | self.preferences = None |
| 143 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 144 | def appendURL(self, url, included=0): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 145 | """Append packages from the database with the given URL. |
| 146 | Only the first database should specify included=0, so the |
| 147 | global information (maintainer, description) get stored.""" |
| 148 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 149 | if url in self._urllist: |
| 150 | return |
| 151 | self._urllist.append(url) |
Jack Jansen | 26bf3ac | 2003-02-10 14:19:14 +0000 | [diff] [blame] | 152 | fp = MyURLopener().open(url).fp |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 153 | dict = plistlib.Plist.fromFile(fp) |
| 154 | # Test here for Pimp version, etc |
| 155 | if not included: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 156 | self._version = dict.get('Version', '0.1') |
| 157 | if self._version != PIMP_VERSION: |
| 158 | sys.stderr.write("Warning: database version %s does not match %s\n" |
| 159 | % (self._version, PIMP_VERSION)) |
| 160 | self._maintainer = dict.get('Maintainer', '') |
| 161 | self._description = dict.get('Description', '') |
| 162 | self._appendPackages(dict['Packages']) |
| 163 | others = dict.get('Include', []) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 164 | for url in others: |
| 165 | self.appendURL(url, included=1) |
| 166 | |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 167 | def _appendPackages(self, packages): |
| 168 | """Given a list of dictionaries containing package |
| 169 | descriptions create the PimpPackage objects and append them |
| 170 | to our internal storage.""" |
| 171 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 172 | for p in packages: |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 173 | p = dict(p) |
| 174 | flavor = p.get('Flavor') |
| 175 | if flavor == 'source': |
| 176 | pkg = PimpPackage_source(self, p) |
| 177 | elif flavor == 'binary': |
| 178 | pkg = PimpPackage_binary(self, p) |
| 179 | else: |
| 180 | pkg = PimpPackage(self, dict(p)) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 181 | self._packages.append(pkg) |
| 182 | |
| 183 | def list(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 184 | """Return a list of all PimpPackage objects in the database.""" |
| 185 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 186 | return self._packages |
| 187 | |
| 188 | def listnames(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 189 | """Return a list of names of all packages in the database.""" |
| 190 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 191 | rv = [] |
| 192 | for pkg in self._packages: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 193 | rv.append(pkg.fullname()) |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 194 | rv.sort() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 195 | return rv |
| 196 | |
| 197 | def dump(self, pathOrFile): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 198 | """Dump the contents of the database to an XML .plist file. |
| 199 | |
| 200 | The file can be passed as either a file object or a pathname. |
| 201 | All data, including included databases, is dumped.""" |
| 202 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 203 | packages = [] |
| 204 | for pkg in self._packages: |
| 205 | packages.append(pkg.dump()) |
| 206 | dict = { |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 207 | 'Version': self._version, |
| 208 | 'Maintainer': self._maintainer, |
| 209 | 'Description': self._description, |
| 210 | 'Packages': packages |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 211 | } |
| 212 | plist = plistlib.Plist(**dict) |
| 213 | plist.write(pathOrFile) |
| 214 | |
| 215 | def find(self, ident): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 216 | """Find a package. The package can be specified by name |
| 217 | or as a dictionary with name, version and flavor entries. |
| 218 | |
| 219 | Only name is obligatory. If there are multiple matches the |
| 220 | best one (higher version number, flavors ordered according to |
| 221 | users' preference) is returned.""" |
| 222 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 223 | if type(ident) == str: |
| 224 | # Remove ( and ) for pseudo-packages |
| 225 | if ident[0] == '(' and ident[-1] == ')': |
| 226 | ident = ident[1:-1] |
| 227 | # Split into name-version-flavor |
| 228 | fields = ident.split('-') |
| 229 | if len(fields) < 1 or len(fields) > 3: |
| 230 | return None |
| 231 | name = fields[0] |
| 232 | if len(fields) > 1: |
| 233 | version = fields[1] |
| 234 | else: |
| 235 | version = None |
| 236 | if len(fields) > 2: |
| 237 | flavor = fields[2] |
| 238 | else: |
| 239 | flavor = None |
| 240 | else: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 241 | name = ident['Name'] |
| 242 | version = ident.get('Version') |
| 243 | flavor = ident.get('Flavor') |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 244 | found = None |
| 245 | for p in self._packages: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 246 | if name == p.name() and \ |
| 247 | (not version or version == p.version()) and \ |
| 248 | (not flavor or flavor == p.flavor()): |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 249 | if not found or found < p: |
| 250 | found = p |
| 251 | return found |
| 252 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 253 | ALLOWED_KEYS = [ |
| 254 | "Name", |
| 255 | "Version", |
| 256 | "Flavor", |
| 257 | "Description", |
| 258 | "Home-page", |
| 259 | "Download-URL", |
| 260 | "Install-test", |
| 261 | "Install-command", |
| 262 | "Pre-install-command", |
| 263 | "Post-install-command", |
| 264 | "Prerequisites", |
| 265 | "MD5Sum" |
| 266 | ] |
| 267 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 268 | class PimpPackage: |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 269 | """Class representing a single package.""" |
| 270 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 271 | def __init__(self, db, dict): |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 272 | self._db = db |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 273 | name = dict["Name"] |
| 274 | for k in dict.keys(): |
| 275 | if not k in ALLOWED_KEYS: |
| 276 | sys.stderr.write("Warning: %s: unknown key %s\n" % (name, k)) |
| 277 | self._dict = dict |
| 278 | |
| 279 | def __getitem__(self, key): |
| 280 | return self._dict[key] |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 281 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 282 | def name(self): return self._dict['Name'] |
| 283 | def version(self): return self._dict['Version'] |
| 284 | def flavor(self): return self._dict['Flavor'] |
| 285 | def description(self): return self._dict['Description'] |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 286 | def homepage(self): return self._dict.get('Home-page') |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 287 | def downloadURL(self): return self._dict['Download-URL'] |
| 288 | |
| 289 | def fullname(self): |
| 290 | """Return the full name "name-version-flavor" of a package. |
| 291 | |
| 292 | If the package is a pseudo-package, something that cannot be |
| 293 | installed through pimp, return the name in (parentheses).""" |
| 294 | |
| 295 | rv = self._dict['Name'] |
| 296 | if self._dict.has_key('Version'): |
| 297 | rv = rv + '-%s' % self._dict['Version'] |
| 298 | if self._dict.has_key('Flavor'): |
| 299 | rv = rv + '-%s' % self._dict['Flavor'] |
| 300 | if not self._dict.get('Download-URL'): |
| 301 | # Pseudo-package, show in parentheses |
| 302 | rv = '(%s)' % rv |
| 303 | return rv |
| 304 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 305 | def dump(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 306 | """Return a dict object containing the information on the package.""" |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 307 | return self._dict |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 308 | |
| 309 | def __cmp__(self, other): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 310 | """Compare two packages, where the "better" package sorts lower.""" |
| 311 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 312 | if not isinstance(other, PimpPackage): |
| 313 | return cmp(id(self), id(other)) |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 314 | if self.name() != other.name(): |
| 315 | return cmp(self.name(), other.name()) |
| 316 | if self.version() != other.version(): |
| 317 | return -cmp(self.version(), other.version()) |
| 318 | return self._db.preferences.compareFlavors(self.flavor(), other.flavor()) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 319 | |
| 320 | def installed(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 321 | """Test wheter the package is installed. |
| 322 | |
| 323 | Returns two values: a status indicator which is one of |
| 324 | "yes", "no", "old" (an older version is installed) or "bad" |
| 325 | (something went wrong during the install test) and a human |
| 326 | readable string which may contain more details.""" |
| 327 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 328 | namespace = { |
| 329 | "NotInstalled": _scriptExc_NotInstalled, |
| 330 | "OldInstalled": _scriptExc_OldInstalled, |
| 331 | "BadInstalled": _scriptExc_BadInstalled, |
| 332 | "os": os, |
| 333 | "sys": sys, |
| 334 | } |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 335 | installTest = self._dict['Install-test'].strip() + '\n' |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 336 | try: |
| 337 | exec installTest in namespace |
| 338 | except ImportError, arg: |
| 339 | return "no", str(arg) |
| 340 | except _scriptExc_NotInstalled, arg: |
| 341 | return "no", str(arg) |
| 342 | except _scriptExc_OldInstalled, arg: |
| 343 | return "old", str(arg) |
| 344 | except _scriptExc_BadInstalled, arg: |
| 345 | return "bad", str(arg) |
| 346 | except: |
Jack Jansen | a460f44 | 2003-02-17 12:21:05 +0000 | [diff] [blame] | 347 | sys.stderr.write("-------------------------------------\n") |
| 348 | sys.stderr.write("---- %s: install test got exception\n" % self.fullname()) |
| 349 | sys.stderr.write("---- source:\n") |
| 350 | sys.stderr.write(installTest) |
| 351 | sys.stderr.write("---- exception:\n") |
| 352 | import traceback |
| 353 | traceback.print_exc(file=sys.stderr) |
| 354 | if self._db._maintainer: |
| 355 | sys.stderr.write("---- Please copy this and mail to %s\n" % self._db._maintainer) |
| 356 | sys.stderr.write("-------------------------------------\n") |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 357 | return "bad", "Package install test got exception" |
| 358 | return "yes", "" |
| 359 | |
| 360 | def prerequisites(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 361 | """Return a list of prerequisites for this package. |
| 362 | |
| 363 | The list contains 2-tuples, of which the first item is either |
| 364 | a PimpPackage object or None, and the second is a descriptive |
| 365 | string. The first item can be None if this package depends on |
| 366 | something that isn't pimp-installable, in which case the descriptive |
| 367 | string should tell the user what to do.""" |
| 368 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 369 | rv = [] |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 370 | if not self._dict.get('Download-URL'): |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 371 | return [(None, |
| 372 | "%s: This package needs to be installed manually (no Download-URL field)" % |
| 373 | self.fullname())] |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 374 | if not self._dict.get('Prerequisites'): |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 375 | return [] |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 376 | for item in self._dict['Prerequisites']: |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 377 | if type(item) == str: |
| 378 | pkg = None |
| 379 | descr = str(item) |
| 380 | else: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 381 | name = item['Name'] |
| 382 | if item.has_key('Version'): |
| 383 | name = name + '-' + item['Version'] |
| 384 | if item.has_key('Flavor'): |
| 385 | name = name + '-' + item['Flavor'] |
| 386 | pkg = self._db.find(name) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 387 | if not pkg: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 388 | descr = "Requires unknown %s"%name |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 389 | else: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 390 | descr = pkg.description() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 391 | rv.append((pkg, descr)) |
| 392 | return rv |
| 393 | |
| 394 | def _cmd(self, output, dir, *cmditems): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 395 | """Internal routine to run a shell command in a given directory.""" |
| 396 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 397 | cmd = ("cd \"%s\"; " % dir) + " ".join(cmditems) |
| 398 | if output: |
| 399 | output.write("+ %s\n" % cmd) |
| 400 | if NO_EXECUTE: |
| 401 | return 0 |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 402 | dummy, fp = os.popen4(cmd, "r") |
| 403 | dummy.close() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 404 | while 1: |
| 405 | line = fp.readline() |
| 406 | if not line: |
| 407 | break |
| 408 | if output: |
| 409 | output.write(line) |
| 410 | rv = fp.close() |
| 411 | return rv |
| 412 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 413 | def downloadPackageOnly(self, output=None): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 414 | """Download a single package, if needed. |
| 415 | |
| 416 | An MD5 signature is used to determine whether download is needed, |
| 417 | and to test that we actually downloaded what we expected. |
| 418 | If output is given it is a file-like object that will receive a log |
| 419 | of what happens. |
| 420 | |
| 421 | If anything unforeseen happened the method returns an error message |
| 422 | string. |
| 423 | """ |
| 424 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 425 | scheme, loc, path, query, frag = urlparse.urlsplit(self._dict['Download-URL']) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 426 | path = urllib.url2pathname(path) |
| 427 | filename = os.path.split(path)[1] |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 428 | self.archiveFilename = os.path.join(self._db.preferences.downloadDir, filename) |
| 429 | if not self._archiveOK(): |
Jack Jansen | 26bf3ac | 2003-02-10 14:19:14 +0000 | [diff] [blame] | 430 | if scheme == 'manual': |
| 431 | return "Please download package manually and save as %s" % self.archiveFilename |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 432 | if self._cmd(output, self._db.preferences.downloadDir, |
| 433 | "curl", |
| 434 | "--output", self.archiveFilename, |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 435 | self._dict['Download-URL']): |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 436 | return "download command failed" |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 437 | if not os.path.exists(self.archiveFilename) and not NO_EXECUTE: |
| 438 | return "archive not found after download" |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 439 | if not self._archiveOK(): |
| 440 | return "archive does not have correct MD5 checksum" |
| 441 | |
| 442 | def _archiveOK(self): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 443 | """Test an archive. It should exist and the MD5 checksum should be correct.""" |
| 444 | |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 445 | if not os.path.exists(self.archiveFilename): |
| 446 | return 0 |
Jack Jansen | e71b9f8 | 2003-02-12 16:37:00 +0000 | [diff] [blame] | 447 | if not self._dict.get('MD5Sum'): |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 448 | sys.stderr.write("Warning: no MD5Sum for %s\n" % self.fullname()) |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 449 | return 1 |
| 450 | data = open(self.archiveFilename, 'rb').read() |
| 451 | checksum = md5.new(data).hexdigest() |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 452 | return checksum == self._dict['MD5Sum'] |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 453 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 454 | def unpackPackageOnly(self, output=None): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 455 | """Unpack a downloaded package archive.""" |
| 456 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 457 | filename = os.path.split(self.archiveFilename)[1] |
| 458 | for ext, cmd in ARCHIVE_FORMATS: |
| 459 | if filename[-len(ext):] == ext: |
| 460 | break |
| 461 | else: |
| 462 | return "unknown extension for archive file: %s" % filename |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 463 | self.basename = filename[:-len(ext)] |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 464 | cmd = cmd % self.archiveFilename |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 465 | if self._cmd(output, self._db.preferences.buildDir, cmd): |
| 466 | return "unpack command failed" |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 467 | |
| 468 | def installPackageOnly(self, output=None): |
| 469 | """Default install method, to be overridden by subclasses""" |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 470 | return "%s: This package needs to be installed manually (no support for flavor=\"%s\")" \ |
| 471 | % (self.fullname(), self._dict.get(flavor, "")) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 472 | |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 473 | def installSinglePackage(self, output=None): |
| 474 | """Download, unpack and install a single package. |
| 475 | |
| 476 | If output is given it should be a file-like object and it |
| 477 | will receive a log of what happened.""" |
| 478 | |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 479 | if not self._dict['Download-URL']: |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 480 | return "%s: This package needs to be installed manually (no Download-URL field)" % _fmtpackagename(self) |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 481 | msg = self.downloadPackageOnly(output) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 482 | if msg: |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 483 | return "%s: download: %s" % (self.fullname(), msg) |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 484 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 485 | msg = self.unpackPackageOnly(output) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 486 | if msg: |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 487 | return "%s: unpack: %s" % (self.fullname(), msg) |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 488 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 489 | return self.installPackageOnly(output) |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 490 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 491 | def beforeInstall(self): |
| 492 | """Bookkeeping before installation: remember what we have in site-packages""" |
| 493 | self._old_contents = os.listdir(self._db.preferences.installDir) |
| 494 | |
| 495 | def afterInstall(self): |
| 496 | """Bookkeeping after installation: interpret any new .pth files that have |
| 497 | appeared""" |
| 498 | |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 499 | new_contents = os.listdir(self._db.preferences.installDir) |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 500 | for fn in new_contents: |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 501 | if fn in self._old_contents: |
Jack Jansen | 53b341f | 2003-02-12 15:36:25 +0000 | [diff] [blame] | 502 | continue |
| 503 | if fn[-4:] != '.pth': |
| 504 | continue |
| 505 | fullname = os.path.join(self._db.preferences.installDir, fn) |
| 506 | f = open(fullname) |
| 507 | for line in f.readlines(): |
| 508 | if not line: |
| 509 | continue |
| 510 | if line[0] == '#': |
| 511 | continue |
| 512 | if line[:6] == 'import': |
| 513 | exec line |
| 514 | continue |
| 515 | if line[-1] == '\n': |
| 516 | line = line[:-1] |
| 517 | if not os.path.isabs(line): |
| 518 | line = os.path.join(self._db.preferences.installDir, line) |
| 519 | line = os.path.realpath(line) |
| 520 | if not line in sys.path: |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 521 | sys.path.append(line) |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 522 | |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 523 | class PimpPackage_binary(PimpPackage): |
| 524 | |
| 525 | def unpackPackageOnly(self, output=None): |
| 526 | """We don't unpack binary packages until installing""" |
| 527 | pass |
| 528 | |
| 529 | def installPackageOnly(self, output=None): |
| 530 | """Install a single source package. |
| 531 | |
| 532 | If output is given it should be a file-like object and it |
| 533 | will receive a log of what happened.""" |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 534 | print 'PimpPackage_binary installPackageOnly' |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 535 | |
| 536 | msgs = [] |
| 537 | if self._dict.has_key('Pre-install-command'): |
| 538 | msg.append("%s: Pre-install-command ignored" % self.fullname()) |
| 539 | if self._dict.has_key('Install-command'): |
| 540 | msgs.append("%s: Install-command ignored" % self.fullname()) |
| 541 | if self._dict.has_key('Post-install-command'): |
| 542 | msgs.append("%s: Post-install-command ignored" % self.fullname()) |
| 543 | |
| 544 | self.beforeInstall() |
| 545 | |
| 546 | # Install by unpacking |
| 547 | filename = os.path.split(self.archiveFilename)[1] |
| 548 | for ext, cmd in ARCHIVE_FORMATS: |
| 549 | if filename[-len(ext):] == ext: |
| 550 | break |
| 551 | else: |
| 552 | return "unknown extension for archive file: %s" % filename |
| 553 | |
Jack Jansen | a460f44 | 2003-02-17 12:21:05 +0000 | [diff] [blame] | 554 | # Extract the files in the root folder. |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 555 | cmd = cmd % self.archiveFilename |
Jack Jansen | a460f44 | 2003-02-17 12:21:05 +0000 | [diff] [blame] | 556 | if self._cmd(output, "/", cmd): |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 557 | return "unpack command failed" |
| 558 | |
| 559 | self.afterInstall() |
| 560 | |
| 561 | if self._dict.has_key('Post-install-command'): |
| 562 | if self._cmd(output, self._buildDirname, self._dict['Post-install-command']): |
| 563 | return "post-install %s: running \"%s\" failed" % \ |
| 564 | (self.fullname(), self._dict['Post-install-command']) |
| 565 | return None |
| 566 | |
| 567 | |
| 568 | class PimpPackage_source(PimpPackage): |
| 569 | |
| 570 | def unpackPackageOnly(self, output=None): |
| 571 | """Unpack a source package and check that setup.py exists""" |
| 572 | PimpPackage.unpackPackageOnly(self, output) |
| 573 | # Test that a setup script has been create |
Jack Jansen | a2125fe | 2003-02-16 23:03:04 +0000 | [diff] [blame] | 574 | self._buildDirname = os.path.join(self._db.preferences.buildDir, self.basename) |
Jack Jansen | 0dacac4 | 2003-02-14 14:11:59 +0000 | [diff] [blame] | 575 | setupname = os.path.join(self._buildDirname, "setup.py") |
| 576 | if not os.path.exists(setupname) and not NO_EXECUTE: |
| 577 | return "no setup.py found after unpack of archive" |
| 578 | |
| 579 | def installPackageOnly(self, output=None): |
| 580 | """Install a single source package. |
| 581 | |
| 582 | If output is given it should be a file-like object and it |
| 583 | will receive a log of what happened.""" |
| 584 | |
| 585 | if self._dict.has_key('Pre-install-command'): |
| 586 | if self._cmd(output, self._buildDirname, self._dict['Pre-install-command']): |
| 587 | return "pre-install %s: running \"%s\" failed" % \ |
| 588 | (self.fullname(), self._dict['Pre-install-command']) |
| 589 | |
| 590 | self.beforeInstall() |
| 591 | installcmd = self._dict.get('Install-command') |
| 592 | if not installcmd: |
| 593 | installcmd = '"%s" setup.py install' % sys.executable |
| 594 | if self._cmd(output, self._buildDirname, installcmd): |
| 595 | return "install %s: running \"%s\" failed" % self.fullname() |
| 596 | |
| 597 | self.afterInstall() |
| 598 | |
| 599 | if self._dict.has_key('Post-install-command'): |
| 600 | if self._cmd(output, self._buildDirname, self._dict['Post-install-command']): |
| 601 | return "post-install %s: running \"%s\" failed" % \ |
| 602 | (self.fullname(), self._dict['Post-install-command']) |
| 603 | return None |
| 604 | |
| 605 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 606 | class PimpInstaller: |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 607 | """Installer engine: computes dependencies and installs |
| 608 | packages in the right order.""" |
| 609 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 610 | def __init__(self, db): |
| 611 | self._todo = [] |
| 612 | self._db = db |
| 613 | self._curtodo = [] |
| 614 | self._curmessages = [] |
| 615 | |
| 616 | def __contains__(self, package): |
| 617 | return package in self._todo |
| 618 | |
| 619 | def _addPackages(self, packages): |
| 620 | for package in packages: |
| 621 | if not package in self._todo: |
| 622 | self._todo.insert(0, package) |
| 623 | |
| 624 | def _prepareInstall(self, package, force=0, recursive=1): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 625 | """Internal routine, recursive engine for prepareInstall. |
| 626 | |
| 627 | Test whether the package is installed and (if not installed |
| 628 | or if force==1) prepend it to the temporary todo list and |
| 629 | call ourselves recursively on all prerequisites.""" |
| 630 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 631 | if not force: |
| 632 | status, message = package.installed() |
| 633 | if status == "yes": |
| 634 | return |
| 635 | if package in self._todo or package in self._curtodo: |
| 636 | return |
| 637 | self._curtodo.insert(0, package) |
| 638 | if not recursive: |
| 639 | return |
| 640 | prereqs = package.prerequisites() |
| 641 | for pkg, descr in prereqs: |
| 642 | if pkg: |
| 643 | self._prepareInstall(pkg, force, recursive) |
| 644 | else: |
| 645 | self._curmessages.append("Requires: %s" % descr) |
| 646 | |
| 647 | def prepareInstall(self, package, force=0, recursive=1): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 648 | """Prepare installation of a package. |
| 649 | |
| 650 | If the package is already installed and force is false nothing |
| 651 | is done. If recursive is true prerequisites are installed first. |
| 652 | |
| 653 | Returns a list of packages (to be passed to install) and a list |
| 654 | of messages of any problems encountered. |
| 655 | """ |
| 656 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 657 | self._curtodo = [] |
| 658 | self._curmessages = [] |
| 659 | self._prepareInstall(package, force, recursive) |
| 660 | rv = self._curtodo, self._curmessages |
| 661 | self._curtodo = [] |
| 662 | self._curmessages = [] |
| 663 | return rv |
| 664 | |
| 665 | def install(self, packages, output): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 666 | """Install a list of packages.""" |
| 667 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 668 | self._addPackages(packages) |
| 669 | status = [] |
| 670 | for pkg in self._todo: |
| 671 | msg = pkg.installSinglePackage(output) |
| 672 | if msg: |
| 673 | status.append(msg) |
| 674 | return status |
| 675 | |
| 676 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 677 | |
| 678 | def _run(mode, verbose, force, args): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 679 | """Engine for the main program""" |
| 680 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 681 | prefs = PimpPreferences() |
| 682 | prefs.check() |
| 683 | db = PimpDatabase(prefs) |
| 684 | db.appendURL(prefs.pimpDatabase) |
| 685 | |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 686 | if mode == 'dump': |
| 687 | db.dump(sys.stdout) |
| 688 | elif mode =='list': |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 689 | if not args: |
| 690 | args = db.listnames() |
| 691 | print "%-20.20s\t%s" % ("Package", "Description") |
| 692 | print |
| 693 | for pkgname in args: |
| 694 | pkg = db.find(pkgname) |
| 695 | if pkg: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 696 | description = pkg.description() |
| 697 | pkgname = pkg.fullname() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 698 | else: |
| 699 | description = 'Error: no such package' |
| 700 | print "%-20.20s\t%s" % (pkgname, description) |
| 701 | if verbose: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 702 | print "\tHome page:\t", pkg.homepage() |
| 703 | print "\tDownload URL:\t", pkg.downloadURL() |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 704 | elif mode =='status': |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 705 | if not args: |
| 706 | args = db.listnames() |
| 707 | print "%-20.20s\t%s\t%s" % ("Package", "Installed", "Message") |
| 708 | print |
| 709 | for pkgname in args: |
| 710 | pkg = db.find(pkgname) |
| 711 | if pkg: |
| 712 | status, msg = pkg.installed() |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 713 | pkgname = pkg.fullname() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 714 | else: |
| 715 | status = 'error' |
| 716 | msg = 'No such package' |
| 717 | print "%-20.20s\t%-9.9s\t%s" % (pkgname, status, msg) |
| 718 | if verbose and status == "no": |
| 719 | prereq = pkg.prerequisites() |
| 720 | for pkg, msg in prereq: |
| 721 | if not pkg: |
| 722 | pkg = '' |
| 723 | else: |
Jack Jansen | e7b33db | 2003-02-11 22:40:59 +0000 | [diff] [blame] | 724 | pkg = pkg.fullname() |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 725 | print "%-20.20s\tRequirement: %s %s" % ("", pkg, msg) |
| 726 | elif mode == 'install': |
| 727 | if not args: |
| 728 | print 'Please specify packages to install' |
| 729 | sys.exit(1) |
| 730 | inst = PimpInstaller(db) |
| 731 | for pkgname in args: |
| 732 | pkg = db.find(pkgname) |
| 733 | if not pkg: |
| 734 | print '%s: No such package' % pkgname |
| 735 | continue |
| 736 | list, messages = inst.prepareInstall(pkg, force) |
| 737 | if messages and not force: |
| 738 | print "%s: Not installed:" % pkgname |
| 739 | for m in messages: |
| 740 | print "\t", m |
| 741 | else: |
| 742 | if verbose: |
| 743 | output = sys.stdout |
| 744 | else: |
| 745 | output = None |
| 746 | messages = inst.install(list, output) |
| 747 | if messages: |
| 748 | print "%s: Not installed:" % pkgname |
| 749 | for m in messages: |
| 750 | print "\t", m |
| 751 | |
| 752 | def main(): |
Jack Jansen | 6a600ab | 2003-02-10 15:55:51 +0000 | [diff] [blame] | 753 | """Minimal commandline tool to drive pimp.""" |
| 754 | |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 755 | import getopt |
| 756 | def _help(): |
| 757 | print "Usage: pimp [-v] -s [package ...] List installed status" |
| 758 | print " pimp [-v] -l [package ...] Show package information" |
| 759 | print " pimp [-vf] -i package ... Install packages" |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 760 | print " pimp -d Dump database to stdout" |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 761 | print "Options:" |
| 762 | print " -v Verbose" |
| 763 | print " -f Force installation" |
| 764 | sys.exit(1) |
| 765 | |
| 766 | try: |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 767 | opts, args = getopt.getopt(sys.argv[1:], "slifvd") |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 768 | except getopt.Error: |
| 769 | _help() |
| 770 | if not opts and not args: |
| 771 | _help() |
| 772 | mode = None |
| 773 | force = 0 |
| 774 | verbose = 0 |
| 775 | for o, a in opts: |
| 776 | if o == '-s': |
| 777 | if mode: |
| 778 | _help() |
| 779 | mode = 'status' |
| 780 | if o == '-l': |
| 781 | if mode: |
| 782 | _help() |
| 783 | mode = 'list' |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 784 | if o == '-d': |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 785 | if mode: |
| 786 | _help() |
Jack Jansen | c4b217d | 2003-02-10 13:38:44 +0000 | [diff] [blame] | 787 | mode = 'dump' |
Jack Jansen | 95839b8 | 2003-02-09 23:10:20 +0000 | [diff] [blame] | 788 | if o == '-i': |
| 789 | mode = 'install' |
| 790 | if o == '-f': |
| 791 | force = 1 |
| 792 | if o == '-v': |
| 793 | verbose = 1 |
| 794 | if not mode: |
| 795 | _help() |
| 796 | _run(mode, verbose, force, args) |
| 797 | |
| 798 | if __name__ == '__main__': |
| 799 | main() |
| 800 | |
| 801 | |