~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-22 01:17:38 UTC
  • Revision ID: mbp@sourcefrog.net-20050322011738-5e778270d06836bb
ignore generated changelog

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
# along with this program; if not, write to the Free Software
17
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
18
 
19
 
import os, types, re, time, errno, sys
 
19
import os, types, re, time, types
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
24
 
import bzrlib
 
22
from errors import bailout
25
23
 
26
24
def make_readonly(filename):
27
25
    """Make a filename read-only."""
38
36
 
39
37
 
40
38
_QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
41
 
 
42
 
_SLASH_RE = re.compile(r'[\\/]+')
43
 
 
44
39
def quotefn(f):
45
 
    """Return a quoted filename filename
46
 
 
47
 
    This previously used backslash quoting, but that works poorly on
48
 
    Windows."""
49
 
    # TODO: I'm not really sure this is the best format either.x
50
 
    if _QUOTE_RE.search(f):
51
 
        return '"' + f + '"'
52
 
    else:
53
 
        return f
 
40
    """Return shell-quoted filename"""
 
41
    ## We could be a bit more terse by using double-quotes etc
 
42
    f = _QUOTE_RE.sub(r'\\\1', f)
 
43
    if f[0] == '~':
 
44
        f[0:1] = r'\~' 
 
45
    return f
54
46
 
55
47
 
56
48
def file_kind(f):
62
54
    elif S_ISLNK(mode):
63
55
        return 'symlink'
64
56
    else:
65
 
        raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
66
 
 
67
 
 
68
 
def kind_marker(kind):
69
 
    if kind == 'file':
70
 
        return ''
71
 
    elif kind == 'directory':
72
 
        return '/'
73
 
    elif kind == 'symlink':
74
 
        return '@'
75
 
    else:
76
 
        raise BzrError('invalid file kind %r' % kind)
77
 
 
78
 
 
79
 
 
80
 
def backup_file(fn):
81
 
    """Copy a file to a backup.
82
 
 
83
 
    Backups are named in GNU-style, with a ~ suffix.
84
 
 
85
 
    If the file is already a backup, it's not copied.
86
 
    """
87
 
    import os
88
 
    if fn[-1] == '~':
89
 
        return
90
 
    bfn = fn + '~'
91
 
 
92
 
    inf = file(fn, 'rb')
93
 
    try:
94
 
        content = inf.read()
95
 
    finally:
96
 
        inf.close()
97
 
    
98
 
    outf = file(bfn, 'wb')
99
 
    try:
100
 
        outf.write(content)
101
 
    finally:
102
 
        outf.close()
103
 
 
104
 
def rename(path_from, path_to):
105
 
    """Basically the same as os.rename() just special for win32"""
106
 
    if sys.platform == 'win32':
107
 
        try:
108
 
            os.remove(path_to)
109
 
        except OSError, e:
110
 
            if e.errno != e.ENOENT:
111
 
                raise
112
 
    os.rename(path_from, path_to)
113
 
 
114
 
 
 
57
        bailout("can't handle file kind with mode %o of %r" % (mode, f)) 
115
58
 
116
59
 
117
60
 
132
75
        return False
133
76
 
134
77
 
135
 
def is_inside(dir, fname):
136
 
    """True if fname is inside dir.
137
 
    
138
 
    The parameters should typically be passed to os.path.normpath first, so
139
 
    that . and .. and repeated slashes are eliminated, and the separators
140
 
    are canonical for the platform.
141
 
    
142
 
    The empty string as a dir name is taken as top-of-tree and matches 
143
 
    everything.
144
 
    
145
 
    >>> is_inside('src', 'src/foo.c')
146
 
    True
147
 
    >>> is_inside('src', 'srccontrol')
148
 
    False
149
 
    >>> is_inside('src', 'src/a/a/a/foo.c')
150
 
    True
151
 
    >>> is_inside('foo.c', 'foo.c')
152
 
    True
153
 
    >>> is_inside('foo.c', '')
154
 
    False
155
 
    >>> is_inside('', 'foo.c')
156
 
    True
157
 
    """
