← 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-10-06 10:53:12 UTC
  • mto: (1185.13.3)
  • mto: This revision was merged to the branch mainline in revision 1418.
  • Revision ID: mbp@sourcefrog.net-20051006105312-06320dbb986e4bb3
- test that we cannot join weaves with different ancestry

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/osutils.py
1
1
# Bazaar-NG -- distributed version control
2
 
 
 
2
#
3
3
# Copyright (C) 2005 by Canonical Ltd
4
 
 
 
4
#
5
5
# This program is free software; you can redistribute it and/or modify
6
6
# it under the terms of the GNU General Public License as published by
7
7
# the Free Software Foundation; either version 2 of the License, or
8
8
# (at your option) any later version.
9
 
 
 
9
#
10
10
# This program is distributed in the hope that it will be useful,
11
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
13
# GNU General Public License for more details.
14
 
 
 
14
#
15
15
# You should have received a copy of the GNU General Public License
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
20
 
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
 
19
from shutil import copyfile
 
20
from stat import (S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE,
 
21
                  S_ISCHR, S_ISBLK, S_ISFIFO, S_ISSOCK)
 
22
from cStringIO import StringIO
 
23
import errno
 
24
import os
 
25
import re
 
26
import sha
 
27
import sys
 
28
import time
 
29
import types
21
30
 
 
31
import bzrlib
22
32
from bzrlib.errors import BzrError
23
33
from bzrlib.trace import mutter
24
 
import bzrlib
 
34
 
25
35
 
26
36
def make_readonly(filename):
27
37
    """Make a filename read-only."""
28
 
    # TODO: probably needs to be fixed for windows
29
38
    mod = os.stat(filename).st_mode
30
39
    mod = mod & 0777555
31
40
    os.chmod(filename, mod)
48
57
    # TODO: I'm not really sure this is the best format either.x
49
58
    global _QUOTE_RE
50
59
    if _QUOTE_RE == None:
51
 
        _QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
 
60
        _QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/\\_~-])')
52
61
        
53
62
    if _QUOTE_RE.search(f):
54
63
        return '"' + f + '"'
64
73
        return 'directory'
65
74
    elif S_ISLNK(mode):
66
75
        return 'symlink'
 
76
    elif S_ISCHR(mode):
 
77
        return 'chardev'
 
78
    elif S_ISBLK(mode):
 
79
        return 'block'
 
80
    elif S_ISFIFO(mode):
 
81
        return 'fifo'
 
82
    elif S_ISSOCK(mode):
 
83
        return 'socket'
67
84
    else:
68
 
        raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
 
85
        return 'unknown'
69
86
 
70
87
 
71
88
def kind_marker(kind):
78
95
    else:
79
96
        raise BzrError('invalid file kind %r' % kind)
80
97
 
 
98
def lexists(f):
 
99
    try:
 
100
        if hasattr(os, 'lstat'):
 
101
            os.lstat(f)
 
102
        else:
 
103
            os.stat(f)
 
104
        return True
 
105
    except OSError,e:
 
106
        if e.errno == errno.ENOENT:
 
107
            return False;
 
108
        else:
 
109
            raise BzrError("lstat/stat of (%r): %r" % (f, e))
81
110
 
 
111
def normalizepath(f):
 
112
    if hasattr(os.path, 'realpath'):
 
113
        F = os.path.realpath
 
114
    else:
 
115
        F = os.path.abspath
 
116
    [p,e] = os.path.split(f)
 
117
    if e == "" or e == "." or e == "..":
 
118
        return F(f)
 
119
    else:
 
120
        return os.path.join(F(p), e)
 
121
    
82
122
 
83
123
def backup_file(fn):
84
124
    """Copy a file to a backup.
87
127
 
88
128
    If the file is already a backup, it's not copied.
89
129
    """
90
 
    import os
91
130
    if fn[-1] == '~':
92
131
        return
93
132
    bfn = fn + '~'
104
143
    finally:
105
144
        outf.close()
106
145
 
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
 
 
118
 
 
 
146
if os.name == 'nt':
 
147
    import shutil
 
148
    rename = shutil.move
 
149
else:
 
150
    rename = os.rename
119
151
 
120
152
 
121
153
def isdir(f):
126
158
        return False
127
159
 
128
160
 
129
 
 
130
161
def isfile(f):
131
162
    """True if f is a regular file."""
132
163
    try:
134
165
    except OSError:
135
166
        return False
136
167
 
 
168
def islink(f):
 
169
    """True if f is a symlink."""
 
170
    try:
 
171
        return S_ISLNK(os.lstat(f)[ST_MODE])
 
172
    except OSError:
 
173
        return False
137
174
 
