/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Vincent Ladeuil
  • Date: 2010-06-20 11:18:38 UTC
  • mto: (5247.1.10 first-try)
  • mto: This revision was merged to the branch mainline in revision 5326.
  • Revision ID: v.ladeuil+lp@free.fr-20100620111838-3s726ne0g2ni42m1
No more use of the get_transport imported *symbol*, all uses are through
the module now, allowing effective wrapping.

* bzrlib/workingtree.py:
(WorkingTree.find_trees): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/transport/decorator.py:
(TransportDecorator.__init__): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_workingtree.py:
(TestWorkingTreeFormat.test_find_format.check_format): More evil
calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/test_upgrade.py:
(TestUpgrade.test_upgrade_simple):
(TestUpgrade.test_upgrade_v6_to_meta_no_workingtree): More evil
calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/test_transport_log.py:
(test_log_transport): More evil calls to transport.get_transport
defeating the wrapping.

* bzrlib/tests/test_smart_request.py:
(TestJailHook.test_jail_hook): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_sftp_transport.py:
(SFTPNonServerTest.test_abspath_root_sibling_server):
(SSHVendorBadConnection.test_bad_connection_paramiko):
(SSHVendorBadConnection.test_bad_connection_ssh):
(TestUsesAuthConfig.get_transport_for_connection): More evil calls
to transport.get_transport defeating the wrapping.

* bzrlib/tests/test_selftest.py:
(test_get_readonly_url_none):
(TestTestCaseWithTransport.test_get_readonly_url_none):
(TestTestCaseWithTransport.test_get_readonly_url_http):
(TestChrootedTest.test_root_is_root): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_remote.py:
(Test_ClientMedium_remote_path_from_transport.assertRemotePath):
(Test_ClientMedium_remote_path_from_transport.assertRemotePathHTTP):
More evil calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/test_reconcile.py:
Unused import deleted.

* bzrlib/tests/test_permissions.py:
(check_mode_r):
(TestSftpPermissions.test_new_files):
(TestSftpPermissions.test_sftp_server_modes): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_lockable_files.py:
(setUp, TestLockableFiles_RemoteLockDir.setUp): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_knit.py:
(TestKnitIndex.test_knit_index_ignores_empty_files): More evil
calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/test_index.py:
(test_build_index_empty_two_element_keys):
(test_build_index_one_reference_list_empty):
(test_build_index_two_reference_list_empty):
(test_build_index_one_node_no_refs):
(test_build_index_one_node_no_refs_accepts_empty_reflist):
(TestGraphIndexBuilder.test_build_index_one_node_2_element_keys):
(test_add_node_empty_value):
(test_build_index_nodes_sorted):
(test_build_index_2_element_key_nodes_sorted):
(test_build_index_reference_lists_are_included_one):
(test_build_index_reference_lists_with_2_element_keys):
(test_build_index_reference_lists_are_included_two):
(test_clear_cache):
(test_node_references_are_cr_delimited):
(test_multiple_reference_lists_are_tab_delimited):
(test_add_node_referencing_missing_key_makes_absent):
(test_node_references_three_digits):
(test_absent_has_no_reference_overhead):
(test_add_node_bad_key):
(TestGraphIndexBuilder.test_add_node_bad_key):
(test_add_node_bad_key_in_reference_lists):
(TestGraphIndexBuilder.test_add_node_bad_key_in_reference_lists):
(make_index):
(test_open_bad_index_no_error):
(test_iter_missing_entry_empty_no_size):
(test_validate_bad_index_errors):
(test_supports_unlimited_cache):
(TestCombinedGraphIndex.make_combined_index_with_missing):
(reload):
(TestCombinedGraphIndex.test_clear_cache):
(test_reorder_after_iter_entries):
(test_validate_reloads):
(TestCombinedGraphIndex.test_find_ancestors_across_indexes):
(TestCombinedGraphIndex.test_find_ancestors_missing_keys):
(test_find_ancestors_no_indexes):
(TestCombinedGraphIndex.test_find_ancestors_ghost_parent):
(test__find_ancestors_empty_index):
(test_construct): More evil calls to transport.get_transport
defeating the wrapping and some more import fixes.

