← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: mbp at sourcefrog
  • Date: 2005-04-08 05:39:46 UTC
  • Revision ID: mbp@sourcefrog.net-20050408053946-1cb3415e1f8f58493034a5cf
- import lovely urlgrabber library

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/osutils.py
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
 
19
import os, types, re, time
20
20
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
21
21
 
22
22
from errors import bailout, BzrError
23
23
from trace import mutter
24
 
import bzrlib
25
24
 
26
25
def make_readonly(filename):
27
26
    """Make a filename read-only."""
84
83
 
85
84
def uuid():
86
85
    """Return a new UUID"""
 
86
    
 
87
    ## XXX: Could alternatively read /proc/sys/kernel/random/uuid on
 
88
    ## Linux, but we need something portable for other systems;
 
89
    ## preferably an implementation in Python.
87
90
    try:
88
 
        return file('/proc/sys/kernel/random/uuid').readline().rstrip('\n')
 
91
        return chomp(file('/proc/sys/kernel/random/uuid').readline())
89
92
    except IOError:
90
93
        return chomp(os.popen('uuidgen').readline())
91
94
 
92
95
 
 
96
def chomp(s):
 
97
    if s and (s[-1] == '\n'):
 
98
        return s[:-1]
 
99
    else:
 
100
        return s
 
101
 
 
102
 
93
103
def sha_file(f):
94
104
    import sha
 
105
    ## TODO: Maybe read in chunks to handle big files
95
106
    if hasattr(f, 'tell'):
96
107
        assert f.tell() == 0
97
108
    s = sha.new()
98
 
    BUFSIZE = 128<<10
99
 
    while True:
100
 
        b = f.read(BUFSIZE)
101
 
        if not b:
102
 
            break
103
 
        s.update(b)
 
109
    s.update(f.read())
104
110
    return s.hexdigest()
105
111
 
106
112
 
122
128
            'sha1': s.hexdigest()}
123
129
 
124
130
 
125
 
def config_dir():
126
 
    """Return per-user configuration directory.
127
 
 
128
 
    By default this is ~/.bzr.conf/
129
 
    
130
 
    TODO: Global option --config-dir to override this.
131
 
    """
132
 
    return os.path.expanduser("~/.bzr.conf")
133
 
 
134
 
 
135
 
def _auto_user_id():
136
 
    """Calculate automatic user identification.
137
 
 
138
 
    Returns (realname, email).
139
 
 
140
 
    Only used when none is set in the environment or the id file.
141
 
 
142
 
    This previously used the FQDN as the default domain, but that can
143
 
    be very slow on machines where DNS is broken.  So now we simply
144
 
    use the hostname.
145
 
    """
 
131
 
 
132
def username():
 
133
    """Return email-style username.
 
134
 
 
135
    Something similar to 'Martin Pool <mbp@sourcefrog.net>'
 
136
 
 
137
    :todo: Check it's reasonably well-formed.
 
138
 
 
139
    :todo: Allow taking it from a dotfile to help people on windows
 
140
           who can't easily set variables.
 
141
 
 
142
    :todo: Cope without pwd module, which is only on unix. 
 
143
    """
 
144
    e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
 
145
    if e: return e
 
146
 
146
147
    import socket
147
 
 
148
 
    # XXX: Any good way to get real user name on win32?
149
 
 
 
148
    
150
149
    try:
151
150
        import pwd
152
151
        uid = os.getuid()
153
152
        w = pwd.getpwuid(uid)
154
 
        gecos = w.pw_gecos.decode(bzrlib.user_encoding)
155
 
        username = w.pw_name.decode(bzrlib.user_encoding)
 
153
        gecos = w.pw_gecos
156
154
        comma = gecos.find(',')
157
155
        if comma == -1:
158
156
            realname = gecos
159
157
        else:
160
158
            realname = gecos[:comma]
161
 
        if not realname:
162
 
            realname = username
163
 
 
 
159
        return '%s <%s@%s>' % (realname, w.pw_name, socket.getfqdn())
164
160
    except ImportError:
165
 
        import getpass
166
 
        realname = username = getpass.getuser().decode(bzrlib.user_encoding)
