← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/atomicfile.py

  • Committer: Martin Pool
  • Date: 2005-05-11 01:03:22 UTC
  • Revision ID: mbp@sourcefrog.net-20050511010322-54654b917bbce05f
- Notes on library dependencies

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/atomicfile.py
24
24
    Open this as for a regular file, then use commit() to move into
25
25
    place or abort() to cancel.
26
26
 
27
 
    You may wish to wrap this in a codecs.EncodedFile to do unicode
28
 
    encoding.
 
27
    An encoding can be specified; otherwise the default is ascii.
29
28
    """
30
29
 
31
 
    def __init__(self, filename, mode='wb'):
 
30
    def __init__(self, filename, mode='wb', encoding=None):
32
31
        if mode != 'wb' and mode != 'wt':
33
32
            raise ValueError("invalid AtomicFile mode %r" % mode)
34
33
 
35
34
        import os, socket
36
 
        self.tmpfilename = '%s.tmp.%d.%s' % (filename, os.getpid(),
 
35
        self.tmpfilename = '%s.%d.%s.tmp' % (filename, os.getpid(),
37
36
                                             socket.gethostname())
38
37
        self.realfilename = filename
39
38
        
40
39
        self.f = open(self.tmpfilename, mode)
 
40
 
 
41
        if encoding:
 
42
            import codecs
 
43
            self.f = codecs.EncodedFile(self.f, encoding)
 
44
        
41
45
        self.write = self.f.write
42
46
        self.closed = property(self.f.closed)
43
47