Mac/Tools/IDE/PackageManager.py

   1 # Prelude to allow running this as a main program
   2 def _init():
   3         import macresource
   4         import sys, os
   5         macresource.need('DITL', 468, "PythonIDE.rsrc")
   6         widgetrespathsegs = [sys.exec_prefix, "Mac", "Tools", "IDE", "Widgets.rsrc"]
   7         widgetresfile = os.path.join(*widgetrespathsegs)
   8         if not os.path.exists(widgetresfile):
   9                 widgetrespathsegs = [os.pardir, "Tools", "IDE", "Widgets.rsrc"]
  10                 widgetresfile = os.path.join(*widgetrespathsegs)
  11         refno = macresource.need('CURS', 468, widgetresfile)
  12         if os.environ.has_key('PYTHONIDEPATH'):
  13                 # For development set this environment variable
  14                 ide_path = os.environ['PYTHONIDEPATH']
  15         elif refno:
  16                 # We're not a fullblown application
  17                 idepathsegs = [sys.exec_prefix, "Mac", "Tools", "IDE"]
  18                 ide_path = os.path.join(*idepathsegs)
  19                 if not os.path.exists(ide_path):
  20                         idepathsegs = [os.pardir, "Tools", "IDE"]
  21                         for p in sys.path:
  22                                 ide_path = os.path.join(*([p]+idepathsegs))
  23                                 if os.path.exists(ide_path):
  24                                         break
  25
  26         else:
  27                 # We are a fully frozen application
  28                 ide_path = sys.argv[0]
  29         if ide_path not in sys.path:
  30                 sys.path.insert(0, ide_path)
  31
  32 if __name__ == '__main__':
  33         _init()
  34
  35 import W
  36 import Wapplication
  37 from Carbon import Evt
  38 import EasyDialogs
  39 import FrameWork
  40
  41 import sys
  42 import string
  43 import os
  44 import urllib
  45
  46 import pimp
  47
  48 PACKMAN_HOMEPAGE="http://www.python.org/packman"
  49
  50 ELIPSES = '...'
  51
  52 USER_INSTALL_DIR = os.path.join(os.environ.get('HOME', ''),
  53                                                                 'Library',
  54                                                                 'Python',
  55                                                                 sys.version[:3],
  56                                                                 'site-packages')
  57
  58 class PackageManagerMain(Wapplication.Application):
  59
  60         def __init__(self):
  61                 self.preffilepath = os.path.join("Python", "Package Install Manager Prefs")
  62                 Wapplication.Application.__init__(self, 'Pimp')
  63                 from Carbon import AE
  64                 from Carbon import AppleEvents
  65                 self.defaulturl = ""
  66
  67                 AE.AEInstallEventHandler(AppleEvents.kCoreEventClass, AppleEvents.kAEOpenApplication,
  68                                 self.ignoreevent)
  69                 AE.AEInstallEventHandler(AppleEvents.kCoreEventClass, AppleEvents.kAEReopenApplication,
  70                                 self.ignoreevent)
  71                 AE.AEInstallEventHandler(AppleEvents.kCoreEventClass, AppleEvents.kAEPrintDocuments,
  72                                 self.ignoreevent)
  73                 AE.AEInstallEventHandler(AppleEvents.kCoreEventClass, AppleEvents.kAEQuitApplication,
  74                                 self.quitevent)
  75                 if 1:
  76                         import PyConsole
  77                         # With -D option (OSX command line only) keep stderr, for debugging the IDE
  78                         # itself.
  79                         debug_stderr = None
  80                         if len(sys.argv) >= 2 and sys.argv[1] == '-D':
  81                                 debug_stderr = sys.stderr
  82                                 del sys.argv[1]
  83                         PyConsole.installoutput()
  84                         if debug_stderr:
  85                                 sys.stderr = debug_stderr
  86                 self.domenu_openstandard()
  87                 self.mainloop()
  88
  89         def makeusermenus(self):
  90                 m = Wapplication.Menu(self.menubar, "File")
  91                 newitem = FrameWork.MenuItem(m, "Open Standard Database", "N", 'openstandard')
  92                 newexpitem = FrameWork.MenuItem(m, "Open Experimental Database", None, 'openexperimental')
  93                 newexpitem.enable(pimp.PIMP_VERSION >= "0.4")
  94                 openitem = FrameWork.MenuItem(m, "Open"+ELIPSES, "O", 'open')
  95                 openURLitem = FrameWork.MenuItem(m, "Open URL"+ELIPSES, "D", 'openURL')
  96                 FrameWork.Separator(m)
  97                 moreinfoitem = FrameWork.MenuItem(m, "More Databases", None, 'opendatabasepage')
  98                 FrameWork.Separator(m)
  99                 closeitem = FrameWork.MenuItem(m, "Close", "W", 'close')
 100 ##              saveitem = FrameWork.MenuItem(m, "Save", "S", 'save')
 101 ##              saveasitem = FrameWork.MenuItem(m, "Save as"+ELIPSES, None, 'save_as')
 102 ##              FrameWork.Separator(m)
 103
 104                 m = Wapplication.Menu(self.menubar, "Edit")
 105                 undoitem = FrameWork.MenuItem(m, "Undo", 'Z', "undo")
 106                 FrameWork.Separator(m)
 107                 cutitem = FrameWork.MenuItem(m, "Cut", 'X', "cut")
 108                 copyitem = FrameWork.MenuItem(m, "Copy", "C", "copy")
 109                 pasteitem = FrameWork.MenuItem(m, "Paste", "V", "paste")
 110                 FrameWork.MenuItem(m, "Clear", None,  "clear")
 111                 FrameWork.Separator(m)
 112                 selallitem = FrameWork.MenuItem(m, "Select all", "A", "selectall")
 113
 114                 m = Wapplication.Menu(self.menubar, "Package")
 115                 runitem = FrameWork.MenuItem(m, "Install", "I", 'install')
 116                 homepageitem = FrameWork.MenuItem(m, "Visit Homepage", None, 'homepage')
 117
 118                 self.openwindowsmenu = Wapplication.Menu(self.menubar, 'Windows')
 119                 self.makeopenwindowsmenu()
 120                 self.makehelpmenu()
 121                 self._menustocheck = [closeitem,
 122                                 undoitem, cutitem, copyitem, pasteitem,
 123                                 selallitem,
 124                                 runitem, homepageitem]
 125
 126         def makehelpmenu(self):
 127                 python_app = os.path.join(sys.prefix, 'Resources/Python.app')
 128                 help_source = os.path.join(python_app, 'Contents/Resources/English.lproj/Documentation')
 129                 hashelp = os.path.isdir(help_source)
 130
 131                 self.helpmenu = m = self.gethelpmenu()
 132                 helpitem1 = FrameWork.MenuItem(m, "PackageManager Help", None, self.domenu_packmanhelp)
 133                 helpitem1.enable(hashelp)
 134                 helpitem2 = FrameWork.MenuItem(m, "MacPython Help", None, self.domenu_pythonhelp)
 135                 helpitem2.enable(hashelp)
 136
 137         def quitevent(self, theAppleEvent, theReply):
 138                 self._quit()
 139
 140         def ignoreevent(self, theAppleEvent, theReply):
 141                 pass
 142
 143         def opendocsevent(self, theAppleEvent, theReply):
 144                 W.SetCursor('watch')
 145                 import aetools
 146                 parameters, args = aetools.unpackevent(theAppleEvent)
 147                 docs = parameters['----']
 148                 if type(docs) <> type([]):
 149                         docs = [docs]
 150                 for doc in docs:
 151                         fsr, a = doc.FSResolveAlias(None)
 152                         path = fsr.as_pathname()
 153                         path = urllib.pathname2url(path)
 154                         self.opendoc(path)
 155
 156         def opendoc(self, url):
 157                 if url:
 158                         self.defaulturl = url
 159                 PackageBrowser(url)
 160
 161         def getabouttext(self):
 162                 return "About Package Manager"+ELIPSES
 163
 164         def do_about(self, id, item, window, event):
 165                 EasyDialogs.Message("Package Install Manager for Python\nPackMan engine (pimp) version: %s" %
 166                         pimp.PIMP_VERSION)
 167
 168         def domenu_openstandard(self, *args):
 169                 if pimp.PIMP_VERSION >= "0.4":
 170                         url = pimp.getDefaultDatabase()
 171                 else:
 172                         # 0.3 compatibility
 173                         url = None
 174                 self.opendoc(url)
 175
 176         def domenu_openexperimental(self, *args):
 177                 database = pimp.getDefaultDatabase(experimental=True)
 178                 self.opendoc(database)
 179
 180         def domenu_open(self, *args):
 181                 filename = EasyDialogs.AskFileForOpen(typeList=("TEXT",))
 182                 if filename:
 183                         filename = urllib.pathname2url(filename)
 184                         self.opendoc(filename)
 185
 186         def domenu_openURL(self, *args):
 187                 ok = EasyDialogs.AskYesNoCancel(
 188                         "Warning: by opening a non-standard database "
 189                         "you are trusting the maintainer of it "
 190                         "to run arbitrary code on your machine.",
 191                         yes="OK", no="")
 192                 if ok <= 0: return
 193                 url = EasyDialogs.AskString("URL of database to open:",
 194                         default=self.defaulturl, ok="Open")
 195                 if url:
 196                         self.opendoc(url)
 197
 198         def domenu_opendatabasepage(self):
 199                 import ic
 200
 201                 icr = ic.IC()
 202                 icr.launchurl(PACKMAN_HOMEPAGE)
 203         def makeopenwindowsmenu(self):
 204                 for i in range(len(self.openwindowsmenu.items)):
 205                         self.openwindowsmenu.menu.DeleteMenuItem(1)
 206                         self.openwindowsmenu.items = []
 207                 windows = []
 208                 self._openwindows = {}
 209                 for window in self._windows.keys():
 210                         title = window.GetWTitle()
 211                         if not title:
 212                                 title = "<no title>"
 213                         windows.append((title, window))
 214                 windows.sort()
 215                 for title, window in windows:
 216                         shortcut = None
 217                         item = FrameWork.MenuItem(self.openwindowsmenu, title, shortcut, callback = self.domenu_openwindows)
 218                         self._openwindows[item.item] = window
 219                 self._openwindowscheckmark = 0
 220                 self.checkopenwindowsmenu()
 221
 222         def domenu_openwindows(self, id, item, window, event):
 223                 w = self._openwindows[item]
 224                 w.ShowWindow()
 225                 w.SelectWindow()
 226
 227         def domenu_quit(self):
 228                 self._quit()
 229
 230         def domenu_save(self, *args):
 231                 print "Save"
 232
 233         def domenu_pythonhelp(self, *args):
 234                 from Carbon import AH
 235                 AH.AHGotoPage("MacPython Help", None, None)
 236
 237         def domenu_packmanhelp(self, *args):
 238                 from Carbon import AH
 239                 AH.AHGotoPage("MacPython Help", "packman.html", None)
 240
 241         def _quit(self):
 242 ##              import PyConsole, PyEdit
 243                 for window in self._windows.values():
 244                         try:
 245                                 rv = window.close() # ignore any errors while quitting
 246                         except:
 247                                 rv = 0   # (otherwise, we can get stuck!)
 248                         if rv and rv > 0:
 249                                 return
 250 ##              try:
 251 ##                      PyConsole.console.writeprefs()
 252 ##                      PyConsole.output.writeprefs()
 253 ##                      PyEdit.searchengine.writeprefs()
 254 ##              except:
 255 ##                      # Write to __stderr__ so the msg end up in Console.app and has
 256 ##                      # at least _some_ chance of getting read...
 257 ##                      # But: this is a workaround for way more serious problems with
 258 ##                      # the Python 2.2 Jaguar addon.
 259 ##                      sys.__stderr__.write("*** PythonIDE: Can't write preferences ***\n")
 260                 self.quitting = 1
 261
 262 class PimpInterface:
 263
 264         def setuppimp(self, url):
 265                 self.pimpprefs = pimp.PimpPreferences()
 266                 self.pimpdb = pimp.PimpDatabase(self.pimpprefs)
 267                 if not url:
 268                         url = self.pimpprefs.pimpDatabase
 269                 try:
 270                         self.pimpdb.appendURL(url)
 271                 except IOError, arg:
 272                         rv = "Cannot open %s: %s\n" % (url, arg)
 273                         rv += "\nSee MacPython Package Manager help page."
 274                         return rv
 275                 except:
 276                         rv = "Unspecified error while parsing database: %s\n" % url
 277                         rv += "Usually, this means the database is not correctly formatted.\n"
 278                         rv += "\nSee MacPython Package Manager help page."
 279                         return rv
 280                 # Check whether we can write the installation directory.
 281                 # If not, set to the per-user directory, possibly
 282                 # creating it, if needed.
 283                 installDir = self.pimpprefs.installDir
 284                 if not os.access(installDir, os.R_OK|os.W_OK|os.X_OK):
 285                         rv = self.setuserinstall(1)
 286                         if rv: return rv
 287                 return self.pimpprefs.check()
 288
 289         def closepimp(self):
 290                 self.pimpdb.close()
 291                 self.pimpprefs = None
 292                 self.pimpdb = None
 293                 self.packages = []
 294
 295         def setuserinstall(self, onoff):
 296                 rv = ""
 297                 if onoff:
 298                         if not os.path.exists(USER_INSTALL_DIR):
 299                                 try:
 300                                         os.makedirs(USER_INSTALL_DIR)
 301                                 except OSError, arg:
 302                                         rv = rv + arg + "\n"
 303                         if not USER_INSTALL_DIR in sys.path:
 304                                 import site
 305                                 reload(site)
 306                         self.pimpprefs.setInstallDir(USER_INSTALL_DIR)
 307                 else:
 308                         self.pimpprefs.setInstallDir(None)
 309                 rv = rv + self.pimpprefs.check()
 310                 return rv
 311
 312         def getuserinstall(self):
 313                 return self.pimpprefs.installDir == USER_INSTALL_DIR
 314
 315         def getbrowserdata(self, show_hidden=1):
 316                 packages = self.pimpdb.list()
 317                 if show_hidden:
 318                         self.packages = packages
 319                 else:
 320                         self.packages = []
 321                         for pkg in packages:
 322                                 name = pkg.fullname()
 323                                 if name[0] == '(' and name[-1] == ')' and not show_hidden:
 324                                         continue
 325                                 self.packages.append(pkg)
 326                 rv = []
 327                 for pkg in self.packages:
 328                         name = pkg.fullname()
 329                         status, _ = pkg.installed()
 330                         description = pkg.description()
 331                         description_line1 = description.split('\n')[0]
 332                         rv.append((status, name, description_line1))
 333                 return rv
 334
 335         def getstatus(self, number):
 336                 pkg = self.packages[number]
 337                 return pkg.installed()
 338
 339         def installpackage(self, sel, output, recursive, force):
 340                 pkg = self.packages[sel]
 341                 pimpinstaller = pimp.PimpInstaller(self.pimpdb)
 342                 list, messages = pimpinstaller.prepareInstall(pkg, force, recursive)
 343                 if messages:
 344                         return messages
 345                 messages = pimpinstaller.install(list, output)
 346                 return messages
 347
 348 class PackageBrowser(PimpInterface):
 349
 350         def __init__(self, url = None):
 351                 self.ic = None
 352                 messages = self.setuppimp(url)
 353                 self.setupwidgets()
 354                 self.updatestatus()
 355                 self.showmessages(messages)
 356
 357         def close(self):
 358                 self.closepimp()
 359
 360         def setupwidgets(self):
 361                 DESCRIPTION_HEIGHT = 140
 362                 INSTALL_POS = -30
 363                 STATUS_POS = INSTALL_POS - (70 + DESCRIPTION_HEIGHT)
 364                 self.w = W.Window((580, 600), "Python Install Manager", minsize = (400, 400), tabbable = 0)
 365                 self.w.titlebar = W.TextBox((4, 8, 60, 18), 'Packages:')
 366                 self.w.hidden_button = W.CheckBox((-100, 4, 0, 18), 'Show Hidden', self.updatestatus)
 367                 data = self.getbrowserdata()
 368                 self.w.packagebrowser = W.MultiList((4, 24, 0, STATUS_POS-2), data, self.listhit, cols=3)
 369
 370                 self.w.installed_l = W.TextBox((4, STATUS_POS, 70, 12), 'Installed:')
 371                 self.w.installed = W.TextBox((74, STATUS_POS, 0, 12), '')
 372                 self.w.message_l = W.TextBox((4, STATUS_POS+20, 70, 12), 'Status:')
 373                 self.w.message = W.TextBox((74, STATUS_POS+20, 0, 12), '')
 374                 self.w.homepage_button = W.Button((4, STATUS_POS+40, 96, 18), 'View homepage', self.do_homepage)
 375                 self.w.description_l = W.TextBox((4, STATUS_POS+70, 70, 12), 'Description:')
 376                 self.w.description = W.EditText((74, STATUS_POS+70, 0, DESCRIPTION_HEIGHT-4))
 377
 378                 self.w.divline = W.HorizontalLine((0, INSTALL_POS-4, 0, 0))
 379                 self.w.verbose_button = W.CheckBox((84, INSTALL_POS+4, 60, 18), 'Verbose')
 380                 self.w.recursive_button = W.CheckBox((146, INSTALL_POS+4, 120, 18), 'Install dependencies', self.updatestatus)
 381                 self.w.recursive_button.set(1)
 382                 self.w.force_button = W.CheckBox((268, INSTALL_POS+4, 70, 18), 'Overwrite', self.updatestatus)
 383                 self.w.user_button = W.CheckBox((340, INSTALL_POS+4, 140, 18), 'For Current User Only', self.do_user)
 384                 self.w.install_button = W.Button((4, INSTALL_POS+4, 56, 18), 'Install:', self.do_install)
 385                 self.w.open()
 386                 self.w.description.enable(0)
 387
 388         def updatestatus(self):
 389                 topcell = self.w.packagebrowser.gettopcell()
 390                 sel = self.w.packagebrowser.getselection()
 391                 data = self.getbrowserdata(self.w.hidden_button.get())
 392                 self.w.packagebrowser.setitems(data)
 393                 self.w.user_button.set(self.getuserinstall())
 394                 if len(sel) != 1:
 395                         self.w.installed.set('')
 396                         self.w.message.set('')
 397                         self.w.install_button.enable(0)
 398                         self.w.homepage_button.enable(0)
 399                         self.w.description.set('')
 400                         self.w.verbose_button.enable(0)
 401                         self.w.recursive_button.enable(0)
 402                         self.w.force_button.enable(0)
 403                         self.w.user_button.enable(0)
 404                 else:
 405                         sel = sel[0]
 406                         if sel >= len(self.packages):
 407                                 sel = 0
 408                         self.w.packagebrowser.setselection([sel])
 409                         installed, message = self.getstatus(sel)
 410                         self.w.installed.set(installed)
 411                         self.w.message.set(message)
 412                         self.w.install_button.enable(installed != "yes" or self.w.force_button.get())
 413                         self.w.homepage_button.enable(not not self.packages[sel].homepage())
 414                         description = self.packages[sel].description()
 415                         description = description.splitlines()
 416                         description = '\r'.join(description)
 417                         self.w.description.set(description)
 418                         self.w.verbose_button.enable(1)
 419                         self.w.recursive_button.enable(1)
 420                         self.w.force_button.enable(1)
 421                         self.w.user_button.enable(1)
 422                 self.w.packagebrowser.settopcell(topcell)
 423
 424         def listhit(self, *args, **kwargs):
 425                 self.updatestatus()
 426
 427         def do_install(self):
 428                 sel = self.w.packagebrowser.getselection()[0]
 429                 if self.w.verbose_button.get():
 430                         output = sys.stdout
 431                 else:
 432                         output = None
 433                 recursive = self.w.recursive_button.get()
 434                 force = self.w.force_button.get()
 435                 messages = self.installpackage(sel, output, recursive, force)
 436
 437                 # Re-read .pth files
 438                 import site
 439                 reload(site)
 440
 441                 self.updatestatus()
 442                 self.showmessages(messages)
 443
 444         def showmessages(self, messages):
 445                 if messages:
 446                         # To be on the safe side we always show the hidden packages,
 447                         # they may be referred to in the error messages.
 448                         if not self.w.hidden_button.get():
 449                                 self.w.hidden_button.set(1)
 450                                 self.updatestatus()
 451                         if type(messages) == list:
 452                                 messages = '\n'.join(messages)
 453                         if self.w.verbose_button.get():
 454                                 sys.stdout.write(messages + '\n')
 455                         EasyDialogs.Message(messages)
 456
 457         def do_homepage(self):
 458                 sel = self.w.packagebrowser.getselection()[0]
 459                 if not self.ic:
 460                         import ic
 461
 462                         self.ic = ic.IC()
 463                 self.ic.launchurl(self.packages[sel].homepage())
 464
 465         def do_user(self):
 466                 messages = self.setuserinstall(self.w.user_button.get())
 467                 self.updatestatus()
 468                 self.showmessages(messages)
 469
 470 if __name__ == '__main__':
 471         PackageManagerMain()