← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Martin Pool
  • Date: 2005-05-11 06:20:05 UTC
  • Revision ID: mbp@sourcefrog.net-20050511062005-297af3451635dae0
- Don't lose first line of command help!

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/osutils.py
19
19
import os, types, re, time, errno, sys
20
20
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
21
21
 
22
 
from bzrlib.errors import BzrError
23
 
from bzrlib.trace import mutter
 
22
from errors import bailout, BzrError
 
23
from trace import mutter
24
24
import bzrlib
25
25
 
26
26
def make_readonly(filename):
37
37
    os.chmod(filename, mod)
38
38
 
39
39
 
40
 
_QUOTE_RE = None
41
 
 
42
 
 
 
40
_QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
43
41
def quotefn(f):
44
 
    """Return a quoted filename filename
45
 
 
46
 
    This previously used backslash quoting, but that works poorly on
47
 
    Windows."""
48
 
    # TODO: I'm not really sure this is the best format either.x
49
 
    global _QUOTE_RE
50
 
    if _QUOTE_RE == None:
51
 
        _QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
52
 
        
53
 
    if _QUOTE_RE.search(f):
54
 
        return '"' + f + '"'
55
 
    else:
56
 
        return f
 
42
    """Return shell-quoted filename"""
 
43
    ## We could be a bit more terse by using double-quotes etc
 
44
    f = _QUOTE_RE.sub(r'\\\1', f)
 
45
    if f[0] == '~':
 
46
        f[0:1] = r'\~' 
 
47
    return f
57
48
 
58
49
 
59
50
def file_kind(f):
65
56
    elif S_ISLNK(mode):
66
57
        return 'symlink'
67
58
    else:
68
 
        raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
69
 
 
70
 
 
71
 
def kind_marker(kind):
72
 
    if kind == 'file':
73
 
        return ''
74
 
    elif kind == 'directory':
75
 
        return '/'
76
 
    elif kind == 'symlink':
77
 
        return '@'
78
 
    else:
79
 
        raise BzrError('invalid file kind %r' % kind)
80
 
 
81
 
 
82
 
 
83
 
def backup_file(fn):
84
 
    """Copy a file to a backup.
85
 
 
86
 
    Backups are named in GNU-style, with a ~ suffix.
87
 
 
88
 
    If the file is already a backup, it's not copied.
89
 
    """
90
 
    import os
91
 
    if fn[-1] == '~':
92
 
        return
93
 
    bfn = fn + '~'
94
 
 
95
 
    inf = file(fn, 'rb')
96
 
    try:
97
 
        content = inf.read()
98
 
    finally:
99
 
        inf.close()
100
 
    
101
 
    outf = file(bfn, 'wb')
102
 
    try:
103
 
        outf.write(content)
104
 
    finally:
105
 
        outf.close()
106
 
 
107
 
def rename(path_from, path_to):
108
 
    """Basically the same as os.rename() just special for win32"""
109
 
    if sys.platform == 'win32':
110
 
        try:
111
 
            os.remove(path_to)
112
 
        except OSError, e:
113
 
            if e.errno != e.ENOENT:
114
 
                raise
115
 
    os.rename(path_from, path_to)
116
 
 
117
 
 
 
59
        raise BzrError("can't handle file kind with mode %o of %r" % (mode, f)) 
118
60
 
119
61
 
120
62
 
135
77
        return False
136
78
 
137
79
 
138
 
def is_inside(dir, fname):
139
 
    """True if fname is inside dir.
140
 
    
141
 
    The parameters should typically be passed to os.path.normpath first, so
142
 
    that . and .. and repeated slashes are eliminated, and the separators
143
 
    are canonical for the platform.
144
 
    
145
 
    The empty string as a dir name is taken as top-of-tree and matches 
146
 
    everything.
147
 
    
148
 
    >>> is_inside('src', 'src/foo.c')
149
 
    True
150
 
    >>> is_inside('src', 'srccontrol')
151
 
    False
152
 
    >>> is_inside('src', 'src/a/a/a/foo.c')
153
 
    True
154
 
    >>> is_inside('foo.c', 'foo.c')
155
 
    True
156
 
    >>> is_inside('foo.c', '')
157
 
    False
158
 
    >>> is_inside('', 'foo.c')
159
 
    True
160
 
    """
161
 
    # XXX: Most callers of this can actually do something smarter by 
162
 
    # looking at the inventory
163
 
    if dir == fname:
164
 
        return True
165
 
    
166
 
    if dir == '':
167
 
        return True
168
 
    
169
 
    if dir[-1] != os.sep:
170
 
        dir += os.sep
171
 
    
172
 
    return fname.startswith(dir)
173
 
 
174
 
 
175
 
def is_inside_any(dir_list, fname):
176
 
    """True if fname is inside any of given dirs."""
177
 
    for dirname in dir_list:
178
 
        if is_inside(dirname, fname):
179
 
            return True
180
 
    else:
181
 
        return False
182
 
 
183
 
 
184
80
def pumpfile(fromfile, tofile):
185
81
    """Copy contents of one file to another."""
