| """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 |
| import Res |
| import os |
| from MACFS import * |
| import MacOS |
| try: |
| openrf = MacOS.openrf |
| except AttributeError: |
| # Backward compatability |
| openrf = open |
| |
| Error = 'macostools.Error' |
| |
| FSSpecType = type(macfs.FSSpec(':')) |
| |
| 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): |
| """Create a finder alias""" |
| srcfss = macfs.FSSpec(src) |
| dstfss = macfs.FSSpec(dst) |
| alias = srcfss.NewAlias() |
| srcfinfo = srcfss.GetFInfo() |
| |
| Res.FSpCreateResFile(dstfss, srcfinfo.Creator, srcfinfo.Type, -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) |
| print 'XX', dst, '->', (head, tail) |
| # XXXX Is this a bug in os.path.split? |
| if not ':' in head: |
| head = head + ':' |
| mkdirs(head) |
| os.mkdir(dst, 0777) |
| |
| def copy(src, dst, createpath=0): |
| """Copy a file, including finder info, resource fork, etc""" |
| if createpath: |
| mkdirs(os.path.split(dst)[0]) |
| srcfss = macfs.FSSpec(src) |
| dstfss = macfs.FSSpec(dst) |
| |
| ifp = open(srcfss.as_pathname(), 'rb') |
| ofp = open(dstfss.as_pathname(), 'wb') |
| d = ifp.read(BUFSIZ) |
| while d: |
| ofp.write(d) |
| d = ifp.read(BUFSIZ) |
| ifp.close() |
| ofp.close() |
| |
| ifp = openrf(srcfss.as_pathname(), '*rb') |
| ofp = openrf(dstfss.as_pathname(), '*wb') |
| d = ifp.read(BUFSIZ) |
| while d: |
| ofp.write(d) |
| d = ifp.read(BUFSIZ) |
| ifp.close() |
| ofp.close() |
| |
| sf = srcfss.GetFInfo() |
| df = dstfss.GetFInfo() |
| df.Creator, df.Type = sf.Creator, sf.Type |
| df.Flags = (sf.Flags & (kIsStationary|kNameLocked|kHasBundle|kIsInvisible|kIsAlias)) |
| dstfss.SetFInfo(df) |
| |
| def copytree(src, dst): |
| """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)) |
| else: |
| copy(src, dst, 1) |