mercurial-scm/hg: mercurial/pure/osutil.py annotate

annotate mercurial/pure/osutil.py @ 19465:004f965630d9

osutil: consider WindowsError's behaviour to support python 2.4 on Windows This change treat the ESRCH error as ENOENT like WindowsError class does in python 2.5 or later. Without this change, some try..execpt code which expects errno is ENOENT may fail. Actually hg command does not work with python 2.4 on Windows. CreateFile() will fail with error code ESRCH when parent directory of specified path is not exist, or ENOENT when parent directory exist but file is not exist. Two errors are same in the mean of "file is not exist". So WindowsError class treats error code ESRCH as ENOENT in python 2.5 or later, but python 2.4 does not. Actual results with python 2.4: >>> errno.ENOENT 2 >>> errno.ESRCH 3 >>> WindowsError(3, 'msg').errno 3 >>> WindowsError(3, 'msg').args (3, 'msg') And with python 2.5 (or later): >>> errno.ENOENT 2 >>> errno.ESRCH 3 >>> WindowsError(3, 'msg').errno 2 >>> WindowsError(3, 'msg').args (3, 'msg') Note that there is no need to fix osutil.c because it never be used with python 2.4.

author	Shun-ichi GOTO <shunichi.goto@gmail.com>
date	Fri, 12 Jul 2013 11:14:42 +0900
parents	2f6418d8a4c9
children	977102cb12fc

