949
945
return os.fstat(f.fileno())[stat.ST_SIZE]
952
# Alias os.urandom to support platforms (which?) without /dev/urandom and
953
# override if it doesn't work. Avoid checking on windows where there is
954
# significant initialisation cost that can be avoided for some bzr calls.
956
rand_bytes = os.urandom
958
if rand_bytes.__module__ != "nt":
948
# Define rand_bytes based on platform.
950
# Python 2.4 and later have os.urandom,
951
# but it doesn't work on some arches
953
rand_bytes = os.urandom
954
except (NotImplementedError, AttributeError):
955
# If python doesn't have os.urandom, or it doesn't work,
956
# then try to first pull random data from /dev/urandom
961
except NotImplementedError:
958
rand_bytes = file('/dev/urandom', 'rb').read
959
# Otherwise, use this hack as a last resort
960
except (IOError, OSError):
962
961
# not well seeded, but better than nothing
963
962
def rand_bytes(n):
1980
1979
_cached_user_encoding = None
1983
def get_user_encoding(use_cache=DEPRECATED_PARAMETER):
1982
def get_user_encoding(use_cache=True):
1984
1983
"""Find out what the preferred user encoding is.
1986
1985
This is generally the encoding that is used for command line parameters
1987
1986
and file contents. This may be different from the terminal encoding
1988
1987
or the filesystem encoding.
1989
:param use_cache: Enable cache for detected encoding.
1990
(This parameter is turned on by default,
1991
and required only for selftesting)
1990
1993
:return: A string defining the preferred user encoding
1992
1995
global _cached_user_encoding
1993
if deprecated_passed(use_cache):
1994
warn_deprecated("use_cache should only have been used for tests",
1995
DeprecationWarning, stacklevel=2)
1996
if _cached_user_encoding is not None:
1996
if _cached_user_encoding is not None and use_cache:
1997
1997
return _cached_user_encoding
1999
if os.name == 'posix' and getattr(locale, 'CODESET', None) is not None:
2000
# Use the existing locale settings and call nl_langinfo directly
2001
# rather than going through getpreferredencoding. This avoids
2002
# <http://bugs.python.org/issue6202> on OSX Python 2.6 and the
2003
# possibility of the setlocale call throwing an error.
2004
user_encoding = locale.nl_langinfo(locale.CODESET)
1999
if sys.platform == 'darwin':
2000
# python locale.getpreferredencoding() always return
2001
# 'mac-roman' on darwin. That's a lie.
2002
sys.platform = 'posix'
2004
if os.environ.get('LANG', None) is None:
2005
# If LANG is not set, we end up with 'ascii', which is bad
2006
# ('mac-roman' is more than ascii), so we set a default which
2007
# will give us UTF-8 (which appears to work in all cases on
2008
# OSX). Users are still free to override LANG of course, as
2009
# long as it give us something meaningful. This work-around
2010
# *may* not be needed with python 3k and/or OSX 10.5, but will
2011
# work with them too -- vila 20080908
2012
os.environ['LANG'] = 'en_US.UTF-8'
2015
sys.platform = 'darwin'
2006
# GZ 2011-12-19: On windows could call GetACP directly instead.
2007
user_encoding = locale.getpreferredencoding(False)
2010
user_encoding = codecs.lookup(user_encoding).name
2012
if user_encoding not in ("", "cp0"):
2020
user_encoding = locale.getpreferredencoding()
2021
except locale.Error, e:
2022
sys.stderr.write('bzr: warning: %s\n'
2023
' Could not determine what text encoding to use.\n'
2024
' This error usually means your Python interpreter\n'
2025
' doesn\'t support the locale set by $LANG (%s)\n'
2026
" Continuing with ascii encoding.\n"
2027
% (e, os.environ.get('LANG')))
2028
user_encoding = 'ascii'
2030
# Windows returns 'cp0' to indicate there is no code page. So we'll just
2031
# treat that as ASCII, and not support printing unicode characters to the
2034
# For python scripts run under vim, we get '', so also treat that as ASCII
2035
if user_encoding in (None, 'cp0', ''):
2036
user_encoding = 'ascii'
2040
codecs.lookup(user_encoding)
2013
2042
sys.stderr.write('bzr: warning:'
2014
2043
' unknown encoding %s.'
2015
2044
' Continuing with ascii encoding.\n'
2016
2045
% user_encoding
2018
user_encoding = 'ascii'
2020
# Get 'ascii' when setlocale has not been called or LANG=C or unset.
2021
if user_encoding == 'ascii':
2022
if sys.platform == 'darwin':
2023
# OSX is special-cased in Python to have a UTF-8 filesystem
2024
# encoding and previously had LANG set here if not present.
2025
user_encoding = 'utf-8'
2026
# GZ 2011-12-19: Maybe UTF-8 should be the default in this case
2027
# for some other posix platforms as well.
2029
_cached_user_encoding = user_encoding
2047
user_encoding = 'ascii'
2050
_cached_user_encoding = user_encoding
2030
2052
return user_encoding
2034
2056
return get_terminal_encoding()
2059
_message_encoding = None
2062
def get_message_encoding():
2063
"""Return the encoding used for messages
2065
While the message encoding is a general setting it should usually only be
2066
needed for decoding system error strings such as from OSError instances.
2068
global _message_encoding
2069
if _message_encoding is None:
2070
if os.name == "posix":
2072
# This is a process-global setting that can change, but should in
2073
# general just get set once at process startup then be constant.
2074
_message_encoding = locale.getlocale(locale.LC_MESSAGES)[1]
2076
# On windows want the result of GetACP() which this boils down to.
2077
_message_encoding = get_user_encoding()
2078
return _message_encoding or "ascii"
2037
2081
def get_host_name():
2038
2082
"""Return the current unicode host name.
added
removed
bzrlib/osutils.py
