cpython/Mac/scripts/EditPythonPrefs.py

"""Edit the Python Preferences file."""
#
# This program is getting more and more clunky. It should really
# be rewritten in a modeless way some time soon.

from Dlg import *
from Events import *
from Res import *
import string
import struct
import macfs
import MacOS
import os
import sys
import Res # For Res.Error

# resource IDs in our own resources (dialogs, etc)
MESSAGE_ID = 256

DIALOG_ID = 512
TEXT_ITEM = 1
OK_ITEM = 2
CANCEL_ITEM = 3
DIR_ITEM = 4
TITLE_ITEM = 5
OPTIONS_ITEM = 7

# The options dialog. There is a correspondence between
# the dialog item numbers and the option.
OPT_DIALOG_ID = 513
# 1 thru 7 are the options
OD_OK_ITEM = 8
OD_CANCEL_ITEM = 9

# Resource IDs in the preferences file
PATH_STRINGS_ID = 128
DIRECTORY_ID = 128
OPTIONS_ID = 128

# Override IDs (in the applet)
OVERRIDE_PATH_STRINGS_ID = 129
OVERRIDE_DIRECTORY_ID = 129
OVERRIDE_OPTIONS_ID = 129

READ = 1
WRITE = 2
smAllScripts = -3
kOnSystemDisk = 0x8000

def restolist(data):
	"""Convert STR# resource data to a list of strings"""
	if not data:
		return []
	num, = struct.unpack('h', data[:2])
	data = data[2:]
	rv = []
	for i in range(num):
		strlen = ord(data[0])
		if strlen < 0: strlen = strlen + 256
		str = data[1:strlen+1]
		data = data[strlen+1:]
		rv.append(str)
	return rv
	
def listtores(list):
	"""Convert a list of strings to STR# resource data"""
	rv = struct.pack('h', len(list))
	for str in list:
		rv = rv + chr(len(str)) + str
	return rv

def message(str = "Hello, world!", id = MESSAGE_ID):
	"""Show a simple alert with a text message"""
	d = GetNewDialog(id, -1)
	d.SetDialogDefaultItem(1)
	print 'd=', d
	tp, h, rect = d.GetDialogItem(2)
	SetDialogItemText(h, str)
	while 1:
		n = ModalDialog(None)
		if n == 1: break
		
def optinteract(options):
	"""Let the user interact with the options dialog"""
	old_options = options[:]
	d = GetNewDialog(OPT_DIALOG_ID, -1)
	d.SetDialogDefaultItem(OD_OK_ITEM)
	d.SetDialogCancelItem(OD_CANCEL_ITEM)
	while 1:
		for i in range(len(options)):
			tp, h, rect = d.GetDialogItem(i+1)
			h.as_Control().SetControlValue(options[i])
		n = ModalDialog(None)
		if n == OD_OK_ITEM:
			return options
		elif n == OD_CANCEL_ITEM:
			return old_options
		elif 1 <= n <= len(options):
			options[n-1] = (not options[n-1])

			
def interact(list, pythondir, options, title):
	"""Let the user interact with the dialog"""
	opythondir = pythondir
	try:
		# Try to go to the "correct" dir for GetDirectory
		os.chdir(pythondir.as_pathname())
	except os.error:
		pass
	d = GetNewDialog(DIALOG_ID, -1)
	tp, h, rect = d.GetDialogItem(TITLE_ITEM)
	SetDialogItemText(h, title)
	tp, h, rect = d.GetDialogItem(TEXT_ITEM)
	SetDialogItemText(h, string.joinfields(list, '\r'))
##	d.SetDialogDefaultItem(OK_ITEM)
	d.SetDialogCancelItem(CANCEL_ITEM)
	while 1:
		n = ModalDialog(None)
		if n == OK_ITEM:
			break
		if n == CANCEL_ITEM:
			return None
##		if n == REVERT_ITEM:
##			return [], pythondir
		if n == DIR_ITEM:
			fss, ok = macfs.GetDirectory('Select python home folder:')
			if ok:
				pythondir = fss
		if n == OPTIONS_ITEM:
			options = optinteract(options)
	tmp = string.splitfields(GetDialogItemText(h), '\r')
	rv = []
	for i in tmp:
		if i:
			rv.append(i)
	return rv, pythondir, options
	
def getprefpath(id):
	# Load the path and directory resources
	try:
		sr = GetResource('STR#', id)
	except (MacOS.Error, Res.Error):
		return None, None
	d = sr.data
	l = restolist(d)
	return l, sr

def getprefdir(id):
	try:
		dr = GetResource('alis', id)
		fss, fss_changed = macfs.RawAlias(dr.data).Resolve()
	except (MacOS.Error, Res.Error):
		return None, None, 1
	return fss, dr, fss_changed

def getoptions(id):
	try:
		opr = GetResource('Popt', id)
	except (MacOS.Error, Res.Error):
		return [0]*7, None
	return map(lambda x: ord(x), opr.data), opr
	