158
 
    # XXX: Most callers of this can actually do something smarter by 
159
 
    # looking at the inventory
160
 
    if dir == fname:
161
 
        return True
162
 
    
163
 
    if dir == '':
164
 
        return True
165
 
    
166
 
    if dir[-1] != os.sep:
167
 
        dir += os.sep
168
 
    
169
 
    return fname.startswith(dir)
170
 
 
171
 
 
172
 
def is_inside_any(dir_list, fname):
173
 
    """True if fname is inside any of given dirs."""
174
 
    for dirname in dir_list:
175
 
        if is_inside(dirname, fname):
176
 
            return True
177
 
    else:
178
 
        return False
179
 
 
180
 
 
181
78
def pumpfile(fromfile, tofile):
182
79
    """Copy contents of one file to another."""
183
80
    tofile.write(fromfile.read())
185
82
 
186
83
def uuid():
187
84
    """Return a new UUID"""
188
 
    try:
189
 
        return file('/proc/sys/kernel/random/uuid').readline().rstrip('\n')
190
 
    except IOError:
191
 
        return chomp(os.popen('uuidgen').readline())
 
85
    
 
86
    ## XXX: Could alternatively read /proc/sys/kernel/random/uuid on
 
87
    ## Linux, but we need something portable for other systems;
 
88
    ## preferably an implementation in Python.
 
89
    bailout('uuids not allowed!')
 
90
    return chomp(os.popen('uuidgen').readline())
 
91
 
 
92
def chomp(s):
 
93
    if s and (s[-1] == '\n'):
 
94
        return s[:-1]
 
95
    else:
 
96
        return s
192
97
 
193
98
 
194
99
def sha_file(f):
195
100
    import sha
 
101
    ## TODO: Maybe read in chunks to handle big files
196
102
    if hasattr(f, 'tell'):
197
103
        assert f.tell() == 0
198
104
    s = sha.new()
199
 
    BUFSIZE = 128<<10
200
 
    while True:
201
 
        b = f.read(BUFSIZE)
202
 
        if not b:
203
 
            break
204
 
        s.update(b)
 
105
    s.update(f.read())
205
106
    return s.hexdigest()
206
107
 
207
108
 
213
114
 
214
115
 
215
116
 
216
 
def fingerprint_file(f):
217
 
    import sha
218
 
    s = sha.new()
219
 
    b = f.read()
220
 
    s.update(b)
221
 
    size = len(b)
222
 
    return {'size': size,
223
 
            'sha1': s.hexdigest()}
224
 
 
225
 
 
226
 
def config_dir():
227
 
    """Return per-user configuration directory.
228
 
 
229
 
    By default this is ~/.bzr.conf/
230
 
    
231
 
    TODO: Global option --config-dir to override this.
232
 
    """
233
 
    return os.path.expanduser("~/.bzr.conf")
234
 
 
235
 
 
236
 
def _auto_user_id():
237
 
    """Calculate automatic user identification.
238
 
 
239
 
    Returns (realname, email).
240
 
 
241
 
    Only used when none is set in the environment or the id file.
242
 
 
243
 
    This previously used the FQDN as the default domain, but that can
244
 
    be very slow on machines where DNS is broken.  So now we simply
245
 
    use the hostname.
246
 
    """
 
117
def username():
 
118
    """Return email-style username.
 
119
 
 
120
    Something similar to 'Martin Pool <mbp@sourcefrog.net>'
 
121
 
 
122
    :todo: Check it's reasonably well-formed.
 
123
 
 
124
    :todo: Allow taking it from a dotfile to help people on windows
 
125
           who can't easily set variables.
 
126
 
 
127
    :todo: Cope without pwd module, which is only on unix. 
 
128
    """
 
129
    e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
 
130
    if e: return e
 
