~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/atomicfile.py

  • Committer: Martin Pool
  • Date: 2010-01-29 10:36:23 UTC
  • mto: This revision was merged to the branch mainline in revision 4992.
  • Revision ID: mbp@sourcefrog.net-20100129103623-hywka5hymo5z13jw
Change url to canonical.com or wiki, plus some doc improvements in passing

Show diffs side-by-side

added added

removed removed

Lines of Context:
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
import os
51
51
        global _hostname
52
52
 
53
53
        self._fd = None
54
 
        assert mode in ('wb', 'wt'), \
55
 
            "invalid AtomicFile mode %r" % mode
56
54
 
57
55
        if _hostname is None:
58
 
            _hostname = socket.gethostname()
 
56
            _hostname = osutils.get_host_name()
59
57
 
60
 
        self.tmpfilename = '%s.%d.%s.tmp' % (filename, _pid, _hostname)
 
58
        self.tmpfilename = '%s.%d.%s.%s.tmp' % (filename, _pid, _hostname,
 
59
                                                osutils.rand_chars(10))
61
60
 
62
61
        self.realfilename = filename
63
 
        
 
62
 
64
63
        flags = os.O_EXCL | os.O_CREAT | os.O_WRONLY
65
64
        if mode == 'wb':
66
65
            flags |= osutils.O_BINARY
 
66
        elif mode != 'wt':
 
67
            raise ValueError("invalid AtomicFile mode %r" % mode)
67
68
 
68
69
        if new_mode is not None:
69
70
            local_mode = new_mode
70
71
        else:
71
72
            local_mode = 0666
72
 
        
 
73
 
73
74
        # Use a low level fd operation to avoid chmodding later.
74
75
        # This may not succeed, but it should help most of the time
75
76
        self._fd = os.open(self.tmpfilename, flags, local_mode)