1
# Bazaar-NG -- distributed version control
3
# Copyright (C) 2005 by Canonical Ltd
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
# GNU General Public License for more details.
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
import os, types, re, time, types
20
from stat import S_ISREG, S_ISDIR, ST_MODE, ST_SIZE
22
from errors import bailout
24
def make_readonly(filename):
25
"""Make a filename read-only."""
26
# TODO: probably needs to be fixed for windows
27
mod = os.stat(filename).st_mode
29
os.chmod(filename, mod)
32
def make_writable(filename):
33
mod = os.stat(filename).st_mode
35
os.chmod(filename, mod)
38
_QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
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)
49
mode = os.lstat(f)[ST_MODE]
55
bailout("can't handle file kind of %r" % fp)
60
"""True if f is an accessible directory."""
62
return S_ISDIR(os.lstat(f)[ST_MODE])
69
"""True if f is a regular file."""
71
return S_ISREG(os.lstat(f)[ST_MODE])
76
def pumpfile(fromfile, tofile):
77
"""Copy contents of one file to another."""
78
tofile.write(fromfile.read())
82
"""Return a new UUID"""
84
## XXX: Could alternatively read /proc/sys/kernel/random/uuid on
85
## Linux, but we need something portable for other systems;
86
## preferably an implementation in Python.
87
bailout('uuids not allowed!')
88
return chomp(os.popen('uuidgen').readline())
91
if s and (s[-1] == '\n'):
99
## TODO: Maybe read in chunks to handle big files
100
if hasattr(f, 'tell'):
116
"""Return email-style username.
118
Something similar to 'Martin Pool <mbp@sourcefrog.net>'
120
:todo: Check it's reasonably well-formed.
122
:todo: Allow taking it from a dotfile to help people on windows
123
who can't easily set variables.
125
:todo: Cope without pwd module, which is only on unix.
127
e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
135
w = pwd.getpwuid(uid)
136
realname, junk = w.pw_gecos.split(',', 1)
137
return '%s <%s@%s>' % (realname, w.pw_name, socket.getfqdn())
141
import getpass, socket
142
return '<%s@%s>' % (getpass.getuser(), socket.getfqdn())
146
"""Return just the email component of a username."""
147
e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
150
m = re.search(r'[\w+.-]+@[\w+.-]+', e)
152
bailout('%r is not a reasonable email address' % e)
156
import getpass, socket
157
return '%s@%s' % (getpass.getuser(), socket.getfqdn())
162
def compare_files(a, b):
163
"""Returns true if equal in contents"""
164
# TODO: don't read the whole thing in one go.
165
result = a.read() == b.read()
170
def local_time_offset():
174
return -time.timezone
177
def format_date(t, offset=0, timezone='original'):
178
## TODO: Perhaps a global option to use either universal or local time?
179
## Or perhaps just let people set $TZ?
182
assert isinstance(t, float)
184
if timezone == 'utc':
187
elif timezone == 'original':
188
tt = time.gmtime(t - offset)
190
assert timezone == 'local'
191
tt = time.localtime(t)
192
offset = local_time_offset()
194
return (time.strftime("%a %Y-%m-%d %H:%M:%S", tt)
195
+ ' %+03d%02d' % (offset / 3600, (offset / 60) % 60))
198
def compact_date(when):
199
return time.strftime('%Y%m%d%H%M%S', time.gmtime(when))
204
"""Return size of given open file."""
205
return os.fstat(f.fileno())[ST_SIZE]
208
if hasattr(os, 'urandom'): # python 2.4 and later
209
rand_bytes = os.urandom
211
# FIXME: No good on non-Linux
212
_rand_file = file('/dev/urandom', 'rb')
213
rand_bytes = _rand_file.read
216
## TODO: We could later have path objects that remember their list
217
## decomposition (might be too tricksy though.)
220
"""Turn string into list of parts.
226
>>> splitpath('a/./b')
228
>>> splitpath('a/.b')
230
>>> splitpath('a/../b')
231
Traceback (most recent call last):
233
BzrError: ("sorry, '..' not allowed in path", [])
235
assert isinstance(p, types.StringTypes)
236
ps = [f for f in p.split('/') if f != '.']
239
bailout("sorry, %r not allowed in path" % f)
243
assert isinstance(p, list)
245
if (f == '..') or (f is None) or (f == ''):
246
bailout("sorry, %r not allowed in path" % f)
250
def appendpath(p1, p2):
257
def extern_command(cmd, ignore_errors = False):
258
mutter('external command: %s' % `cmd`)
260
if not ignore_errors:
261
bailout('command failed')
added
removed
bzrlib/osutils.py