186
82
    tofile.write(fromfile.read())
272
168
    return realname, (username + '@' + socket.gethostname())
273
169
 
274
170
 
275
 
def _get_user_id(branch):
 
171
def _get_user_id():
276
172
    """Return the full user id from a file or environment variable.
277
173
 
278
 
    e.g. "John Hacker <jhacker@foo.org>"
279
 
 
280
 
    branch
281
 
        A branch to use for a per-branch configuration, or None.
282
 
 
283
 
    The following are searched in order:
284
 
 
285
 
    1. $BZREMAIL
286
 
    2. .bzr/email for this branch.
287
 
    3. ~/.bzr.conf/email
288
 
    4. $EMAIL
289
 
    """
 
174
    TODO: Allow taking this from a file in the branch directory too
 
175
    for per-branch ids."""
290
176
    v = os.environ.get('BZREMAIL')
291
177
    if v:
292
178
        return v.decode(bzrlib.user_encoding)
293
 
 
294
 
    if branch:
295
 
        try:
296
 
            return (branch.controlfile("email", "r") 
297
 
                    .read()
298
 
                    .decode(bzrlib.user_encoding)
299
 
                    .rstrip("\r\n"))
300
 
        except IOError, e:
301
 
            if e.errno != errno.ENOENT:
302
 
                raise
303
 
        except BzrError, e:
304
 
            pass
305
179
    
306
180
    try:
307
181
        return (open(os.path.join(config_dir(), "email"))
319
193
        return None
320
194
 
321
195
 
322
 
def username(branch):
 
196
def username():
323
197
    """Return email-style username.
324
198
 
325
199
    Something similar to 'Martin Pool <mbp@sourcefrog.net>'
326
200
 
327
201
    TODO: Check it's reasonably well-formed.
328
202
    """
329
 
    v = _get_user_id(branch)
 
203
    v = _get_user_id()
330
204
    if v:
331
205
        return v
332
206
    
337
211
        return email
338
212
 
339
213
 
340
 
def user_email(branch):
 
214
_EMAIL_RE = re.compile(r'[\w+.-]+@[\w+.-]+')
 
215
def user_email():
341
216
    """Return just the email component of a username."""
342
 
    e = _get_user_id(branch)
 
217
    e = _get_user_id()
343
218
    if e:
344
 
        m = re.search(r'[\w+.-]+@[\w+.-]+', e)
 
219
        m = _EMAIL_RE.search(e)
345
220
        if not m:
346
 
            raise BzrError("%r doesn't seem to contain a reasonable email address" % e)
 
221
            bailout("%r doesn't seem to contain a reasonable email address" % e)
347
222
        return m.group(0)
348
223
 
349
224
    return _auto_user_id()[1]
391
266
        tt = time.localtime(t)
392
267
        offset = local_time_offset(t)
393
268
    else:
394
 
        raise BzrError("unsupported timezone format %r" % timezone,
395
 
                       ['options are "utc", "original", "local"'])
 
269
        bailout("unsupported timezone format %r",
 
270
                ['options are "utc", "original", "local"'])
396
271
 
397
272
    return (time.strftime("%a %Y-%m-%d %H:%M:%S", tt)
398
273
            + ' %+03d%02d' % (offset / 3600, (offset / 60) % 60))
440
315
    >>> splitpath('a/../b')
441
316
    Traceback (most recent call last):
442
317
    ...
443
 
    BzrError: sorry, '..' not allowed in path
 
318
    BzrError: ("sorry, '..' not allowed in path", [])
444
319
    """
445
320
    assert isinstance(p, types.StringTypes)
446
321
 
451
326
    rps = []
452
327
    for f in ps:
453
328
        if f == '..':
454
 
            raise BzrError("sorry, %r not allowed in path" % f)
 
329
            bailout("sorry, %r not allowed in path" % f)
455
330
        elif (f == '.') or (f == ''):
456
331
            pass
457
332
        else:
462
337
    assert isinstance(p, list)
463
338
    for f in p:
464
339
        if (f == '..') or (f == None) or (f == ''):
465
 
            raise BzrError("sorry, %r not allowed in path" % f)
 
340
            bailout("sorry, %r not allowed in path" % f)
466
341
    return os.path.join(*p)
467
342
 
468
343
 
477
352
    mutter('external command: %s' % `cmd`)
478
353
    if os.system(cmd):
479
354
        if not ignore_errors:
480
 
            raise BzrError('command failed')
481
 
 
482
 
 
483
 
def _read_config_value(name):
484
 
    """Read a config value from the file ~/.bzr.conf/<name>
485
 
    Return None if the file does not exist"""
486
 
    try:
487
 
        f = file(os.path.join(config_dir(), name), "r")
488
 
        return f.read().decode(bzrlib.user_encoding).rstrip("\r\n")
489
 
    except IOError, e:
490
 
        if e.errno == errno.ENOENT:
491
 
            return None
492
 
        raise
493
 
 
 
355
            bailout('command failed')
494
356