167
 
 
168
 
    return realname, (username + '@' + socket.gethostname())
169
 
 
170
 
 
171
 
def _get_user_id():
172
 
    """Return the full user id from a file or environment variable.
173
 
 
174
 
    TODO: Allow taking this from a file in the branch directory too
175
 
    for per-branch ids."""
176
 
    v = os.environ.get('BZREMAIL')
177
 
    if v:
178
 
        return v.decode(bzrlib.user_encoding)
179
 
    
180
 
    try:
181
 
        return (open(os.path.join(config_dir(), "email"))
182
 
                .read()
183
 
                .decode(bzrlib.user_encoding)
184
 
                .rstrip("\r\n"))
185
 
    except IOError, e:
186
 
        if e.errno != errno.ENOENT:
187
 
            raise e
188
 
 
189
 
    v = os.environ.get('EMAIL')
190
 
    if v:
191
 
        return v.decode(bzrlib.user_encoding)
192
 
    else:    
193
 
        return None
194
 
 
195
 
 
196
 
def username():
197
 
    """Return email-style username.
198
 
 
199
 
    Something similar to 'Martin Pool <mbp@sourcefrog.net>'
200
 
 
201
 
    TODO: Check it's reasonably well-formed.
202
 
    """
203
 
    v = _get_user_id()
204
 
    if v:
205
 
        return v
206
 
    
207
 
    name, email = _auto_user_id()
208
 
    if name:
209
 
        return '%s <%s>' % (name, email)
210
 
    else:
211
 
        return email
 
161
        pass
 
162
 
 
163
    import getpass, socket
 
164
    return '<%s@%s>' % (getpass.getuser(), socket.getfqdn())
212
165
 
213
166
 
214
167
_EMAIL_RE = re.compile(r'[\w+.-]+@[\w+.-]+')
215
168
def user_email():
216
169
    """Return just the email component of a username."""
217
 
    e = _get_user_id()
 
170
    e = os.environ.get('BZREMAIL') or os.environ.get('EMAIL')
218
171
    if e:
219
172
        m = _EMAIL_RE.search(e)
220
173
        if not m:
221
 
            bailout("%r doesn't seem to contain a reasonable email address" % e)
 
174
            bailout('%r is not a reasonable email address' % e)
222
175
        return m.group(0)
223
176
 
224
 
    return _auto_user_id()[1]
 
177
 
 
178
    import getpass, socket
 
179
    return '%s@%s' % (getpass.getuser(), socket.getfqdn())
 
180
 
225
181
    
226
182
 
227
183
 
228
184
def compare_files(a, b):
229
185
    """Returns true if equal in contents"""
 
186
    # TODO: don't read the whole thing in one go.
230
187
    BUFSIZE = 4096
231
188
    while True:
232
189
        ai = a.read(BUFSIZE)
311
268
    BzrError: ("sorry, '..' not allowed in path", [])
312
269
    """
313
270
    assert isinstance(p, types.StringTypes)
314
 
 
315
 
    # split on either delimiter because people might use either on
316
 
    # Windows
317
 
    ps = re.split(r'[\\/]', p)
318
 
 
319
 
    rps = []
 
271
    ps = [f for f in p.split('/') if (f != '.' and f != '')]
320
272
    for f in ps:
321
273
        if f == '..':
322
274
            bailout("sorry, %r not allowed in path" % f)
323
 
        elif (f == '.') or (f == ''):
324
 
            pass
325
 
        else:
326
 
            rps.append(f)
327
 
    return rps
 
275
    return ps
328
276
 
329
277
def joinpath(p):
330
278
    assert isinstance(p, list)
331
279
    for f in p:
332
280
        if (f == '..') or (f == None) or (f == ''):
333
281
            bailout("sorry, %r not allowed in path" % f)
334
 
    return os.path.join(*p)
 
282
    return '/'.join(p)
335
283
 
336
284
 
337
285
def appendpath(p1, p2):
338
286
    if p1 == '':
339
287
        return p2
340
288
    else:
341
 
        return os.path.join(p1, p2)
 
289
        return p1 + '/' + p2
342
290
    
343
291
 
344
292
def extern_command(cmd, ignore_errors = False):