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