131
 
247
132
    import socket
248
 
 
249
 
    # XXX: Any good way to get real user name on win32?
250
 
 
 
133
    
251
134
    try:
252
135
        import pwd
253
136
        uid = os.getuid()
254
137
        w = pwd.getpwuid(uid)
255
 
        gecos = w.pw_gecos.decode(bzrlib.user_encoding)
256
 
        username = w.pw_name.decode(bzrlib.user_encoding)
 
138
        gecos = w.pw_gecos
257
139
        comma = gecos.find(',')
258
140
        if comma == -1:
259
141
            realname = gecos
260
142
        else:
261
143
            realname = gecos[:comma]
262
 
        if not realname:
263
 
            realname = username
264
 
 
 
144
        return '%s <%s@%s>' % (realname, w.pw_name, socket.getfqdn())
265
145
    except ImportError:
266
 
        import getpass
267
 
        realname = username = getpass.getuser().decode(bzrlib.user_encoding)
268
 
 
269
 
    return realname, (username + '@' + socket.gethostname())
270
 
 
271
 
 
272
 
def _get_user_id():
273
 
    """Return the full user id from a file or environment variable.
274
 
 
275
 
    TODO: Allow taking this from a file in the branch directory too
276
 
    for per-branch ids."""
277
 
    v = os.environ.get('BZREMAIL')
278
 
    if v:
279
 
        return v.decode(bzrlib.user_encoding)
280
 
    
281
 
    try:
282
 
        return (open(os.path.join(config_dir(), "email"))
283
 
                .read()
284
 
                .decode(bzrlib.user_encoding)
285
 
                .rstrip("\r\n"))
286
 
    except IOError, e:
287
 
        if e.errno != errno.ENOENT:
288
 
            raise e
289
 
 
290
 
    v = os.environ.get('EMAIL')
291
 
    if v:
292
 
        return v.decode(bzrlib.user_encoding)
293
 
    else:    
294
 
        return None
295
 
 
296
 
 
297
 
def username():
298
 
    """Return email-style username.
299
 
 
300
 
    Something similar to 'Martin Pool <mbp@sourcefrog.net>'
301
 
 
302
 
    TODO: Check it's reasonably well-formed.
303
 
    """
304
 
    v = _get_user_id()
305
 
    if v:
306
 
        return v
307
 
    
308
 
    name, email = _auto_user_id()
309
 
    if name:
310
 
        return '%s <%s>' % (name, email)
311
 
    else:
312
 
        return email
313
 
 
314
 
 
315
 
_EMAIL_RE = re.compile(r'[\w+.-]+@[\w+.-]+')
 
146
        pass
 
147
 
 
148
    import getpass, socket
 
149
    return '<%s@%s>' % (getpass.getuser(), socket.getfqdn())
 
150
 
 
151
 
316
152
def user_email():
317
153
    """Return just the email component of a username."""
318
 
    e = _get_user_id()
 
154
    e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
319
155
    if e:
320
 
        m = _EMAIL_RE.search(e)
 
156
        import re
 
157
        m = re.search(r'[\w+.-]+@[\w+.-]+', e)
321
158
        if not m:
322
 
            raise BzrError("%r doesn't seem to contain a reasonable email address" % e)
 
159
            bailout('%r is not a reasonable email address' % e)
323
160
        return m.group(0)
324
161
 
325
 
    return _auto_user_id()[1]
 
162
 
 
163
    import getpass, socket
 
164
    return '%s@%s' % (getpass.getuser(), socket.getfqdn())
 
165
 
326
166
    
327
167
 
328
168
 
329
169
def compare_files(a, b):
330
170
    """Returns true if equal in contents"""
331
 
    BUFSIZE = 4096
332
 
    while True:
333
 
        ai = a.read(BUFSIZE)
334
 
        bi = b.read(BUFSIZE)
335
 
        if ai != bi:
336
 
            return False
337
 
        if ai == '':