138
175
def is_inside(dir, fname):
139
176
    """True if fname is inside dir.
145
182
    The empty string as a dir name is taken as top-of-tree and matches 
146
183
    everything.
147
184
    
148
 
    >>> is_inside('src', 'src/foo.c')
 
185
    >>> is_inside('src', os.path.join('src', 'foo.c'))
149
186
    True
150
187
    >>> is_inside('src', 'srccontrol')
151
188
    False
152
 
    >>> is_inside('src', 'src/a/a/a/foo.c')
 
189
    >>> is_inside('src', os.path.join('src', 'a', 'a', 'a', 'foo.c'))
153
190
    True
154
191
    >>> is_inside('foo.c', 'foo.c')
155
192
    True
165
202
    
166
203
    if dir == '':
167
204
        return True
168
 
    
 
205
 
169
206
    if dir[-1] != os.sep:
170
207
        dir += os.sep
171
 
    
 
208
 
172
209
    return fname.startswith(dir)
173
210
 
174
211
 
186
223
    tofile.write(fromfile.read())
187
224
 
188
225
 
189
 
def uuid():
190
 
    """Return a new UUID"""
191
 
    try:
192
 
        return file('/proc/sys/kernel/random/uuid').readline().rstrip('\n')
193
 
    except IOError:
194
 
        return chomp(os.popen('uuidgen').readline())
195
 
 
196
 
 
197
226
def sha_file(f):
198
 
    import sha
199
227
    if hasattr(f, 'tell'):
200
228
        assert f.tell() == 0
201
229
    s = sha.new()
208
236
    return s.hexdigest()
209
237
 
210
238
 
 
239
 
 
240
def sha_strings(strings):
 
241
    """Return the sha-1 of concatenation of strings"""
 
242
    s = sha.new()
 
243
    map(s.update, strings)
 
244
    return s.hexdigest()
 
245
 
 
246
 
211
247
def sha_string(f):
212
 
    import sha
213
248
    s = sha.new()
214
249
    s.update(f)
215
250
    return s.hexdigest()
216
251
 
217
252
 
218
 
 
219
253
def fingerprint_file(f):
220
 
    import sha
221
254
    s = sha.new()
222
255
    b = f.read()
223
256
    s.update(b)
233
266
    
234
267
    TODO: Global option --config-dir to override this.
235
268
    """
236
 
    return os.path.expanduser("~/.bzr.conf")
 
269
    return os.path.join(os.path.expanduser("~"), ".bzr.conf")
237
270
 
238
271
 
239
272
def _auto_user_id():
343
376
    if e:
344
377
        m = re.search(r'[\w+.-]+@[\w+.-]+', e)
345
378
        if not m:
346
 
            raise BzrError("%r doesn't seem to contain a reasonable email address" % e)
 
379
            raise BzrError("%r doesn't seem to contain "
 
380
                           "a reasonable email address" % e)
347
381
        return m.group(0)
348
382
 
349
383
    return _auto_user_id()[1]
350
 
    
351
384
 
352
385
 
353
386
def compare_files(a, b):
362
395
            return True
363
396
 
364
397
 
365
 
 
366
398
def local_time_offset(t=None):
367
399
    """Return offset of local zone from GMT, either at present or at time t."""
368
400
    # python2.3 localtime() can't take None
479
511
        return os.path.join(p1, p2)
480
512
    
481
513
 
482
 
def extern_command(cmd, ignore_errors = False):
483
 
    mutter('external command: %s' % `cmd`)
484
 
    if os.system(cmd):
485
 
        if not ignore_errors:
486
 
            raise BzrError('command failed')
487
 
 
488
 
 
489
514
def _read_config_value(name):
490
515
    """Read a config value from the file ~/.bzr.conf/<name>
491
516
    Return None if the file does not exist"""
498
523
        raise
499
524
 
500
525
 
 
526
def split_lines(s):
 
527
    """Split s into lines, but without removing the newline characters."""
 
528
    return StringIO(s).readlines()
 
529
 
 
530
 
 
531
def hardlinks_good():
 
532
    return sys.platform not in ('win32', 'cygwin', 'darwin')
 
533
 
 
534
 
 
535
def link_or_copy(src, dest):
 
536
    """Hardlink a file, or copy it if it can't be hardlinked."""
 
537
    if not hardlinks_good():
 
538
        copyfile(src, dest)
 
539
        return
 
540
    try:
 
541
        os.link(src, dest)
 
542
    except (OSError, IOError), e:
 
543
        if e.errno != errno.EXDEV:
 
544
            raise
 
545
        copyfile(src, dest)
 
546
 
 
547
 
 
548
def has_symlinks():
 
549
    if hasattr(os, 'symlink'):
 
550
        return True
 
551
    else:
 
552
        return False