def runWizard():
"""Run the wizard dialog."""
# Make sure the help panel isn't updated during the wizard.
#events.send(events.Command('freeze'))
suggested = {
'doom1': 'DOOM.WAD',
'doom1-share': 'DOOM1.WAD',
'doom1-ultimate': 'DOOM.WAD',
'doom2': 'DOOM2.WAD',
'doom2-tnt': 'TNT.WAD',
'doom2-plut': 'PLUTONIA.WAD',
'heretic-ext': 'HERETIC.WAD',
'heretic-share': 'HERETIC.WAD',
'heretic': 'HERETIC.WAD',
'hexen': 'HEXEN.WAD',
'hexen-demo': 'HEXEN.WAD',
'hexen-dk': 'HEXEN.WAD',
'hacx': 'HACX.WAD',
'chex': 'CHEX.WAD'
}
events.mute()
# Make the Defaults profile active.
pr.setActive(pr.getDefaults())
wiz = WizardDialog(language.translate('setup-wizard'),
paths.findBitmap('wizard'))
# Language selection page.
langPage = WizardPage(wiz, 'wizard-language')
area = langPage.getArea()
area.createText('wizard-language-explanation', maxLineLength=65).resizeToBestSize()
sar, languageCheckBox = area.createSetting(st.getSetting('language'))
languageCheckBox.getFromProfile(pr.getActive())
# Game selection.
gamePage = ProfileWizardPage(wiz, 'wizard-games', None)
gamePage.follows(langPage)
area = gamePage.getArea()
area.createText('wizard-select-games', maxLineLength=65).resizeToBestSize()
area.setBorderDirs(ui.BORDER_NOT_BOTTOM)
games = area.createList('', style=sb.widget.list.List.STYLE_CHECKBOX)
area.setBorderDirs(ui.BORDER_NOT_TOP)
games.setMinSize(50, 180)
gamePage.setGameList(games)
def allGames():
# Check the entire list.
for item in games.getItems():
games.checkItem(item, True)
def clearGames():
# Uncheck the entire list.
for item in games.getItems():
games.checkItem(item, False)
controls = area.createArea(alignment=ui.ALIGN_HORIZONTAL, border=2)
controls.setWeight(0)
button = controls.createButton('wizard-games-all', wb.Button.STYLE_MINI)
button.addReaction(allGames)
button.resizeToBestSize()
button = controls.createButton('wizard-games-clear', wb.Button.STYLE_MINI)
button.addReaction(clearGames)
button.resizeToBestSize()
# The pages will be linked together.
previousPage = gamePage
deathKingsWad = None
# We'll do this dynamically.
checkedProfiles = ['doom1', 'doom2', 'heretic', 'hexen']
# Only display the system profiles in the wizard (not any user
# profiles).
profiles = pr.getProfiles(lambda p: p.isSystemProfile())
for p in profiles:
if p is not pr.getDefaults():
games.addItem(p.getId(), p.getId() in checkedProfiles)
# Create a page for the profile.
page = ProfileWizardPage(wiz, p.getId(), games)
# Link the pages together.
page.follows(previousPage)
previousPage = page
area = page.getArea()
area.createText('wizard-locate-iwad', maxLineLength=65).resizeToBestSize()
# The suggestion.
if suggested.has_key(p.getId()):
sugArea = area.createArea(alignment=ui.ALIGN_HORIZONTAL, border=2)
sugArea.setExpanding(False)
sugArea.setWeight(1)
sugArea.createText('wizard-suggested-iwad', ':', align=wt.Text.RIGHT)
sugArea.setWeight(2)
sug = sugArea.createText('')
sug.setText(suggested[p.getId()])
sar, page.iwadText = area.createSetting(st.getSetting('iwad'))
if p.getId() == 'hexen-dk':
area.setBorder(12, ui.BORDER_ALL)
area.createLine()
area.setBorder(6, ui.BORDER_ALL)
area.createText('deathkings-selection-title').setHeadingStyle()
# Death Kings is an extension to Hexen. It uses the
# same Hexen IWAD, but also an addon IWAD.
area.createText('wizard-locate-iwad-deathkings', maxLineLength=65).resizeToBestSize()
entry = area.createArea(alignment=ui.ALIGN_HORIZONTAL, border=4)
entry.setExpanding(False)
entry.setWeight(1)
entry.setBorderDirs(ui.BORDER_NOT_LEFT)
deathKingsWad = entry.createTextField('')
entry.setWeight(0)
entry.setBorderDirs(ui.BORDER_TOP_BOTTOM)
browseButton = entry.createButton('browse-button', wb.Button.STYLE_MINI)
def browseDeathKings():
# Open a file browser.
selection = sb.util.dialog.chooseFile('deathkings-selection-title',
'', True,
[('file-type-wad', 'wad')])
if len(selection) > 0:
deathKingsWad.setText(selection)
browseButton.addReaction(browseDeathKings)
# Addon paths.
pathsPage = ProfileWizardPage(wiz, 'wizard-addon-paths', games)
pathsPage.follows(previousPage)
area = pathsPage.getArea()
area.createText('wizard-addon-paths-explanation',
maxLineLength=65).resizeToBestSize()
area.setBorderDirs(ui.BORDER_NOT_BOTTOM)
pathList = area.createList('addon-paths-list')
pathList.setMinSize(100, 120)
# Insert the current custom paths into the list.
for p in paths.getAddonPaths():
pathList.addItem(p)
def addAddonPath():
selection = sb.util.dialog.chooseFolder('addon-paths-add-prompt', '')
if selection:
pathList.addItem(selection)
pathList.selectItem(selection)
def removeAddonPath():
selection = pathList.getSelectedItem()
if selection:
pathList.removeItem(selection)
area.setWeight(0)
area.setBorderDirs(ui.BORDER_NOT_TOP)
commands = area.createArea(alignment=ui.ALIGN_HORIZONTAL, border=2)
commands.setWeight(0)
# Button for adding new paths.
button = commands.createButton('new-addon-path', wb.Button.STYLE_MINI)
button.addReaction(addAddonPath)
# Button for removing a path.
button = commands.createButton('delete-addon-path', wb.Button.STYLE_MINI)
button.addReaction(removeAddonPath)
# Launch options.
quitPage = WizardPage(wiz, 'wizard-launching')
quitPage.follows(pathsPage)
area = quitPage.getArea()
area.createText('wizard-launching-explanation').resizeToBestSize()
sar, quitCheckBox = area.createSetting(st.getSetting('quit-on-launch'))
quitCheckBox.getFromProfile(pr.getActive())
# List of unusable profiles, due to a missing IWAD.
unusableProfiles = []
# When the page changes in the wizard, change the active profile.
def changeActiveProfile(page):
prof = pr.get(page.getId())
if prof:
pr.setActive(prof)
else:
pr.setActive(pr.getDefaults())
# Update page with values from the current profile.
page.update()
wiz.setPageReaction(changeActiveProfile)
if wiz.run(langPage) == 'ok':
events.unmute()
# Show the profiles that now have an IWAD.
for prof in profiles:
if prof.getValue('iwad', False) != None:
pr.show(prof.getId())
if prof.getId() in unusableProfiles:
unusableProfiles.remove(prof.getId())
else:
pr.hide(prof.getId())
if prof.getId() not in unusableProfiles and \
prof.getId() in games.getSelectedItems():
unusableProfiles.append(prof.getId())
# Install the Death Kings WAD?
if deathKingsWad:
try:
ident = ao.install(deathKingsWad.getText())
# Attach the WAD as an addon.
kingsProfile = pr.get('hexen-dk')
kingsProfile.useAddon(ident)
except:
# TODO: Handle error.
pass
# Update custom addon paths.
currentAddonPaths = paths.getAddonPaths()
wasModified = False
for path in pathList.getItems():
if path not in currentAddonPaths:
paths.addAddonPath(path)
wasModified = True
for path in currentAddonPaths:
if path not in pathList.getItems():
paths.removeAddonPath(path)
wasModified = True
if wasModified:
# Load addons from new paths.
ao.refresh()
events.send(events.Notify('addon-paths-changed'))
# The wizard will only be shown once automatically.
pr.getDefaults().setValue(HAS_BEEN_RUN, 'yes', False)
else:
# Wizard was canceled.
events.unmute()
pr.refresh()
# This'll destroy all the pages of the wizard as well.
wiz.destroy()
# Enable help panel updates again.
#events.send(events.Command('unfreeze'))
# Tell the user about unusable profiles.
if len(unusableProfiles) > 0:
dialog, area = sb.util.dialog.createButtonDialog(
'wizard-unlaunchable-profiles',
['ok'], 'ok', resizable=False)
# Compose a list of the unlaunchable profiles.
profList = ''
unusableProfiles.sort(lambda a, b: cmp(language.translate(a),
language.translate(b)))
for p in unusableProfiles:
profList += "\n" + language.translate(p)
msg = area.createText()
msg.setText(language.translate('wizard-unlaunchable-profiles-listed') + "\n" +
profList)
msg.resizeToBestSize()
dialog.run()
def chooseAddons(dialogId, title, actionButton):
"""Opens an addon selection dialog.
@param title Title of the dialog.
@param actionButton The button that will perform the affirmative
action of the dialog.
"""
dialog, area = sb.util.dialog.createButtonDialog(dialogId,
['cancel', actionButton],
actionButton)
dialog.addEndCommand('addon-dialog-add-to-custom-folders')
# The action button is disabled until a valid selection has been
# made.
dialog.disableWidget(actionButton)
area.setWeight(0)
folderArea = area.createArea(alignment=ui.ALIGN_HORIZONTAL)
folderArea.setExpanding(False)
folderArea.setBorder(2, ui.BORDER_NOT_LEFT)
folderArea.setWeight(0)
folderArea.createText('addon-dialog-folder').resizeToBestSize()
folderArea.setBorderDirs(ui.BORDER_ALL)
folderArea.setWeight(1)
pathField = folderArea.createTextField('')
pathField.setText(os.getcwd())
pathField.select()
pathField.focus()
folderArea.setWeight(0)
folderArea.setBorderDirs(ui.BORDER_NOT_RIGHT)
browseButton = folderArea.createButton('addon-dialog-folder-browse',
style=wg.Button.STYLE_MINI)
#folderArea.addSpacer()
#folderArea.setBorderDirs(ui.BORDER_NOT_RIGHT)
folderCmdArea = area.createArea(alignment=ui.ALIGN_HORIZONTAL)
folderCmdArea.setExpanding(False)
folderCmdArea.setBorder(6, ui.BORDER_NOT_TOP)
folderCmdArea.setWeight(1)
folderCmdArea.addSpacer()
folderCmdArea.setWeight(0)
uninstButton = folderCmdArea.createButton('addon-dialog-folder-uninstalled')
# Add to Custom Folders button.
folderCmdArea.setBorder(6, ui.BORDER_LEFT | ui.BORDER_BOTTOM)
addToMyButton = folderCmdArea.createButton('addon-dialog-add-to-custom-folders')
def goToUninstalled():
pathField.setText(paths.getUserPath(paths.UNINSTALLED))
pathField.select()
addToMyButton.disable()
uninstButton.addReaction(goToUninstalled)
area.createText('addon-dialog-found')
area.setWeight(1)
foundList = area.createList('', style=sb.widget.list.List.STYLE_COLUMNS)
foundList.setMinSize(500, 300)
area.setWeight(0)
area.createText('addon-dialog-addons-copied', maxLineLength=70).resizeToBestSize()
def selectAction():
dialog.enableWidget(actionButton)
foundList.addReaction(selectAction)
for col, width in [('name', None), ('type', 180)]:
foundList.addColumn('addon-dialog-' + col, width)
def updateList():
# Update the found addons list.
foundList.clear()
dialog.disableWidget(actionButton)
extensions = ao.getAddonExtensions() + ['manifest']
# This should be done in addons.py.
fileNames = os.listdir(pathField.getText())
for name in fileNames:
type = ''
for ext in extensions:
if paths.hasExtension(ext, name):
type = ext
break
if not type:
# Unknown files are skipped.
continue
# Manifests don't appear in the list if the corresponding
# addon is in the same directory.
if paths.hasExtension('manifest', name):
foundSame = False
# Identifier of the addon the manifest belongs to.
manifestId = paths.getBase(name)
# See if the addon is in the list.
for other in fileNames:
if other == name:
continue
if manifestId == ao.formIdentifier(other)[0]:
foundSame = True
break
if foundSame:
# Don't add it.
continue
foundList.addItemWithColumns(
name, name, language.translate('addon-dialog-type-' + type))
# Update reactions.
def pathChanged():
if os.path.exists(pathField.getText()):
updateList()
if pathField.getText() != paths.getUserPath(paths.UNINSTALLED):
addToMyButton.enable()
else:
addToMyButton.disable()
def browseAction():
# Show a directory browser.
selection = sb.util.dialog.chooseFolder('addon-dialog-browse-prompt',
pathField.getText())
if len(selection):
pathField.setText(selection)
pathField.select()
# The initial contents of the list.
updateList()
pathField.addReaction(pathChanged)
browseButton.addReaction(browseAction)
dialog.addEndCommand(actionButton)
result = dialog.run()
if result == actionButton:
addonFiles = map(lambda name: os.path.join(pathField.getText(), name),
foundList.getSelectedItems())
# Include any associated manifests.
for name in addonFiles:
manifest = ao.formIdentifier(name)[0] + '.manifest'
manifestFile = os.path.join(pathField.getText(), manifest)
if manifest not in addonFiles and os.path.exists(manifestFile):
addonFiles.append(manifestFile)
#print 'including ' + manifestFile + ' due to ' + name
return addonFiles
elif result == 'addon-dialog-add-to-custom-folders':
paths.addAddonPath(pathField.getText())
events.send(events.Notify('addon-paths-changed'))
ao.refresh()
return []
# The dialog was canceled.
return []
def chooseAddons(dialogId, title, actionButton):
"""Opens an addon selection dialog.
@param title Title of the dialog.
@param actionButton The button that will perform the affirmative
action of the dialog.
"""
dialog, area = sb.util.dialog.createButtonDialog(dialogId,
['cancel', actionButton],
actionButton)
dialog.addEndCommand('addon-dialog-add-to-custom-folders')
# The action button is disabled until a valid selection has been
# made.
dialog.disableWidget(actionButton)
area.setWeight(0)
folderArea = area.createArea(alignment=ui.ALIGN_HORIZONTAL)
folderArea.setExpanding(False)
folderArea.setBorder(2, ui.BORDER_NOT_LEFT)
folderArea.setWeight(0)
folderArea.createText('addon-dialog-folder').resizeToBestSize()
folderArea.setBorderDirs(ui.BORDER_ALL)
folderArea.setWeight(1)
pathField = folderArea.createTextField('')
pathField.setText(os.getcwd())
pathField.select()
pathField.focus()
folderArea.setWeight(0)
folderArea.setBorderDirs(ui.BORDER_NOT_RIGHT)
browseButton = folderArea.createButton('addon-dialog-folder-browse',
style=wg.Button.STYLE_MINI)
#folderArea.addSpacer()
#folderArea.setBorderDirs(ui.BORDER_NOT_RIGHT)
folderCmdArea = area.createArea(alignment=ui.ALIGN_HORIZONTAL)
folderCmdArea.setExpanding(False)
folderCmdArea.setBorder(6, ui.BORDER_NOT_TOP)
folderCmdArea.setWeight(1)
folderCmdArea.addSpacer()
folderCmdArea.setWeight(0)
uninstButton = folderCmdArea.createButton(
'addon-dialog-folder-uninstalled')
# Add to Custom Folders button.
folderCmdArea.setBorder(6, ui.BORDER_LEFT | ui.BORDER_BOTTOM)
addToMyButton = folderCmdArea.createButton(
'addon-dialog-add-to-custom-folders')
def goToUninstalled():
pathField.setText(paths.getUserPath(paths.UNINSTALLED))
pathField.select()
addToMyButton.disable()
uninstButton.addReaction(goToUninstalled)
area.createText('addon-dialog-found')
area.setWeight(1)
foundList = area.createList('', style=sb.widget.list.List.STYLE_COLUMNS)
foundList.setMinSize(500, 300)
area.setWeight(0)
area.createText('addon-dialog-addons-copied',
maxLineLength=70).resizeToBestSize()
def selectAction():
dialog.enableWidget(actionButton)
foundList.addReaction(selectAction)
for col, width in [('name', None), ('type', 180)]:
foundList.addColumn('addon-dialog-' + col, width)
def updateList():
# Update the found addons list.
foundList.clear()
dialog.disableWidget(actionButton)
extensions = ao.getAddonExtensions() + ['manifest']
# This should be done in addons.py.
fileNames = os.listdir(pathField.getText())
for name in fileNames:
type = ''
for ext in extensions:
if paths.hasExtension(ext, name):
type = ext
break
if not type:
# Unknown files are skipped.
continue
# Manifests don't appear in the list if the corresponding
# addon is in the same directory.
if paths.hasExtension('manifest', name):
foundSame = False
# Identifier of the addon the manifest belongs to.
manifestId = paths.getBase(name)
# See if the addon is in the list.
for other in fileNames:
if other == name:
continue
if manifestId == ao.formIdentifier(other)[0]:
foundSame = True
break
if foundSame:
# Don't add it.
continue
foundList.addItemWithColumns(
name, name, language.translate('addon-dialog-type-' + type))
# Update reactions.
def pathChanged():
if os.path.exists(pathField.getText()):
updateList()
if pathField.getText() != paths.getUserPath(paths.UNINSTALLED):
addToMyButton.enable()
else:
addToMyButton.disable()
def browseAction():
# Show a directory browser.
selection = sb.util.dialog.chooseFolder('addon-dialog-browse-prompt',
pathField.getText())
if len(selection):
pathField.setText(selection)
pathField.select()
# The initial contents of the list.
updateList()
pathField.addReaction(pathChanged)
browseButton.addReaction(browseAction)
dialog.addEndCommand(actionButton)
result = dialog.run()
if result == actionButton:
addonFiles = map(lambda name: os.path.join(pathField.getText(), name),
foundList.getSelectedItems())
# Include any associated manifests.
for name in addonFiles:
manifest = ao.formIdentifier(name)[0] + '.manifest'
manifestFile = os.path.join(pathField.getText(), manifest)
if manifest not in addonFiles and os.path.exists(manifestFile):
addonFiles.append(manifestFile)
#print 'including ' + manifestFile + ' due to ' + name
return addonFiles
elif result == 'addon-dialog-add-to-custom-folders':
paths.addAddonPath(pathField.getText())
events.send(events.Notify('addon-paths-changed'))
ao.refresh()
return []
# The dialog was canceled.
return []
def handleCommand(event):
"""This is called when someone sends a command event.
@param event An events.Command object."""
# Figure out which addon has been currently selected in the addons
# tree. The action will target this addon.
selected = ''
if event.hasId('refresh-addon-database'):
ao.refresh()
elif event.hasId('install-addon'):
tab30.installer.run()
elif event.hasId('uninstall-addon'):
selectedItems = addonList.getSelectedItems()
# Only take the uninstallable addons.
addons = filter(lambda i: ao.isUninstallable(i), selectedItems)
if len(addons) == 0:
return
# Make sure the user wants to uninstall.
dialog, area = sb.util.dialog.createButtonDialog(
'uninstall-addon-dialog', ['no', 'yes'], 'no', resizable=False)
if len(addons) == 1:
text = language.translate('uninstall-addon-query')
area.setExpanding(True)
area.createRichText(
language.expand(text, language.translate(addons[0])))
else:
area.setWeight(0)
area.createText('uninstall-addon-query-several')
msg = '<html><ul>'
for addon in addons:
msg += '<li>' + language.translate(addon) + '</li>'
msg += '</ul></html>'
ft = area.createFormattedText()
ft.setText(msg)
ft.setMinSize(400, 150)
if dialog.run() == 'yes':
for addon in addons:
ao.uninstall(addon)
ao.refresh()
elif event.hasId('addon-info'):
sel = addonList.getSelectedItems()
if len(sel) > 0:
# Show the Addon Inspector.
tab30.inspector.run(ao.get(sel[0]))
elif event.hasId('addon-settings'):
sel = addonList.getSelectedItems()
if len(sel) > 0:
command = events.Command('show-addon-settings')
command.addon = sel[0]
events.send(command)
elif event.hasId('load-order'):
# Open the Load Order dialog.
tab30.loadorder.run(pr.getActive())
elif event.hasId('addon-list-check-selected'):
for addon in addonList.getSelectedItems():
pr.getActive().useAddon(addon)
elif event.hasId('addon-list-uncheck-selected'):
for addon in addonList.getSelectedItems():
pr.getActive().dontUseAddon(addon)
elif event.hasId('addon-list-check-all'):
for addon in addonList.getItems():
pr.getActive().useAddon(addon)
elif event.hasId('addon-list-uncheck-all'):
for addon in addonList.getItems():
pr.getActive().dontUseAddon(addon)
elif event.hasId('addon-show-parent-box'):
for addon in addonList.getSelectedItems():
a = ao.get(addon)
if a.getBox():
showInAddonList(a.getBox().getId())
break
elif event.hasId('addon-show-box-category'):
for addon in addonList.getSelectedItems():
a = ao.get(addon)
if a.getType() == 'addon-type-box':
addonList.deselectAllItems()
tree.selectItem(a.getContentCategoryLongId())
refreshListIfVisible()
break
def handleCommand(event):
"""This is called when someone sends a command event.
@param event An events.Command object."""
# Figure out which addon has been currently selected in the addons
# tree. The action will target this addon.
selected = ''
if event.hasId('refresh-addon-database'):
ao.refresh()
elif event.hasId('install-addon'):
tab30.installer.run()
elif event.hasId('uninstall-addon'):
selectedItems = addonList.getSelectedItems()
# Only take the uninstallable addons.
addons = filter(lambda i: ao.isUninstallable(i), selectedItems)
if len(addons) == 0:
return
# Make sure the user wants to uninstall.
dialog, area = sb.util.dialog.createButtonDialog(
'uninstall-addon-dialog',
['no', 'yes'], 'no', resizable=False)
if len(addons) == 1:
text = language.translate('uninstall-addon-query')
area.setExpanding(True)
area.createRichText(language.expand(text, language.translate(addons[0])))
else:
area.setWeight(0)
area.createText('uninstall-addon-query-several')
msg = '<html><ul>'
for addon in addons:
msg += '<li>' + language.translate(addon) + '</li>'
msg += '</ul></html>'
ft = area.createFormattedText()
ft.setText(msg)
ft.setMinSize(400, 150)
if dialog.run() == 'yes':
for addon in addons:
ao.uninstall(addon)
ao.refresh()
elif event.hasId('addon-info'):
sel = addonList.getSelectedItems()
if len(sel) > 0:
# Show the Addon Inspector.
tab30.inspector.run(ao.get(sel[0]))
elif event.hasId('addon-settings'):
sel = addonList.getSelectedItems()
if len(sel) > 0:
command = events.Command('show-addon-settings')
command.addon = sel[0]
events.send(command)
elif event.hasId('load-order'):
# Open the Load Order dialog.
tab30.loadorder.run(pr.getActive())
elif event.hasId('addon-list-check-selected'):
for addon in addonList.getSelectedItems():
pr.getActive().useAddon(addon)
elif event.hasId('addon-list-uncheck-selected'):
for addon in addonList.getSelectedItems():
pr.getActive().dontUseAddon(addon)
elif event.hasId('addon-list-check-all'):
for addon in addonList.getItems():
pr.getActive().useAddon(addon)
elif event.hasId('addon-list-uncheck-all'):
for addon in addonList.getItems():
pr.getActive().dontUseAddon(addon)
elif event.hasId('addon-show-parent-box'):
for addon in addonList.getSelectedItems():
a = ao.get(addon)
if a.getBox():
showInAddonList(a.getBox().getId())
break
elif event.hasId('addon-show-box-category'):
for addon in addonList.getSelectedItems():
a = ao.get(addon)
if a.getType() == 'addon-type-box':
addonList.deselectAllItems()
tree.selectItem(a.getContentCategoryLongId())
refreshListIfVisible()
break