338
 
            return True
 
171
    # TODO: don't read the whole thing in one go.
 
172
    result = a.read() == b.read()
 
173
    return result
339
174
 
340
175
 
341
176
 
342
177
def local_time_offset(t=None):
343
178
    """Return offset of local zone from GMT, either at present or at time t."""
344
 
    # python2.3 localtime() can't take None
345
 
    if t == None:
346
 
        t = time.time()
347
 
        
348
179
    if time.localtime(t).tm_isdst and time.daylight:
349
180
        return -time.altzone
350
181
    else:
354
185
def format_date(t, offset=0, timezone='original'):
355
186
    ## TODO: Perhaps a global option to use either universal or local time?
356
187
    ## Or perhaps just let people set $TZ?
 
188
    import time
 
189
    
357
190
    assert isinstance(t, float)
358
191
    
359
192
    if timezone == 'utc':
367
200
        tt = time.localtime(t)
368
201
        offset = local_time_offset(t)
369
202
    else:
370
 
        raise BzrError("unsupported timezone format %r" % timezone,
371
 
                       ['options are "utc", "original", "local"'])
 
203
        bailout("unsupported timezone format %r",
 
204
                ['options are "utc", "original", "local"'])
372
205
 
373
206
    return (time.strftime("%a %Y-%m-%d %H:%M:%S", tt)
374
207
            + ' %+03d%02d' % (offset / 3600, (offset / 60) % 60))
386
219
 
387
220
if hasattr(os, 'urandom'): # python 2.4 and later
388
221
    rand_bytes = os.urandom
389
 
elif sys.platform == 'linux2':
390
 
    rand_bytes = file('/dev/urandom', 'rb').read
391
222
else:
392
 
    # not well seeded, but better than nothing
393
 
    def rand_bytes(n):
394
 
        import random
395
 
        s = ''
396
 
        while n:
397
 
            s += chr(random.randint(0, 255))
398
 
            n -= 1
399
 
        return s
 
223
    # FIXME: No good on non-Linux
 
224
    _rand_file = file('/dev/urandom', 'rb')
 
225
    rand_bytes = _rand_file.read
400
226
 
401
227
 
402
228
## TODO: We could later have path objects that remember their list
416
242
    >>> splitpath('a/../b')
417
243
    Traceback (most recent call last):
418
244
    ...
419
 
    BzrError: sorry, '..' not allowed in path
 
245
    BzrError: ("sorry, '..' not allowed in path", [])
420
246
    """
421
247
    assert isinstance(p, types.StringTypes)
422
 
 
423
 
    # split on either delimiter because people might use either on
424
 
    # Windows
425
 
    ps = re.split(r'[\\/]', p)
426
 
 
427
 
    rps = []
 
248
    ps = [f for f in p.split('/') if f != '.']
428
249
    for f in ps:
429
250
        if f == '..':
430
 
            raise BzrError("sorry, %r not allowed in path" % f)
431
 
        elif (f == '.') or (f == ''):
432
 
            pass
433
 
        else:
434
 
            rps.append(f)
435
 
    return rps
 
251
            bailout("sorry, %r not allowed in path" % f)
 
252
    return ps
436
253
 
437
254
def joinpath(p):
438
255
    assert isinstance(p, list)
439
256
    for f in p:
440
 
        if (f == '..') or (f == None) or (f == ''):
441
 
            raise BzrError("sorry, %r not allowed in path" % f)
442
 
    return os.path.join(*p)
 
257
        if (f == '..') or (f is None) or (f == ''):
 
258
            bailout("sorry, %r not allowed in path" % f)
 
259
    return '/'.join(p)
443
260
 
444
261
 
445
262
def appendpath(p1, p2):
446
263
    if p1 == '':
447
264
        return p2
448
265
    else:
449
 
        return os.path.join(p1, p2)
 
266
        return p1 + '/' + p2
450
267
    
451
268
 
452
269
def extern_command(cmd, ignore_errors = False):
453
270
    mutter('external command: %s' % `cmd`)
454
271
    if os.system(cmd):
455
272
        if not ignore_errors:
456
 
            raise BzrError('command failed')
457
 
 
458
 
 
459
 
def _read_config_value(name):
460
 
    """Read a config value from the file ~/.bzr.conf/<name>