* bzrlib/tests/test_directory_service.py:
(TestDirectoryLookup.test_get_transport): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/test_debug.py:
Unused import deleted.

* bzrlib/tests/test_btree_index.py:
(TestBTreeBuilder.test_three_level_tree_details):
(TestBTreeBuilder.test_spill_index_stress_2_2):
(TestBTreeIndex.make_index):
(test_trivial_constructor):
(test__read_nodes_no_size_one_page_reads_once):
(TestBTreeIndex.test__read_nodes_no_size_multiple_pages):
(TestBTreeIndex.test_2_levels_key_count_2_2):
(TestBTreeIndex.test_validate_two_pages):
(TestBTreeIndex.test_iter_all_entries_reads):
(TestBTreeIndex.test_iter_all_entries_reads):
(TestBTreeIndex.test_iter_entries_references_2_refs_resolved):
(TestBTreeIndex.test_iter_entries_references_2_refs_resolved):
(TestBTreeIndex.test_supports_unlimited_cache):
(TestExpandOffsets.make_index): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/per_workingtree/test_pull.py:
Unused import deleted.

* bzrlib/tests/per_versionedfile.py:
(test_readonly_mode):
(TestWeave.get_file):
(reopen_file):
(TestReadonlyHttpMixin.test_readonly_http_works):
(TestReadonlyHttpMixin.test_readonly_http_works_with_feeling):
(TestWeaveHTTP.get_file):
(TestWeaveMerge.get_file): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/per_repository/test_reconcile.py:
(setUp):
(TestReconcileWithIncorrectRevisionCache.setUp):
(test_reconcile_wrong_order):
(test_reconcile_wrong_order_secondary_inventory): More evil calls
to transport.get_transport defeating the wrapping.

* bzrlib/tests/per_repository/test_has_same_location.py:
(TestHasSameLocation.test_different_format_not_equal): More evil
calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/per_repository/test_fetch.py:
Unused import deleted.

* bzrlib/tests/per_intertree/__init__.py:
Unused import deleted.

* bzrlib/tests/per_interrepository/__init__.py:
(TestCaseWithInterRepository.make_bzrdir): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/per_interbranch/test_push.py:
Unused import deleted.

* bzrlib/tests/per_interbranch/__init__.py:
Unused import deleted.

* bzrlib/tests/per_bzrdir/test_bzrdir.py:
(TestBzrDir.sproutOrSkip):
(TestBzrDir.test_format_initialize_find_open):
(TestBzrDir.test_open_not_bzrdir):
(TestBzrDir.test_open_branch):
(TestBzrDir.test_open_branch):
(TestBzrDir.test_list_branches):
(TestBzrDir.test_create_repository):
(TestBzrDir.test_create_repository_shared):
(TestBzrDir.test_create_repository_nonshared):
(TestBzrDir.test_open_repository):
(test_root_transport):
(TestBzrDir.test_find_repository_no_repo_under_standalone_branch):
(TestBzrDir.test_find_repository_containing_shared_repository):
(TestBzrDir.test_find_repository_shared_within_shared_repository):
(TestBzrDir.test_find_repository_with_nested_dirs_works):
(ChrootedBzrDirTests.test_find_repository_no_repository):
(TestBzrDirControlComponent): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/tests/per_branch/test_stacking.py: 
Unused import deleted.

* bzrlib/tests/per_branch/test_permissions.py: 
Unused import deleted.

* bzrlib/tests/per_branch/test_bound_sftp.py:
(test_unbinding): More evil calls to transport.get_transport
defeating the wrapping.

* bzrlib/tests/blackbox/test_upgrade.py:
(TestWithUpgradableBranches.test_upgrade_up_to_date_checkout_warns_branch_left_alone):
(test_upgrade_explicit_metaformat):
(TestWithUpgradableBranches.test_upgrade_explicit_knit):
(test_upgrade_with_existing_backup_dir):
(TestWithUpgradableBranches.test_upgrade_with_existing_backup_dir):
More evil calls to transport.get_transport defeating the wrapping.

