Mac/Lib/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 import macfs
   8 import Res
   9 import os
  10 from MACFS import *
  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 FSSpecType = type(macfs.FSSpec(':'))
  22
  23 BUFSIZ=0x80000          # Copy in 0.5Mb chunks
  24
  25 #
  26 # Not guaranteed to be correct or stay correct (Apple doesn't tell you
  27 # how to do this), but it seems to work.
  28 #
  29 def mkalias(src, dst, relative=None):
  30         """Create a finder alias"""
  31         srcfss = macfs.FSSpec(src)
  32         dstfss = macfs.FSSpec(dst)
  33         if relative:
  34                 relativefss = macfs.FSSpec(relative)
  35                 # ik mag er geen None in stoppen :-(
  36                 alias = srcfss.NewAlias(relativefss)
  37         else:
  38                 alias = srcfss.NewAlias()
  39         srcfinfo = srcfss.GetFInfo()
  40
  41         Res.FSpCreateResFile(dstfss, srcfinfo.Creator, srcfinfo.Type, -1)
  42         h = Res.FSpOpenResFile(dstfss, 3)
  43         resource = Res.Resource(alias.data)
  44         resource.AddResource('alis', 0, '')
  45         Res.CloseResFile(h)
  46
  47         dstfinfo = dstfss.GetFInfo()
  48         dstfinfo.Flags = dstfinfo.Flags|0x8000    # Alias flag
  49         dstfss.SetFInfo(dstfinfo)
  50
  51 def mkdirs(dst):
  52         """Make directories leading to 'dst' if they don't exist yet"""
  53         if dst == '' or os.path.exists(dst):
  54                 return
  55         head, tail = os.path.split(dst)
  56         if not ':' in head:
  57                 head = head + ':'
  58         mkdirs(head)
  59         os.mkdir(dst, 0777)
  60
  61 def touched(dst):
  62         """Tell the finder a file has changed"""
  63         file_fss = macfs.FSSpec(dst)
  64         vRefNum, dirID, name = file_fss.as_tuple()
  65         dir_fss = macfs.FSSpec((vRefNum, dirID, ''))
  66         crdate, moddate, bkdate = dir_fss.GetDates()
  67         now = time.time()
  68         if now == moddate:
  69                 now = now + 1
  70         dir_fss.SetDates(crdate, now, bkdate)
  71
  72 def touched_ae(dst):
  73         """Tell the finder a file has changed"""
  74         import Finder
  75         f = Finder.Finder()
  76         file_fss = macfs.FSSpec(dst)
  77         vRefNum, dirID, name = file_fss.as_tuple()
  78         dir_fss = macfs.FSSpec((vRefNum, dirID, ''))
  79         f.update(dir_fss)
  80
  81 def copy(src, dst, createpath=0, copydates=1, forcetype=None):
  82         """Copy a file, including finder info, resource fork, etc"""
  83         if createpath:
  84                 mkdirs(os.path.split(dst)[0])
  85         srcfss = macfs.FSSpec(src)
  86         dstfss = macfs.FSSpec(dst)
  87
  88         ifp = open(srcfss.as_pathname(), 'rb')
  89         ofp = open(dstfss.as_pathname(), 'wb')
  90         d = ifp.read(BUFSIZ)
  91         while d:
  92                 ofp.write(d)
  93                 d = ifp.read(BUFSIZ)
  94         ifp.close()
  95         ofp.close()
  96
  97         ifp = openrf(srcfss.as_pathname(), '*rb')
  98         ofp = openrf(dstfss.as_pathname(), '*wb')
  99         d = ifp.read(BUFSIZ)
 100         while d:
 101                 ofp.write(d)
 102                 d = ifp.read(BUFSIZ)
 103         ifp.close()
 104         ofp.close()
 105
 106         sf = srcfss.GetFInfo()
 107         df = dstfss.GetFInfo()
 108         df.Creator, df.Type = sf.Creator, sf.Type
 109         if forcetype != None:
 110                 df.Type = forcetype
 111         df.Flags = (sf.Flags & (kIsStationary|kNameLocked|kHasBundle|kIsInvisible|kIsAlias))
 112         dstfss.SetFInfo(df)
 113         if copydates:
 114                 crdate, mddate, bkdate = srcfss.GetDates()
 115                 dstfss.SetDates(crdate, mddate, bkdate)
 116         touched(dstfss)
 117
 118 def copytree(src, dst, copydates=1):
 119         """Copy a complete file tree to a new destination"""
 120         if os.path.isdir(src):
 121                 mkdirs(dst)
 122                 files = os.listdir(src)
 123                 for f in files:
 124                         copytree(os.path.join(src, f), os.path.join(dst, f), copydates)
 125         else:
 126                 copy(src, dst, 1, copydates)