py3: remove cast_bytes method

This commit is contained in:
Mr. Outis
2019-12-31 20:14:25 -06:00
parent 326ba21ed0
commit b844aea42c
8 changed files with 9 additions and 44 deletions
+2 -5
View File
@@ -9,7 +9,6 @@ from subprocess import Popen
from dvc.env import DVC_DAEMON
from dvc.utils import fix_env
from dvc.utils import is_binary
from dvc.utils.compat import cast_bytes_py2
logger = logging.getLogger(__name__)
@@ -98,9 +97,7 @@ def daemon(args):
env = fix_env()
file_path = os.path.abspath(inspect.stack()[0][1])
env[cast_bytes_py2("PYTHONPATH")] = cast_bytes_py2(
os.path.dirname(os.path.dirname(file_path))
)
env[cast_bytes_py2(DVC_DAEMON)] = cast_bytes_py2("1")
env["PYTHONPATH"] = os.path.dirname(os.path.dirname(file_path))
env[DVC_DAEMON] = "1"
_spawn(cmd, env)
-1
View File
@@ -7,7 +7,6 @@ import logging.handlers
import colorama
from dvc.progress import Tqdm
from dvc.utils.compat import RecursionError
FOOTER = (
+1 -2
View File
@@ -17,7 +17,6 @@ from dvc.utils import fix_env
from dvc.utils import is_binary
from dvc.utils import relpath
from dvc.utils.fs import path_isin
from dvc.utils.compat import cast_bytes_py2
from dvc.utils.compat import open
@@ -76,7 +75,7 @@ class Git(Base):
# LD_LIBRARY_PATH that has been set by PyInstaller.
# See [1] for more info.
# [1] https://github.com/gitpython-developers/GitPython/issues/924
env[cast_bytes_py2(ld_key)] = ""
env[ld_key] = ""
try:
tmp_repo = git.Repo.clone_from(
+1 -3
View File
@@ -15,7 +15,6 @@ import nanotime
from ruamel.yaml import YAML
from shortuuid import uuid
from dvc.utils.compat import cast_bytes_py2
from dvc.utils.compat import fspath
from dvc.utils.compat import fspath_py35
from dvc.utils.compat import makedirs as _makedirs
@@ -205,8 +204,7 @@ def fix_env(env=None):
lp_key = "LD_LIBRARY_PATH"
lp_orig = env.get(lp_key + "_ORIG", None)
if lp_orig is not None:
# NOTE: py2 doesn't like unicode strings in environ
env[cast_bytes_py2(lp_key)] = cast_bytes_py2(lp_orig)
env[lp_key] = lp_orig
else:
env.pop(lp_key, None)
-26
View File
@@ -14,23 +14,6 @@ is_py2 = _ver[0] == 2
#: Python 3.x?
is_py3 = _ver[0] == 3
# simplified version of ipython_genutils/encoding.py
DEFAULT_ENCODING = sys.getdefaultencoding()
if _ver[:2] < (3, 5):
RecursionError = RuntimeError
else:
RecursionError = RecursionError
def no_code(x, encoding=None):
return x
def encode(u, encoding=None):
encoding = encoding or DEFAULT_ENCODING
return u.encode(encoding, "replace")
def csv_reader(unicode_csv_data, dialect=None, **kwargs):
"""csv.reader doesn't support Unicode input, so need to use some tricks
@@ -63,13 +46,6 @@ def utf_8_encoder(unicode_csv_data):
yield line.encode("utf-8")
def cast_bytes(s, encoding=None):
"""Source: https://github.com/ipython/ipython_genutils"""
if not isinstance(s, bytes):
return encode(s, encoding)
return s
def _makedirs(name, mode=0o777, exist_ok=False):
"""Source: https://github.com/python/cpython/blob/
3ce3dea60646d8a5a1c952469a2eb65f937875b3/Lib/os.py#L196-L226
@@ -118,7 +94,6 @@ if is_py2:
numeric_types = (int, long, float) # noqa: F821
integer_types = (int, long) # noqa: F821
input = raw_input # noqa: F821
cast_bytes_py2 = cast_bytes
makedirs = _makedirs
range = xrange # noqa: F821
FileNotFoundError = IOError
@@ -174,7 +149,6 @@ elif is_py3:
integer_types = (int,) # noqa: F821
input = input # noqa: F821
open = open # noqa: F821
cast_bytes_py2 = no_code
range = range # noqa: F821
FileNotFoundError = FileNotFoundError
+1 -1
View File
@@ -49,7 +49,7 @@ def _is_release(dir_path, base_version):
["git", "describe", "--tags", "--exact-match"],
cwd=dir_path,
stderr=subprocess.STDOUT,
).decode("utf-8")
)
tag = output.strip()
return tag == base_version
except subprocess.CalledProcessError:
+2 -3
View File
@@ -7,15 +7,14 @@ from git.exc import GitCommandNotFound
from dvc.remote.ssh.connection import SSHConnection
from dvc.repo import Repo as DvcRepo
from dvc.utils.compat import cast_bytes_py2
from .basic_env import TestDirFixture, TestDvcGitFixture, TestGitFixture
from .dir_helpers import * # noqa
# Prevent updater and analytics from running their processes
os.environ[cast_bytes_py2("DVC_TEST")] = cast_bytes_py2("true")
os.environ["DVC_TEST"] = "true"
# Ensure progress output even when not outputting to raw sys.stderr console
os.environ[cast_bytes_py2("DVC_IGNORE_ISATTY")] = cast_bytes_py2("true")
os.environ["DVC_IGNORE_ISATTY"] = "true"
@pytest.fixture(autouse=True)
+2 -3
View File
@@ -3,7 +3,6 @@ import subprocess
import mock
import dvc.version
from dvc.utils.compat import cast_bytes
def test_is_release():
@@ -13,10 +12,10 @@ def test_is_release():
assert ret is False
m.side_effect = None
m.return_value = cast_bytes(dvc.version._BASE_VERSION)
m.return_value = dvc.version._BASE_VERSION
ret = dvc.version._is_release(None, dvc.version._BASE_VERSION)
assert ret
m.return_value = cast_bytes("630d1741c2d5dd89a3176bd15b63121b905d35c9")
m.return_value = "630d1741c2d5dd89a3176bd15b63121b905d35c9"
ret = dvc.version._is_release(None, dvc.version._BASE_VERSION)
assert ret is False