rev	line source
8232 823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	1 # osutil.py - pure Python version of osutil.c
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	2 #
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	3 # Copyright 2009 Matt Mackall <mpm@selenic.com> and others
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	4 #
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	5 # This software may be used and distributed according to the terms of the
10263 25e572394f5c Update license to GPLv2+ Matt Mackall <mpm@selenic.com> parents: 9031 diff changeset	6 # GNU General Public License version 2 or any later version.
8232 823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	7
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	8 import os
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	9 import stat as statmod
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	10
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	11 def _mode_to_kind(mode):
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	12 if statmod.S_ISREG(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	13 return statmod.S_IFREG
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	14 if statmod.S_ISDIR(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	15 return statmod.S_IFDIR
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	16 if statmod.S_ISLNK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	17 return statmod.S_IFLNK
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	18 if statmod.S_ISBLK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	19 return statmod.S_IFBLK
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	20 if statmod.S_ISCHR(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	21 return statmod.S_IFCHR
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	22 if statmod.S_ISFIFO(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	23 return statmod.S_IFIFO
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	24 if statmod.S_ISSOCK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	25 return statmod.S_IFSOCK
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	26 return mode
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	27
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	28 def listdir(path, stat=False, skip=None):
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	29 '''listdir(path, stat=False) -> list_of_tuples
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	30
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	31 Return a sorted list containing information about the entries
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	32 in the directory.
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	33
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	34 If stat is True, each element is a 3-tuple:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	35
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	36 (name, type, stat object)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	37
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	38 Otherwise, each element is a 2-tuple:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	39
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	40 (name, type)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	41 '''
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	42 result = []
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	43 prefix = path
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	44 if not prefix.endswith(os.sep):
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	45 prefix += os.sep
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	46 names = os.listdir(path)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	47 names.sort()
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	48 for fn in names:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	49 st = os.lstat(prefix + fn)
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	50 if fn == skip and statmod.S_ISDIR(st.st_mode):
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	51 return []
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	52 if stat:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	53 result.append((fn, _mode_to_kind(st.st_mode), st))
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	54 else:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	55 result.append((fn, _mode_to_kind(st.st_mode)))
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	56 return result
8421 b6d0fa8c7685 posixfile: remove posixfile_nt and fix import bug in windows.py Sune Foldager <cryo@cyanite.org> parents: 8232 diff changeset	57
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	58 if os.name != 'nt':
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	59 posixfile = open
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	60 else:
16474 ee553e6cd8c4 pure/osutil: use Python's msvcrt module (issue3380) Adrian Buehlmann <adrian@cadifra.com> parents: 15040 diff changeset	61 import ctypes, msvcrt
19465 004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	62 from errno import ESRCH, ENOENT
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	63
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	64 _kernel32 = ctypes.windll.kernel32
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	65
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	66 _DWORD = ctypes.c_ulong
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	67 _LPCSTR = _LPSTR = ctypes.c_char_p
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	68 _HANDLE = ctypes.c_void_p
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	69
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	70 _INVALID_HANDLE_VALUE = _HANDLE(-1).value
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	71
18959 2f6418d8a4c9 check-code: catch trailing space in comments Mads Kiilerich <madski@unity3d.com> parents: 17429 diff changeset	72 # CreateFile
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	73 _FILE_SHARE_READ = 0x00000001
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	74 _FILE_SHARE_WRITE = 0x00000002
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	75 _FILE_SHARE_DELETE = 0x00000004
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	76
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	77 _CREATE_ALWAYS = 2
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	78 _OPEN_EXISTING = 3
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	79 _OPEN_ALWAYS = 4
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	80
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	81 _GENERIC_READ = 0x80000000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	82 _GENERIC_WRITE = 0x40000000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	83
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	84 _FILE_ATTRIBUTE_NORMAL = 0x80
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	85
17429 72fa4ef2245f declare local constants instead of using magic values and comments Mads Kiilerich <mads@kiilerich.com> parents: 16686 diff changeset	86 # open_osfhandle flags
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	87 _O_RDONLY = 0x0000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	88 _O_RDWR = 0x0002
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	89 _O_APPEND = 0x0008
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	90
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	91 _O_TEXT = 0x4000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	92 _O_BINARY = 0x8000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	93
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	94 # types of parameters of C functions used (required by pypy)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	95
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	96 _kernel32.CreateFileA.argtypes = [_LPCSTR, _DWORD, _DWORD, ctypes.c_void_p,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	97 _DWORD, _DWORD, _HANDLE]
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	98 _kernel32.CreateFileA.restype = _HANDLE
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	99
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	100 def _raiseioerror(name):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	101 err = ctypes.WinError()
19465 004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	102 # For python 2.4, treat ESRCH as ENOENT like WindowsError does
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	103 # in python 2.5 or later.
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	104 # py24: WindowsError(3, '').errno => 3
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	105 # py25 or later: WindowsError(3, '').errno => 2
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	106 errno = err.errno
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	107 if errno == ESRCH:
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	108 errno = ENOENT
004f965630d9 osutil: consider WindowsError's behaviour to support python 2.4 on Windows Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 18959 diff changeset	109 raise IOError(errno, '%s: %s' % (name, err.strerror))
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	110
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	111 class posixfile(object):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	112 '''a file object aiming for POSIX-like semantics
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	113
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	114 CPython's open() returns a file that was opened without setting the
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	115 _FILE_SHARE_DELETE flag, which causes rename and unlink to abort.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	116 This even happens if any hardlinked copy of the file is in open state.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	117 We set _FILE_SHARE_DELETE here, so files opened with posixfile can be
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	118 renamed and deleted while they are held open.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	119 Note that if a file opened with posixfile is unlinked, the file
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	120 remains but cannot be opened again or be recreated under the same name,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	121 until all reading processes have closed the file.'''
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	122
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	123 def __init__(self, name, mode='r', bufsize=-1):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	124 if 'b' in mode:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	125 flags = _O_BINARY
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	126 else:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	127 flags = _O_TEXT
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	128
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	129 m0 = mode[0]
16686 67964cda8701 cleanup: "not x in y" -> "x not in y" Brodie Rao <brodie@sf.io> parents: 16474 diff changeset	130 if m0 == 'r' and '+' not in mode:
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	131 flags \|= _O_RDONLY
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	132 access = _GENERIC_READ
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	133 else:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	134 # work around http://support.microsoft.com/kb/899149 and
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	135 # set _O_RDWR for 'w' and 'a', even if mode has no '+'
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	136 flags \|= _O_RDWR
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	137 access = _GENERIC_READ \| _GENERIC_WRITE
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	138
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	139 if m0 == 'r':
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	140 creation = _OPEN_EXISTING
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	141 elif m0 == 'w':
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	142 creation = _CREATE_ALWAYS
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	143 elif m0 == 'a':
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	144 creation = _OPEN_ALWAYS
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	145 flags \|= _O_APPEND
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	146 else:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	147 raise ValueError("invalid mode: %s" % mode)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	148
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	149 fh = _kernel32.CreateFileA(name, access,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	150 _FILE_SHARE_READ \| _FILE_SHARE_WRITE \| _FILE_SHARE_DELETE,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	151 None, creation, _FILE_ATTRIBUTE_NORMAL, None)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	152 if fh == _INVALID_HANDLE_VALUE:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	153 _raiseioerror(name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	154
16474 ee553e6cd8c4 pure/osutil: use Python's msvcrt module (issue3380) Adrian Buehlmann <adrian@cadifra.com> parents: 15040 diff changeset	155 fd = msvcrt.open_osfhandle(fh, flags)
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	156 if fd == -1:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	157 _kernel32.CloseHandle(fh)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	158 _raiseioerror(name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	159
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	160 f = os.fdopen(fd, mode, bufsize)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	161 # unfortunately, f.name is '<fdopen>' at this point -- so we store
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	162 # the name on this wrapper. We cannot just assign to f.name,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	163 # because that attribute is read-only.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	164 object.__setattr__(self, 'name', name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	165 object.__setattr__(self, '_file', f)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	166
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	167 def __iter__(self):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	168 return self._file
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	169
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	170 def __getattr__(self, name):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	171 return getattr(self._file, name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	172
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	173 def __setattr__(self, name, value):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	174 '''mimics the read-only attributes of Python file objects
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	175 by raising 'TypeError: readonly attribute' if someone tries:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	176 f = posixfile('foo.txt')
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	177 f.name = 'bla' '''
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	178 return self._file.__setattr__(name, value)

Mercurial > public > mercurial-scm > hg

annotate mercurial/pure/osutil.py @ 19465:004f965630d9