Lib/plat-mac/macostools.py

   1 """macostools - Various utility functions for MacOS.
   2
   3 mkalias(src, dst) - Create a finder alias 'dst' pointing to 'src'
   4 copy(src, dst) - Full copy of 'src' to 'dst'
   5 """
   6
   7 from Carbon import Res
   8 from Carbon import File, Files
   9 import os
  10 import sys
  11 import MacOS
  12 import time
  13 try:
  14     openrf = MacOS.openrf
  15 except AttributeError:
  16     # Backward compatability
  17     openrf = open
  18
  19 Error = 'macostools.Error'
  20
  21 BUFSIZ=0x80000      # Copy in 0.5Mb chunks
  22
  23 COPY_FLAGS = (Files.kIsStationary|Files.kNameLocked|Files.kHasBundle|
  24               Files.kIsInvisible|Files.kIsAlias)
  25
  26 #
  27 # Not guaranteed to be correct or stay correct (Apple doesn't tell you
  28 # how to do this), but it seems to work.
  29 #
  30 def mkalias(src, dst, relative=None):
  31     """Create a finder alias"""
  32     srcfsr = File.FSRef(src)
  33     # The next line will fail under unix-Python if the destination
  34     # doesn't exist yet. We should change this code to be fsref-based.
  35     dstdir, dstname = os.path.split(dst)
  36     if not dstdir: dstdir = os.curdir
  37     dstdirfsr = File.FSRef(dstdir)
  38     if relative:
  39         relativefsr = File.FSRef(relative)
  40         # ik mag er geen None in stoppen :-(
  41         alias = File.FSNewAlias(relativefsr, srcfsr)
  42     else:
  43         alias = srcfsr.FSNewAliasMinimal()
  44
  45     dstfsr, dstfss = Res.FSCreateResourceFile(dstdirfsr, unicode(dstname),
  46         File.FSGetResourceForkName())
  47     h = Res.FSOpenResourceFile(dstfsr, File.FSGetResourceForkName(), 3)
  48     resource = Res.Resource(alias.data)
  49     resource.AddResource('alis', 0, '')
  50     Res.CloseResFile(h)
  51
  52     dstfinfo = dstfss.FSpGetFInfo()
  53     dstfinfo.Flags = dstfinfo.Flags|0x8000    # Alias flag
  54     dstfss.FSpSetFInfo(dstfinfo)
  55
  56 def mkdirs(dst):
  57     """Make directories leading to 'dst' if they don't exist yet"""
  58     if dst == '' or os.path.exists(dst):
  59         return
  60     head, tail = os.path.split(dst)
  61     if os.sep == ':' and not ':' in head:
  62         head = head + ':'
  63     mkdirs(head)
  64     os.mkdir(dst, 0777)
  65
  66 def touched(dst):
  67     """Tell the finder a file has changed. No-op on MacOSX."""
  68     if sys.platform != 'mac': return
  69     import warnings
  70     warnings.filterwarnings("ignore", "macfs.*", DeprecationWarning, __name__)
  71     import macfs
  72     file_fss = macfs.FSSpec(dst)
  73     vRefNum, dirID, name = file_fss.as_tuple()
  74     dir_fss = macfs.FSSpec((vRefNum, dirID, ''))
  75     crdate, moddate, bkdate = dir_fss.GetDates()
  76     now = time.time()
  77     if now == moddate:
  78         now = now + 1
  79     try:
  80         dir_fss.SetDates(crdate, now, bkdate)
  81     except macfs.error:
  82         pass
  83
  84 def touched_ae(dst):
  85     """Tell the finder a file has changed"""
  86     pardir = os.path.split(dst)[0]
  87     if not pardir:
  88         pardir = os.curdir
  89     import Finder
  90     f = Finder.Finder()
  91     f.update(File.FSRef(pardir))
  92
  93 def copy(src, dst, createpath=0, copydates=1, forcetype=None):
  94     """Copy a file, including finder info, resource fork, etc"""
  95     src = File.pathname(src)
  96     dst = File.pathname(dst)
  97     if createpath:
  98         mkdirs(os.path.split(dst)[0])
  99
 100     ifp = open(src, 'rb')
 101     ofp = open(dst, 'wb')
 102     d = ifp.read(BUFSIZ)
 103     while d:
 104         ofp.write(d)
 105         d = ifp.read(BUFSIZ)
 106     ifp.close()
 107     ofp.close()
 108
 109     ifp = openrf(src, '*rb')
 110     ofp = openrf(dst, '*wb')
 111     d = ifp.read(BUFSIZ)
 112     while d:
 113         ofp.write(d)
 114         d = ifp.read(BUFSIZ)
 115     ifp.close()
 116     ofp.close()
 117
 118     srcfss = File.FSSpec(src)
 119     dstfss = File.FSSpec(dst)
 120     sf = srcfss.FSpGetFInfo()
 121     df = dstfss.FSpGetFInfo()
 122     df.Creator, df.Type = sf.Creator, sf.Type
 123     if forcetype != None:
 124         df.Type = forcetype
 125     df.Flags = (sf.Flags & COPY_FLAGS)
 126     dstfss.FSpSetFInfo(df)
 127     if copydates:
 128         srcfsr = File.FSRef(src)
 129         dstfsr = File.FSRef(dst)
 130         catinfo, _, _, _ = srcfsr.FSGetCatalogInfo(Files.kFSCatInfoAllDates)
 131         dstfsr.FSSetCatalogInfo(Files.kFSCatInfoAllDates, catinfo)
 132     touched(dstfss)
 133
 134 def copytree(src, dst, copydates=1):
 135     """Copy a complete file tree to a new destination"""
 136     if os.path.isdir(src):
 137         mkdirs(dst)
 138         files = os.listdir(src)
 139         for f in files:
 140             copytree(os.path.join(src, f), os.path.join(dst, f), copydates)
 141     else:
 142         copy(src, dst, 1, copydates)