* bzrlib/tests/blackbox/test_serve.py:
(test_bzr_serve_port_readonly): More evil calls to
transport.get_transport defeating the wrapping.

* bzrlib/plugins/launchpad/test_lp_directory.py:
(DirectoryOpenBranchTests.test_directory_open_branch): Use
transport.get_transport instead of get_transport.

* bzrlib/plugins/launchpad/account.py:
(check_lp_login): Use transport.get_transport instead of
get_transport.

* bzrlib/builtins.py (run): Use transport.get_transport instead of
get_transport.

* bzrlib/btree_index.py:
(BTreeBuilder._spill_mem_keys_to_disk): Use
transport.get_transport instead of get_transport.

* bzrlib/benchmarks/bench_commit.py:
Unused import deleted.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/osutils.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
import errno
17
18
import os
18
19
import re
19
20
import stat
20
 
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
21
21
import sys
22
22
import time
23
23
import codecs
25
25
from bzrlib.lazy_import import lazy_import
26
26
lazy_import(globals(), """
27
27
from datetime import datetime
28
 
import errno
29
 
from ntpath import (abspath as _nt_abspath,
30
 
                    join as _nt_join,
31
 
                    normpath as _nt_normpath,
32
 
                    realpath as _nt_realpath,
33
 
                    splitdrive as _nt_splitdrive,
34
 
                    )
 
28
import getpass
 
29
import ntpath
35
30
import posixpath
36
31
import shutil
37
 
from shutil import (
38
 
    rmtree,
39
 
    )
40
32
import socket
41
33
import subprocess
42
34
import tempfile
43
 
from tempfile import (
44
 
    mkdtemp,
45
 
    )
46
35
import unicodedata
47
36
 
48
37
from bzrlib import (
303
292
    running python.exe under cmd.exe return capital C:\\
304
293
    running win32 python inside a cygwin shell returns lowercase c:\\
305
294
    """
306
 
    drive, path = _nt_splitdrive(path)
 
295
    drive, path = ntpath.splitdrive(path)
307
296
    return drive.upper() + path
308
297
 
309
298
 
310
299
def _win32_abspath(path):
311
 
    # Real _nt_abspath doesn't have a problem with a unicode cwd
312
 
    return _win32_fixdrive(_nt_abspath(unicode(path)).replace('\\', '/'))
 
300
    # Real ntpath.abspath doesn't have a problem with a unicode cwd
 
301
    return _win32_fixdrive(ntpath.abspath(unicode(path)).replace('\\', '/'))
313
302
 
314
303
 
315
304
def _win98_abspath(path):
326
315
    #   /path       => C:/path
327
316
    path = unicode(path)
328
317
    # check for absolute path
329
 
    drive = _nt_splitdrive(path)[0]
 
318
    drive = ntpath.splitdrive(path)[0]
330
319
    if drive == '' and path[:2] not in('//','\\\\'):
331
320
        cwd = os.getcwdu()
332
321
        # we cannot simply os.path.join cwd and path
333
322
        # because os.path.join('C:','/path') produce '/path'
334
323
        # and this is incorrect
335
324
        if path[:1] in ('/','\\'):
336
 
            cwd = _nt_splitdrive(cwd)[0]
 
325
            cwd = ntpath.splitdrive(cwd)[0]
337
326
            path = path[1:]
338
327
        path = cwd + '\\' + path
339
 
    return _win32_fixdrive(_nt_normpath(path).replace('\\', '/'))
 
328
    return _win32_fixdrive(ntpath.normpath(path).replace('\\', '/'))
340
329
 
341
330
 
342
331
def _win32_realpath(path):
343
 
    # Real _nt_realpath doesn't have a problem with a unicode cwd
344
 
    return _win32_fixdrive(_nt_realpath(unicode(path)).replace('\\', '/'))
 
332
    # Real ntpath.realpath doesn't have a problem with a unicode cwd
 
333
    return _win32_fixdrive(ntpath.realpath(unicode(path)).replace('\\', '/'))
345
334
 
346
335
 
347
336
def _win32_pathjoin(*args):
348
 
    return _nt_join(*args).replace('\\', '/')
 
337
    return ntpath.join(*args).replace('\\', '/')
349
338
 
350
339
 
351
340
def _win32_normpath(path):
352
 
    return _win32_fixdrive(_nt_normpath(unicode(path)).replace('\\', '/'))
 
341
    return _win32_fixdrive(ntpath.normpath(unicode(path)).replace('\\', '/'))
353
342
 
354
343
 
355
344
def _win32_getcwd():
360
349
    return _win32_fixdrive(tempfile.mkdtemp(*args, **kwargs).replace('\\', '/'))
361
350
 
362
351
 
363
 
def _add_rename_error_details(e, old, new):
364
 
    new_e = OSError(e.errno, "failed to rename %s to %s: %s"
365
 
        % (old, new, e.strerror))
366
 
    new_e.filename = old
367
 
    new_e.to_filename = new
368
 
    return new_e
369
 
 
370
 
 
371
352
def _win32_rename(old, new):
372
353
    """We expect to be able to atomically replace 'new' with old.
373
354
 
375
356
    and then deleted.
376
357
    """
377
358
    try:
378
 
        fancy_rename(old, new, rename_func=_wrapped_rename, unlink_func=os.unlink)
 
359
        fancy_rename(old, new, rename_func=os.rename, unlink_func=os.unlink)
379
360
    except OSError, e:
380
361
        if e.errno in (errno.EPERM, errno.EACCES, errno.EBUSY, errno.EINVAL):
381
362
            # If we try to rename a non-existant file onto cwd, we get
386
367
        raise
387
368
 
388
369
 
389
 
def _wrapped_rename(old, new):
390
 
    """Rename a file or directory"""
391
 
    try:
392
 
        os.rename(old, new)
393
 
    except (IOError, OSError), e:
394
 
        # this is eventually called by all rename-like functions, so should 
395
 
        # catch all of them
396
 
        raise _add_rename_error_details(e, old, new)
397
 
 
398
 
 
399
370
def _mac_getcwd():
400
371
    return unicodedata.normalize('NFC', os.getcwdu())
401
372
 
406
377
realpath = _posix_realpath
407
378
pathjoin = os.path.join
408
379
normpath = os.path.normpath
409
 
rename = _wrapped_rename # overridden below on win32
410
380
getcwd = os.getcwdu
 
381
rename = os.rename
411
382
dirname = os.path.dirname
412
383
basename = os.path.basename
413
384
split = os.path.split
414
385
splitext = os.path.splitext
415
 
# These were already imported into local scope
416
 
# mkdtemp = tempfile.mkdtemp
417
 
# rmtree = shutil.rmtree
 
386
mkdtemp = tempfile.mkdtemp
 
387
rmtree = shutil.rmtree
418
388
 
419
389
MIN_ABS_PATHLENGTH = 1
420
390
 
519
489
def isdir(f):
520
490
    """True if f is an accessible directory."""
521
491
    try:
522
 
        return S_ISDIR(os.lstat(f)[ST_MODE])
 
492
        return stat.S_ISDIR(os.lstat(f)[stat.ST_MODE])
523
493
    except OSError:
524
494
        return False
525
495
 
527
497
def isfile(f):
528
498
    """True if f is a regular file."""
529
499
    try:
530
 
        return S_ISREG(os.lstat(f)[ST_MODE])
 
500
        return stat.S_ISREG(os.lstat(f)[stat.ST_MODE])
531
501
    except OSError:
532
502
        return False
533
503
 
534
504
def islink(f):
535
505
    """True if f is a symlink."""
536
506
    try:
537
 
        return S_ISLNK(os.lstat(f)[ST_MODE])
 
507
        return stat.S_ISLNK(os.lstat(f)[stat.ST_MODE])
538
508
    except OSError:
539
509
        return False
540
510
 
880
850
 
881
851
def filesize(f):
882
852
    """Return size of given open file."""
883
 
    return os.fstat(f.fileno())[ST_SIZE]
 
853
    return os.fstat(f.fileno())[stat.ST_SIZE]
884
854
 
885
855
 
886
856
# Define rand_bytes based on platform.
948
918
 
949
919
def parent_directories(filename):
950
920
    """Return the list of parent directories, deepest first.
951
 
    
 
921
 
952
922
    For example, parent_directories("a/b/c") -> ["a/b", "a"].
953
923
    """
954
924
    parents = []
978
948
    # NB: This docstring is just an example, not a doctest, because doctest
979
949
    # currently can't cope with the use of lazy imports in this namespace --
980
950
    # mbp 20090729
981
 
    
 
951
 
982
952
    # This currently doesn't report the failure at the time it occurs, because
983
953
    # they tend to happen very early in startup when we can't check config
984
954
    # files etc, and also we want to report all failures but not spam the user
1054
1024
 
1055
1025
 
1056
1026
def delete_any(path):
1057
 
    """Delete a file, symlink or directory.  
1058
 
    
 
1027
    """Delete a file, symlink or directory.
 
1028
 
1059
1029
    Will delete even if readonly.
1060
1030
    """
1061
1031
    try:
1250
1220
    # but for now, we haven't optimized...
1251
1221
    return [canonical_relpath(base, p) for p in paths]
1252
1222
 
 
1223
 
 
1224
def decode_filename(filename):
 
1225
    """Decode the filename using the filesystem encoding
 
1226
 
 
1227
    If it is unicode, it is returned.
 
1228
    Otherwise it is decoded from the the filesystem's encoding. If decoding
 
1229
    fails, a errors.BadFilenameEncoding exception is raised.
 
1230
    """
 
1231
    if type(filename) is unicode:
 
1232
        return filename
 
1233
    try:
 
1234
        return filename.decode(_fs_enc)
 
1235
    except UnicodeDecodeError:
 
1236
        raise errors.BadFilenameEncoding(filename, _fs_enc)
 
1237
 
 
1238
 
1253
1239
def safe_unicode(unicode_or_utf8_string):
1254
1240
    """Coerce unicode_or_utf8_string into unicode.
1255
1241
 
1338
1324
def normalizes_filenames():
1339
1325
    """Return True if this platform normalizes unicode filenames.
1340
1326
 
1341
 
    Mac OSX does, Windows/Linux do not.
 
1327
    Only Mac OSX.
1342
1328
    """
1343
1329
    return _platform_normalizes_filenames
1344
1330
 
1349
1335
    On platforms where the system normalizes filenames (Mac OSX),
1350
1336
    you can access a file by any path which will normalize correctly.
1351
1337
    On platforms where the system does not normalize filenames
1352
 
    (Windows, Linux), you have to access a file by its exact path.
 
1338
    (everything else), you have to access a file by its exact path.
1353
1339
 
1354
1340
    Internally, bzr only supports NFC normalization, since that is
1355
1341
    the standard for XML documents.
1417
1403
terminal_width() returns None.
1418
1404
"""
1419
1405
 
 
1406
# Keep some state so that terminal_width can detect if _terminal_size has
 
1407
# returned a different size since the process started.  See docstring and
 
1408
# comments of terminal_width for details.
 
1409
# _terminal_size_state has 3 possible values: no_data, unchanged, and changed.
 
1410
_terminal_size_state = 'no_data'
 
1411
_first_terminal_size = None
1420
1412
 
1421
1413
def terminal_width():
1422
1414
    """Return terminal width.
1426
1418
    The rules are:
1427
1419
    - if BZR_COLUMNS is set, returns its value
1428
1420
    - if there is no controlling terminal, returns None
 
1421
    - query the OS, if the queried size has changed since the last query,
 
1422
      return its value,
1429
1423
    - if COLUMNS is set, returns its value,
 
1424
    - if the OS has a value (even though it's never changed), return its value.
1430
1425
 
1431
1426
    From there, we need to query the OS to get the size of the controlling
1432
1427
    terminal.
1433
1428
 
1434
 
    Unices:
 
1429
    On Unices we query the OS by:
1435
1430
    - get termios.TIOCGWINSZ
1436
1431
    - if an error occurs or a negative value is obtained, returns None
1437
1432
 
1438
 
    Windows:
1439
 
    
 
1433
    On Windows we query the OS by:
1440
1434
    - win32utils.get_console_size() decides,
1441
1435
    - returns None on error (provided default value)
1442
1436
    """
 
1437
    # Note to implementors: if changing the rules for determining the width,
 
1438
    # make sure you've considered the behaviour in these cases:
 
1439
    #  - M-x shell in emacs, where $COLUMNS is set and TIOCGWINSZ returns 0,0.
 
1440
    #  - bzr log | less, in bash, where $COLUMNS not set and TIOCGWINSZ returns
 
1441
    #    0,0.
 
1442
    #  - (add more interesting cases here, if you find any)
 
1443
    # Some programs implement "Use $COLUMNS (if set) until SIGWINCH occurs",
 
1444
    # but we don't want to register a signal handler because it is impossible
 
1445
    # to do so without risking EINTR errors in Python <= 2.6.5 (see
 
1446
    # <http://bugs.python.org/issue8354>).  Instead we check TIOCGWINSZ every
 
1447
    # time so we can notice if the reported size has changed, which should have
 
1448
    # a similar effect.
1443
1449
 
1444
1450
    # If BZR_COLUMNS is set, take it, user is always right
1445
1451
    try:
1448
1454
        pass
1449
1455
 
1450
1456
    isatty = getattr(sys.stdout, 'isatty', None)
1451
 
    if  isatty is None or not isatty():
 
1457
    if isatty is None or not isatty():
1452
1458
        # Don't guess, setting BZR_COLUMNS is the recommended way to override.
1453
1459
        return None
1454
1460
 
1455
 
    # If COLUMNS is set, take it, the terminal knows better (even inside a
1456
 
    # given terminal, the application can decide to set COLUMNS to a lower
1457
 
    # value (splitted screen) or a bigger value (scroll bars))
 
1461
    # Query the OS
 
1462
    width, height = os_size = _terminal_size(None, None)
 
1463
    global _first_terminal_size, _terminal_size_state
 
1464
    if _terminal_size_state == 'no_data':
 
1465
        _first_terminal_size = os_size
 
1466
        _terminal_size_state = 'unchanged'
 
1467
    elif (_terminal_size_state == 'unchanged' and
 
1468
          _first_terminal_size != os_size):
 
1469
        _terminal_size_state = 'changed'
 
1470
 
 
1471
    # If the OS claims to know how wide the terminal is, and this value has
 
1472
    # ever changed, use that.
 
1473
    if _terminal_size_state == 'changed':
 
1474
        if width is not None and width > 0:
 
1475
            return width
 
1476
 
 
1477
    # If COLUMNS is set, use it.
1458
1478
    try:
1459
1479
        return int(os.environ['COLUMNS'])
1460
1480
    except (KeyError, ValueError):
1461
1481
        pass
1462
1482
 
1463
 
    width, height = _terminal_size(None, None)
1464
 
    if width <= 0:
1465
 
        # Consider invalid values as meaning no width
1466
 
        return None
 
1483
    # Finally, use an unchanged size from the OS, if we have one.
 
1484
    if _terminal_size_state == 'unchanged':
 
1485
        if width is not None and width > 0:
 
1486
            return width
1467
1487
 
1468
 
    return width
 
1488
    # The width could not be determined.
 
1489
    return None
1469
1490
 
1470
1491
 
1471
1492
def _win32_terminal_size(width, height):
1498
1519
    _terminal_size = _ioctl_terminal_size
1499
1520
 
1500
1521
 
1501
 
def _terminal_size_changed(signum, frame):
1502
 
    """Set COLUMNS upon receiving a SIGnal for WINdow size CHange."""
1503
 
    width, height = _terminal_size(None, None)
1504
 
    if width is not None:
1505
 
        os.environ['COLUMNS'] = str(width)
1506
 
 
1507
 
 
1508
 
_registered_sigwinch = False
1509
 
def watch_sigwinch():
1510
 
    """Register for SIGWINCH, once and only once.
1511
 
 
1512
 
    Do nothing if the signal module is not available.
1513
 
    """
1514
 
    global _registered_sigwinch
1515
 
    if not _registered_sigwinch:
1516
 
        try:
1517
 
            import signal
1518
 
            if getattr(signal, "SIGWINCH", None) is not None:
1519
 
                set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1520
 
        except ImportError:
1521
 
            # python doesn't provide signal support, nothing we can do about it
1522
 
            pass
1523
 
        _registered_sigwinch = True
1524
 
 
1525
 
 
1526
1522
def supports_executable():
1527
1523
    return sys.platform != "win32"
1528
1524
 
1651
1647
        dirblock = []
1652
1648
        append = dirblock.append
1653
1649
        try:
1654
 
            names = sorted(_listdir(top))
 
1650
            names = sorted(map(decode_filename, _listdir(top)))
1655
1651
        except OSError, e:
1656
1652
            if not _is_error_enotdir(e):
1657
1653
                raise
2027
2023
 
2028
2024
def send_all(sock, bytes, report_activity=None):
2029
2025
    """Send all bytes on a socket.
2030
 
 
 
2026
 
2031
2027
    Breaks large blocks in smaller chunks to avoid buffering limitations on
2032
2028
    some platforms, and catches EINTR which may be thrown if the send is
2033
2029
    interrupted by a signal.
2034
2030
 
2035
2031
    This is preferred to socket.sendall(), because it avoids portability bugs
2036
2032
    and provides activity reporting.
2037
 
 
 
2033
 
2038
2034
    :param report_activity: Call this as bytes are read, see
2039
2035
        Transport._report_activity
2040
2036
    """
2095
2091
    base = dirname(bzrlib.__file__)
2096
2092
    if getattr(sys, 'frozen', None):    # bzr.exe
2097
2093
        base = abspath(pathjoin(base, '..', '..'))
2098
 
    filename = pathjoin(base, resource_relpath)
2099
 
    return open(filename, 'rU').read()
2100
 
 
 
2094
    f = file(pathjoin(base, resource_relpath), "rU")
 
2095
    try:
 
2096
        return f.read()
 
2097
    finally:
 
2098
        f.close()
2101
2099
 
2102
2100
def file_kind_from_stat_mode_thunk(mode):
2103
2101
    global file_kind_from_stat_mode
2126
2124
 
2127
2125
def until_no_eintr(f, *a, **kw):
2128
2126
    """Run f(*a, **kw), retrying if an EINTR error occurs.
2129
 
    
 
2127
 
2130
2128
    WARNING: you must be certain that it is safe to retry the call repeatedly
2131
2129
    if EINTR does occur.  This is typically only true for low-level operations
2132
2130
    like os.read.  If in any doubt, don't use this.
2263
2261
if sys.platform == 'win32':
2264
2262
    def open_file(filename, mode='r', bufsize=-1):
2265
2263
        """This function is used to override the ``open`` builtin.
2266
 
        
 
2264
 
2267
2265
        But it uses O_NOINHERIT flag so the file handle is not inherited by
2268
2266
        child processes.  Deleting or renaming a closed file opened with this
2269
2267
        function is not blocking child processes.
2302
2300
        return os.fdopen(os.open(filename, flags), mode, bufsize)
2303
2301
else:
2304
2302
    open_file = open
 
2303
 
 
2304
 
 
2305
def getuser_unicode():
 
2306
    """Return the username as unicode.
 
2307
    """
 
2308
    try:
 
2309
        user_encoding = get_user_encoding()
 
2310
        username = getpass.getuser().decode(user_encoding)
 
2311
    except UnicodeDecodeError:
 
2312
        raise errors.BzrError("Can't decode username as %s." % \
 
2313
                user_encoding)
 
2314
    return username