945
949
return os.fstat(f.fileno())[stat.ST_SIZE]
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
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":
958
rand_bytes = file('/dev/urandom', 'rb').read
959
# Otherwise, use this hack as a last resort
960
except (IOError, OSError):
961
except NotImplementedError:
961
962
# not well seeded, but better than nothing
962
963
def rand_bytes(n):
1979
1980
_cached_user_encoding = None
1982
def get_user_encoding(use_cache=True):
1983
def get_user_encoding(use_cache=DEPRECATED_PARAMETER):
1983
1984
"""Find out what the preferred user encoding is.
1985
1986
This is generally the encoding that is used for command line parameters
1986
1987
and file contents. This may be different from the terminal encoding
1987
1988
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)
1993
1990
:return: A string defining the preferred user encoding
1995
1992
global _cached_user_encoding
1996
if _cached_user_encoding is not None and use_cache:
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:
1997
1997
return _cached_user_encoding
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'
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)
2006
# GZ 2011-12-19: On windows could call GetACP directly instead.
2007
user_encoding = locale.getpreferredencoding(False)
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)
2010
user_encoding = codecs.lookup(user_encoding).name
2012
if user_encoding not in ("", "cp0"):
2042
2013
sys.stderr.write('bzr: warning:'
2043
2014
' unknown encoding %s.'
2044
2015
' Continuing with ascii encoding.\n'
2045
2016
% user_encoding
2047
user_encoding = 'ascii'
2050
_cached_user_encoding = 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
2052
2030
return user_encoding
2056
2034
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"
2081
2037
def get_host_name():
2082
2038
"""Return the current unicode host name.
added
removed
bzrlib/osutils.py
