diff options
author | Ned Batchelder <ned@nedbatchelder.com> | 2011-03-15 19:18:12 -0400 |
---|---|---|
committer | Ned Batchelder <ned@nedbatchelder.com> | 2011-03-15 19:18:12 -0400 |
commit | 786ddc82a22314ebaa93f0f63a54fd443d09812e (patch) | |
tree | f3b5ba631f2933b8f14e9d33d358d25aec708a22 /coverage/backward.py | |
parent | 930b770abfdf92343abfac72e97be967502fc183 (diff) | |
download | python-coveragepy-786ddc82a22314ebaa93f0f63a54fd443d09812e.tar.gz |
A file opened in 'b' mode needs bytes, not strings. Try to do this in a clean-ish way...
Diffstat (limited to 'coverage/backward.py')
-rw-r--r-- | coverage/backward.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/coverage/backward.py b/coverage/backward.py index c363f21..af6da62 100644 --- a/coverage/backward.py +++ b/coverage/backward.py @@ -81,3 +81,19 @@ except AttributeError: """Open a source file the best way.""" return open(fname, "rU") +# Python 3.x is picky about bytes and strings, so provide methods to +# get them right, and make them no-ops in 2.x +if sys.version_info >= (3, 0): + def to_bytes(s): + return s.encode('utf8') + + def to_string(b): + return b.decode('utf8') + +else: + def to_bytes(s): + return s + + def to_string(b): + return b + |