1 files changed, 137 insertions, 0 deletions
diff --git a/Lib/plat-mac/macostools.py b/Lib/plat-mac/macostools.py
new file mode 100644
index 0000000..8822ab8
--- /dev/null
+++ b/Lib/plat-mac/macostools.py
@@ -0,0 +1,137 @@
+"""macostools - Various utility functions for MacOS.
+
+mkalias(src, dst) - Create a finder alias 'dst' pointing to 'src'
+copy(src, dst) - Full copy of 'src' to 'dst'
+"""
+
+import macfs
+from Carbon import Res
+import os
+from MACFS import *
+import MacOS
+import time
+try:
+	openrf = MacOS.openrf
+except AttributeError:
+	# Backward compatability
+	openrf = open
+
+Error = 'macostools.Error'
+
+BUFSIZ=0x80000		# Copy in 0.5Mb chunks
+
+#
+# Not guaranteed to be correct or stay correct (Apple doesn't tell you
+# how to do this), but it seems to work.
+#
+def mkalias(src, dst, relative=None):
+	"""Create a finder alias"""
+	srcfss = macfs.FSSpec(src)
+	# The next line will fail under unix-Python if the destination
+	# doesn't exist yet. We should change this code to be fsref-based.
+	dstfss = macfs.FSSpec(dst)
+	if relative:
+		relativefss = macfs.FSSpec(relative)
+		# ik mag er geen None in stoppen :-(
+		alias = srcfss.NewAlias(relativefss)
+	else:
+		alias = srcfss.NewAlias()
+	
+	if os.path.isdir(src):
+		cr, tp = 'MACS', 'fdrp'
+	else:
+		cr, tp = srcfss.GetCreatorType()
+	
+	Res.FSpCreateResFile(dstfss, cr, tp, -1)
+	h = Res.FSpOpenResFile(dstfss, 3)
+	resource = Res.Resource(alias.data)
+	resource.AddResource('alis', 0, '')
+	Res.CloseResFile(h)
+	
+	dstfinfo = dstfss.GetFInfo()
+	dstfinfo.Flags = dstfinfo.Flags|0x8000    # Alias flag
+	dstfss.SetFInfo(dstfinfo)
+	
+def mkdirs(dst):
+	"""Make directories leading to 'dst' if they don't exist yet"""
+	if dst == '' or os.path.exists(dst):
+		return
+	head, tail = os.path.split(dst)
+	if os.sep == ':' and not ':' in head:
+		head = head + ':'
+	mkdirs(head)
+	os.mkdir(dst, 0777)
+	
+def touched(dst):
+	"""Tell the finder a file has changed"""
+	file_fss = macfs.FSSpec(dst)
+	vRefNum, dirID, name = file_fss.as_tuple()
+	dir_fss = macfs.FSSpec((vRefNum, dirID, ''))
+	crdate, moddate, bkdate = dir_fss.GetDates()
+	now = time.time()
+	if now == moddate:
+		now = now + 1
+	try:
+		dir_fss.SetDates(crdate, now, bkdate)
+	except macfs.error:
+		pass
+	
+def touched_ae(dst):
+	"""Tell the finder a file has changed"""
+	import Finder
+	f = Finder.Finder()
+	file_fss = macfs.FSSpec(dst)
+	vRefNum, dirID, name = file_fss.as_tuple()
+	dir_fss = macfs.FSSpec((vRefNum, dirID, ''))
+	f.update(dir_fss)
+	
+def copy(src, dst, createpath=0, copydates=1, forcetype=None):
+	"""Copy a file, including finder info, resource fork, etc"""
+	if hasattr(src, 'as_pathname'):
+		src = src.as_pathname()
+	if hasattr(dst, 'as_pathname'):
+		dst = dst.as_pathname()
+	if createpath:
+		mkdirs(os.path.split(dst)[0])
+	
+	ifp = open(src, 'rb')
+	ofp = open(dst, 'wb')
+	d = ifp.read(BUFSIZ)
+	while d:
+		ofp.write(d)
+		d = ifp.read(BUFSIZ)
+	ifp.close()
+	ofp.close()
+	
+	ifp = openrf(src, '*rb')
+	ofp = openrf(dst, '*wb')
+	d = ifp.read(BUFSIZ)
+	while d:
+		ofp.write(d)
+		d = ifp.read(BUFSIZ)
+	ifp.close()
+	ofp.close()
+	
+	srcfss = macfs.FSSpec(src)
+	dstfss = macfs.FSSpec(dst)
+	sf = srcfss.GetFInfo()
+	df = dstfss.GetFInfo()
+	df.Creator, df.Type = sf.Creator, sf.Type
+	if forcetype != None:
+		df.Type = forcetype
+	df.Flags = (sf.Flags & (kIsStationary|kNameLocked|kHasBundle|kIsInvisible|kIsAlias))
+	dstfss.SetFInfo(df)
+	if copydates:
+		crdate, mddate, bkdate = srcfss.GetDates()
+		dstfss.SetDates(crdate, mddate, bkdate)
+	touched(dstfss)
+	
+def copytree(src, dst, copydates=1):
+	"""Copy a complete file tree to a new destination"""
+	if os.path.isdir(src):
+		mkdirs(dst)
+		files = os.listdir(src)
+		for f in files:
+			copytree(os.path.join(src, f), os.path.join(dst, f), copydates)
+	else:
+		copy(src, dst, 1, copydates)