461
 
    Return None if the file does not exist"""
462
 
    try:
463
 
        f = file(os.path.join(config_dir(), name), "r")
464
 
        return f.read().decode(bzrlib.user_encoding).rstrip("\r\n")
465
 
    except IOError, e:
466
 
        if e.errno == errno.ENOENT:
467
 
            return None
468
 
        raise
469
 
 
470
 
 
471
 
def _get_editor():
472
 
    """Return a sequence of possible editor binaries for the current platform"""
473
 
    e = _read_config_value("editor")
474
 
    if e is not None:
475
 
        yield e
476
 
        
477
 
    if os.name == "windows":
478
 
        yield "notepad.exe"
479
 
    elif os.name == "posix":
480
 
        try:
481
 
            yield os.environ["EDITOR"]
482
 
        except KeyError:
483
 
            yield "/usr/bin/vi"
484
 
 
485
 
 
486
 
def _run_editor(filename):
487
 
    """Try to execute an editor to edit the commit message. Returns True on success,
488
 
    False on failure"""
489
 
    for e in _get_editor():
490
 
        x = os.spawnvp(os.P_WAIT, e, (e, filename))
491
 
        if x == 0:
492
 
            return True
493
 
        elif x == 127:
494
 
            continue
495
 
        else:
496
 
            break
497
 
    raise BzrError("Could not start any editor. Please specify $EDITOR or use ~/.bzr.conf/editor")
498
 
    return False
499
 
                          
500
 
 
501
 
def get_text_message(infotext, ignoreline = "default"):
502
 
    import tempfile
503
 
    
504
 
    if ignoreline == "default":
505
 
        ignoreline = "-- This line and the following will be ignored --"
506
 
        
507
 
    try:
508
 
        tmp_fileno, msgfilename = tempfile.mkstemp()
509
 
        msgfile = os.close(tmp_fileno)
510
 
        if infotext is not None and infotext != "":
511
 
            hasinfo = True
512
 
            msgfile = file(msgfilename, "w")
513
 
            msgfile.write("\n\n%s\n\n%s" % (ignoreline, infotext))
514
 
            msgfile.close()
515
 
        else:
516
 
            hasinfo = False
517
 
 
518
 
        if not _run_editor(msgfilename):
519
 
            return None
520
 
        
521
 
        started = False
522
 
        msg = []
523
 
        lastline, nlines = 0, 0
524
 
        for line in file(msgfilename, "r"):
525
 
            stripped_line = line.strip()
526
 
            # strip empty line before the log message starts
527
 
            if not started:
528
 
                if stripped_line != "":
529
 
                    started = True
530
 
                else:
531
 
                    continue
532
 
            # check for the ignore line only if there
533
 
            # is additional information at the end
534
 
            if hasinfo and stripped_line == ignoreline:
535
 
                break
536
 
            nlines += 1
537
 
            # keep track of the last line that had some content
538
 
            if stripped_line != "":
539
 
                lastline = nlines
540
 
            msg.append(line)
541
 
            
542
 
        if len(msg) == 0:
543
 
            return None
544
 
        # delete empty lines at the end
545
 
        del msg[lastline:]
546
 
        # add a newline at the end, if needed
547
 
        if not msg[-1].endswith("\n"):
548
 
            return "%s%s" % ("".join(msg), "\n")
549
 
        else:
550
 
            return "".join(msg)
551
 
    finally:
552
 
        # delete the msg file in any case
553
 
        try: os.unlink(msgfilename)
554
 
        except IOError: pass
 
273
            bailout('command failed')
 
274