def openpreffile(rw):
	# Find the preferences folder and our prefs file, create if needed.	
	vrefnum, dirid = macfs.FindFolder(kOnSystemDisk, 'pref', 0)
	preff_fss = macfs.FSSpec((vrefnum, dirid, 'Python Preferences'))
	try:
		preff_handle = FSpOpenResFile(preff_fss, rw)
	except Res.Error:
		# Create it
		message('No preferences file, creating one...')
		FSpCreateResFile(preff_fss, 'Pyth', 'pref', smAllScripts)
		preff_handle = FSpOpenResFile(preff_fss, rw)
	return preff_handle
	
def openapplet(name):
	fss = macfs.FSSpec(name)
	try:
		app_handle = FSpOpenResFile(fss, WRITE)
	except Res.Error:
		message('File does not have a resource fork.')
		sys.exit(0)
	return app_handle
		
	
def edit_preferences():
	preff_handle = openpreffile(WRITE)
	
	l, sr = getprefpath(PATH_STRINGS_ID)
	if l == None:	
		message('Cannot find any sys.path resource! (Old python?)')
		sys.exit(0)
		
	fss, dr, fss_changed = getprefdir(DIRECTORY_ID)
	if fss == None:
		fss = macfs.FSSpec(os.getcwd())
		fss_changed = 1
		
	options, opr = getoptions(OPTIONS_ID)
	saved_options = options[:]
	
	# Let the user play away
	result = interact(l, fss, options, 'System-wide preferences')
	
	# See what we have to update, and how
	if result == None:
		sys.exit(0)
		
	pathlist, nfss, options = result
	if nfss != fss:
		fss_changed = 1
		
	if fss_changed:
		alias = nfss.NewAlias()
		if dr:
			dr.data = alias.data
			dr.ChangedResource()
		else:
			dr = Resource(alias.data)
			dr.AddResource('alis', DIRECTORY_ID, '')
			
	if pathlist != l:
		if pathlist == []:
			if sr.HomeResFile() == preff_handle:
				sr.RemoveResource()
		elif sr.HomeResFile() == preff_handle:
			sr.data = listtores(pathlist)
			sr.ChangedResource()
		else:
			sr = Resource(listtores(pathlist))
			sr.AddResource('STR#', PATH_STRINGS_ID, '')
			
	if options != saved_options:
		newdata = reduce(lambda x, y: x+chr(y), options, '')
		if opr and opr.HomeResFile() == preff_handle:
			opr.data = newdata
			opr.ChangedResource()
		else:
			opr = Resource(newdata)
			opr.AddResource('Popt', OPTIONS_ID, '')
				
	CloseResFile(preff_handle)
	
def edit_applet(name):
	pref_handle = openpreffile(READ)
	app_handle = openapplet(name)
	
	notfound = ''
	l, sr = getprefpath(OVERRIDE_PATH_STRINGS_ID)
	if l == None:
		notfound = 'path'
		
		l, dummy = getprefpath(PATH_STRINGS_ID)
		if l == None:	
			message('Cannot find any sys.path resource! (Old python?)')
			sys.exit(0)
		
	fss, dr, fss_changed = getprefdir(OVERRIDE_DIRECTORY_ID)
	if fss == None:
		if notfound:
			notfound = notfound + ', directory'
		else:
			notfound = 'directory'
		fss, dummy, dummy2 = getprefdir(DIRECTORY_ID)
		if fss == None:
			fss = macfs.FSSpec(os.getcwd())
			fss_changed = 1

	options, opr = getoptions(OVERRIDE_OPTIONS_ID)
	if not opr:
		if notfound:
			notfound = notfound + ', options'
		else:
			notfound = 'options'
		options, dummy = getoptions(OPTIONS_ID)
	saved_options = options[:]
	
	dummy = dummy2 = None # Discard them.
	
	if notfound:
		message('Warning: initial %s taken from system-wide defaults'%notfound)
	# Let the user play away
	result = interact(l, fss, options, name)
	
	# See what we have to update, and how
	if result == None:
		sys.exit(0)
		
	pathlist, nfss, options = result
	if nfss != fss:
		fss_changed = 1
		
	if fss_changed:
		alias = nfss.NewAlias()
		if dr:
			dr.data = alias.data
			dr.ChangedResource()
		else:
			dr = Resource(alias.data)
			dr.AddResource('alis', OVERRIDE_DIRECTORY_ID, '')
			
	if pathlist != l:
		if pathlist == []:
			if sr.HomeResFile() == app_handle:
				sr.RemoveResource()
		elif sr and sr.HomeResFile() == app_handle:
			sr.data = listtores(pathlist)
			sr.ChangedResource()
		else:
			sr = Resource(listtores(pathlist))
			sr.AddResource('STR#', OVERRIDE_PATH_STRINGS_ID, '')
			
	if options != saved_options:
		newdata = reduce(lambda x, y: x+chr(y), options, '')
		if opr and opr.HomeResFile() == app_handle:
			opr.data = newdata
			opr.ChangedResource()
		else:
			opr = Resource(newdata)
			opr.AddResource('Popt', OVERRIDE_OPTIONS_ID, '')
				
	CloseResFile(app_handle)

def main():
	try:
		h = OpenResFile('EditPythonPrefs.rsrc')
	except Res.Error:
		pass	# Assume we already have acces to our own resource
	
	if len(sys.argv) <= 1:
		edit_preferences()
	else:
		for appl in sys.argv[1:]:
			edit_applet(appl)
		

if __name__ == '__main__':
	print # Stupid, to init toolboxes...
	main()