blob: 8cebc087a3df090aa56f68f3e81298fc431adbcf [file] [log] [blame]
Guido van Rossum54f22ed2000-02-04 15:10:34 +00001"""Pathname and path-related operations for the Macintosh."""
Guido van Rossum217a5fa1990-12-26 15:40:07 +00002
Guido van Rossum5c1d2291998-03-03 21:49:01 +00003import os
Guido van Rossum217a5fa1990-12-26 15:40:07 +00004from stat import *
5
Skip Montanaro17ab1232001-01-24 06:27:27 +00006__all__ = ["normcase","isabs","join","splitdrive","split","splitext",
7 "basename","dirname","commonprefix","getsize","getmtime",
Martin v. Löwis96a60e42002-12-31 13:11:54 +00008 "getatime","getctime", "islink","exists","isdir","isfile",
Mark Hammond8696ebc2002-10-08 02:44:31 +00009 "walk","expanduser","expandvars","normpath","abspath",
Neal Norwitz61cdac62003-01-03 18:01:57 +000010 "realpath","supports_unicode_filenames"]
Guido van Rossumb5e05e91991-01-01 18:10:40 +000011
Fred Drakeb4e460a2000-09-28 16:25:20 +000012# Normalize the case of a pathname. Dummy in Posix, but <s>.lower() here.
Guido van Rossum599f2ed1992-01-14 18:28:18 +000013
Fred Drakeb4e460a2000-09-28 16:25:20 +000014def normcase(path):
15 return path.lower()
Guido van Rossum599f2ed1992-01-14 18:28:18 +000016
17
Guido van Rossum217a5fa1990-12-26 15:40:07 +000018def isabs(s):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000019 """Return true if a path is absolute.
20 On the Mac, relative paths begin with a colon,
21 but as a special case, paths with no colons at all are also relative.
22 Anything else is absolute (the string up to the first colon is the
23 volume name)."""
24
Fred Drake8152d322000-12-12 23:20:45 +000025 return ':' in s and s[0] != ':'
Guido van Rossum217a5fa1990-12-26 15:40:07 +000026
Guido van Rossumb5e05e91991-01-01 18:10:40 +000027
Barry Warsaw384d2491997-02-18 21:53:25 +000028def join(s, *p):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000029 path = s
30 for t in p:
31 if (not s) or isabs(t):
32 path = t
33 continue
34 if t[:1] == ':':
35 t = t[1:]
36 if ':' not in path:
37 path = ':' + path
Fred Drake8152d322000-12-12 23:20:45 +000038 if path[-1:] != ':':
Guido van Rossum54f22ed2000-02-04 15:10:34 +000039 path = path + ':'
40 path = path + t
41 return path
Guido van Rossum217a5fa1990-12-26 15:40:07 +000042
Guido van Rossumb5e05e91991-01-01 18:10:40 +000043
Guido van Rossumb5e05e91991-01-01 18:10:40 +000044def split(s):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000045 """Split a pathname into two parts: the directory leading up to the final
46 bit, and the basename (the filename, without colons, in that directory).
47 The result (s, t) is such that join(s, t) yields the original argument."""
Guido van Rossumb5e05e91991-01-01 18:10:40 +000048
Guido van Rossum54f22ed2000-02-04 15:10:34 +000049 if ':' not in s: return '', s
50 colon = 0
51 for i in range(len(s)):
Fred Drakeb4e460a2000-09-28 16:25:20 +000052 if s[i] == ':': colon = i + 1
Guido van Rossum54f22ed2000-02-04 15:10:34 +000053 path, file = s[:colon-1], s[colon:]
54 if path and not ':' in path:
55 path = path + ':'
56 return path, file
Guido van Rossumb5e05e91991-01-01 18:10:40 +000057
Guido van Rossuma48bf791996-07-23 02:28:32 +000058
59def splitext(p):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000060 """Split a path into root and extension.
61 The extension is everything starting at the last dot in the last
62 pathname component; the root is everything before that.
63 It is always true that root + ext == p."""
Guido van Rossuma48bf791996-07-23 02:28:32 +000064
Martin v. Löwisde333792002-12-12 20:30:20 +000065 i = p.rfind('.')
66 if i<=p.rfind(':'):
67 return p, ''
68 else:
69 return p[:i], p[i:]
Guido van Rossuma48bf791996-07-23 02:28:32 +000070
Guido van Rossum0ec31261995-08-10 18:09:16 +000071
72def splitdrive(p):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000073 """Split a pathname into a drive specification and the rest of the
74 path. Useful on DOS/Windows/NT; on the Mac, the drive is always
75 empty (don't use the volume name -- it doesn't have the same
76 syntactic and semantic oddities as DOS drive letters, such as there
77 being a separate current directory per drive)."""
78
79 return '', p
Guido van Rossum0ec31261995-08-10 18:09:16 +000080
81
Guido van Rossumc629d341992-11-05 10:43:02 +000082# Short interfaces to split()
83
84def dirname(s): return split(s)[0]
85def basename(s): return split(s)[1]
86
Jack Jansen791f7d42003-01-15 22:45:48 +000087def ismount(s):
88 if not isabs(s):
89 return False
90 components = split(s)
91 return len(components) == 2 and components[1] == ''
Guido van Rossumc629d341992-11-05 10:43:02 +000092
Guido van Rossum217a5fa1990-12-26 15:40:07 +000093def isdir(s):
Guido van Rossum54f22ed2000-02-04 15:10:34 +000094 """Return true if the pathname refers to an existing directory."""
95
96 try:
97 st = os.stat(s)
98 except os.error:
99 return 0
Raymond Hettinger32200ae2002-06-01 19:51:15 +0000100 return S_ISDIR(st.st_mode)
Guido van Rossum217a5fa1990-12-26 15:40:07 +0000101
Guido van Rossumb5e05e91991-01-01 18:10:40 +0000102
Guido van Rossum2bc1f8f1998-07-24 20:49:26 +0000103# Get size, mtime, atime of files.
104
105def getsize(filename):
106 """Return the size of a file, reported by os.stat()."""
Raymond Hettinger32200ae2002-06-01 19:51:15 +0000107 return os.stat(filename).st_size
Guido van Rossum2bc1f8f1998-07-24 20:49:26 +0000108
109def getmtime(filename):
110 """Return the last modification time of a file, reported by os.stat()."""
Raymond Hettinger32200ae2002-06-01 19:51:15 +0000111 return os.stat(filename).st_mtime
Guido van Rossum2bc1f8f1998-07-24 20:49:26 +0000112
113def getatime(filename):
114 """Return the last access time of a file, reported by os.stat()."""
Raymond Hettinger32200ae2002-06-01 19:51:15 +0000115 return os.stat(filename).st_atime
Guido van Rossum2bc1f8f1998-07-24 20:49:26 +0000116
117
Guido van Rossum7e4b2de1995-01-27 02:41:45 +0000118def islink(s):
Jack Jansen992d58b2002-04-22 13:55:43 +0000119 """Return true if the pathname refers to a symbolic link."""
Guido van Rossum7e4b2de1995-01-27 02:41:45 +0000120
Jack Jansen992d58b2002-04-22 13:55:43 +0000121 try:
Anthony Baxterae7639d2002-04-23 02:38:39 +0000122 import macfs
Jack Jansen992d58b2002-04-22 13:55:43 +0000123 return macfs.ResolveAliasFile(s)[2]
124 except:
125 return False
Guido van Rossum7e4b2de1995-01-27 02:41:45 +0000126
Guido van Rossumb5e05e91991-01-01 18:10:40 +0000127
Guido van Rossum217a5fa1990-12-26 15:40:07 +0000128def isfile(s):
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000129 """Return true if the pathname refers to an existing regular file."""
Guido van Rossum217a5fa1990-12-26 15:40:07 +0000130
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000131 try:
132 st = os.stat(s)
133 except os.error:
Guido van Rossum8ca162f2002-04-07 06:36:23 +0000134 return False
Raymond Hettinger32200ae2002-06-01 19:51:15 +0000135 return S_ISREG(st.st_mode)
Guido van Rossumb5e05e91991-01-01 18:10:40 +0000136
Martin v. Löwis96a60e42002-12-31 13:11:54 +0000137def getctime(filename):
138 """Return the creation time of a file, reported by os.stat()."""
139 return os.stat(filename).st_ctime
Guido van Rossumb5e05e91991-01-01 18:10:40 +0000140
Guido van Rossum217a5fa1990-12-26 15:40:07 +0000141def exists(s):
Tim Petersbc0e9102002-04-04 22:55:58 +0000142 """Return True if the pathname refers to an existing file or directory."""
Guido van Rossumc629d341992-11-05 10:43:02 +0000143
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000144 try:
145 st = os.stat(s)
146 except os.error:
Tim Petersbc0e9102002-04-04 22:55:58 +0000147 return False
148 return True
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000149
Jack Jansen03c06ee2000-08-23 09:13:40 +0000150# Return the longest prefix of all list elements.
151
152def commonprefix(m):
153 "Given a list of pathnames, returns the longest common leading component"
154 if not m: return ''
155 prefix = m[0]
156 for item in m:
157 for i in range(len(prefix)):
Fred Drake8152d322000-12-12 23:20:45 +0000158 if prefix[:i+1] != item[:i+1]:
Jack Jansen03c06ee2000-08-23 09:13:40 +0000159 prefix = prefix[:i]
160 if i == 0: return ''
161 break
162 return prefix
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000163
Jack Jansenf4e7d2a1995-12-15 13:23:37 +0000164def expandvars(path):
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000165 """Dummy to retain interface-compatibility with other operating systems."""
166 return path
Jack Jansenf4e7d2a1995-12-15 13:23:37 +0000167
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000168
Jack Jansenf4e7d2a1995-12-15 13:23:37 +0000169def expanduser(path):
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000170 """Dummy to retain interface-compatibility with other operating systems."""
171 return path
Guido van Rossum0ec31261995-08-10 18:09:16 +0000172
Neal Norwitz93cf79f2002-03-31 14:06:41 +0000173class norm_error(Exception):
174 """Path cannot be normalized"""
Guido van Rossumc629d341992-11-05 10:43:02 +0000175
176def normpath(s):
Jack Jansen2fc01092000-08-06 21:18:35 +0000177 """Normalize a pathname. Will return the same result for
178 equivalent paths."""
Jack Jansena68bfe21995-08-07 14:09:27 +0000179
Jack Jansen2fc01092000-08-06 21:18:35 +0000180 if ":" not in s:
181 return ":"+s
182
Fred Drakeb4e460a2000-09-28 16:25:20 +0000183 comps = s.split(":")
Jack Jansen2fc01092000-08-06 21:18:35 +0000184 i = 1
185 while i < len(comps)-1:
186 if comps[i] == "" and comps[i-1] != "":
187 if i > 1:
188 del comps[i-1:i+1]
Fred Drakeb4e460a2000-09-28 16:25:20 +0000189 i = i - 1
Jack Jansen2fc01092000-08-06 21:18:35 +0000190 else:
191 # best way to handle this is to raise an exception
Greg Ward034cbf12001-08-08 20:55:10 +0000192 raise norm_error, 'Cannot use :: immediately after volume name'
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000193 else:
Jack Jansen2fc01092000-08-06 21:18:35 +0000194 i = i + 1
195
Fred Drakeb4e460a2000-09-28 16:25:20 +0000196 s = ":".join(comps)
Jack Jansen2fc01092000-08-06 21:18:35 +0000197
198 # remove trailing ":" except for ":" and "Volume:"
199 if s[-1] == ":" and len(comps) > 2 and s != ":"*len(s):
200 s = s[:-1]
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000201 return s
Guido van Rossum0ec31261995-08-10 18:09:16 +0000202
Jack Jansena68bfe21995-08-07 14:09:27 +0000203
204def walk(top, func, arg):
Tim Peterscf5e6a42001-10-10 04:16:20 +0000205 """Directory tree walk with callback function.
206
207 For each directory in the directory tree rooted at top (including top
208 itself, but excluding '.' and '..'), call func(arg, dirname, fnames).
209 dirname is the name of the directory, and fnames a list of the names of
210 the files and subdirectories in dirname (excluding '.' and '..'). func
211 may modify the fnames list in-place (e.g. via del or slice assignment),
212 and walk will only recurse into the subdirectories whose names remain in
213 fnames; this can be used to implement a filter, or to impose a specific
214 order of visiting. No semantics are defined for, or required of, arg,
215 beyond that arg is always passed to func. It can be used, e.g., to pass
216 a filename pattern, or a mutable object designed to accumulate
217 statistics. Passing None for arg is common."""
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000218
219 try:
220 names = os.listdir(top)
221 except os.error:
222 return
223 func(arg, top, names)
224 for name in names:
225 name = join(top, name)
Jack Jansen992d58b2002-04-22 13:55:43 +0000226 if isdir(name) and not islink(name):
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000227 walk(name, func, arg)
Guido van Rossume294cf61999-01-29 18:05:18 +0000228
229
Guido van Rossume294cf61999-01-29 18:05:18 +0000230def abspath(path):
Guido van Rossum54f22ed2000-02-04 15:10:34 +0000231 """Return an absolute path."""
Guido van Rossume294cf61999-01-29 18:05:18 +0000232 if not isabs(path):
233 path = join(os.getcwd(), path)
234 return normpath(path)
Guido van Rossum83eeef42001-09-17 15:16:09 +0000235
236# realpath is a no-op on systems without islink support
Jack Jansen992d58b2002-04-22 13:55:43 +0000237def realpath(path):
Tim Peters8ac14952002-05-23 15:15:30 +0000238 path = abspath(path)
239 try:
240 import macfs
241 except ImportError:
242 return path
243 if not path:
244 return path
245 components = path.split(':')
246 path = components[0] + ':'
247 for c in components[1:]:
248 path = join(path, c)
249 path = macfs.ResolveAliasFile(path)[0].as_pathname()
250 return path
Mark Hammond8696ebc2002-10-08 02:44:31 +0000251
252supports_unicode_filenames = False