105 lines
2.2 KiB
Python
105 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# config.py
|
||
|
|
# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
|
||
|
|
#
|
||
|
|
# This module is part of GitPython and is released under
|
||
|
|
# the BSD License: http://www.opensource.org/licenses/bsd-license.php
|
||
|
|
"""utilities to help provide compatibility with python 3"""
|
||
|
|
# flake8: noqa
|
||
|
|
|
||
|
|
import locale
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from gitdb.utils.encoding import (
|
||
|
|
force_bytes, # @UnusedImport
|
||
|
|
force_text, # @UnusedImport
|
||
|
|
)
|
||
|
|
|
||
|
|
# typing --------------------------------------------------------------------
|
||
|
|
|
||
|
|
from typing import (
|
||
|
|
Any,
|
||
|
|
AnyStr,
|
||
|
|
Dict,
|
||
|
|
IO,
|
||
|
|
Optional,
|
||
|
|
Tuple,
|
||
|
|
Type,
|
||
|
|
Union,
|
||
|
|
overload,
|
||
|
|
)
|
||
|
|
|
||
|
|
# ---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
|
||
|
|
is_win: bool = os.name == "nt"
|
||
|
|
is_posix = os.name == "posix"
|
||
|
|
is_darwin = os.name == "darwin"
|
||
|
|
defenc = sys.getfilesystemencoding()
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def safe_decode(s: None) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def safe_decode(s: AnyStr) -> str:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
def safe_decode(s: Union[AnyStr, None]) -> Optional[str]:
|
||
|
|
"""Safely decodes a binary string to unicode"""
|
||
|
|
if isinstance(s, str):
|
||
|
|
return s
|
||
|
|
elif isinstance(s, bytes):
|
||
|
|
return s.decode(defenc, "surrogateescape")
|
||
|
|
elif s is None:
|
||
|
|
return None
|
||
|
|
else:
|
||
|
|
raise TypeError("Expected bytes or text, but got %r" % (s,))
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def safe_encode(s: None) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def safe_encode(s: AnyStr) -> bytes:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
def safe_encode(s: Optional[AnyStr]) -> Optional[bytes]:
|
||
|
|
"""Safely encodes a binary string to unicode"""
|
||
|
|
if isinstance(s, str):
|
||
|
|
return s.encode(defenc)
|
||
|
|
elif isinstance(s, bytes):
|
||
|
|
return s
|
||
|
|
elif s is None:
|
||
|
|
return None
|
||
|
|
else:
|
||
|
|
raise TypeError("Expected bytes or text, but got %r" % (s,))
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def win_encode(s: None) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
@overload
|
||
|
|
def win_encode(s: AnyStr) -> bytes:
|
||
|
|
...
|
||
|
|
|
||
|
|
|
||
|
|
def win_encode(s: Optional[AnyStr]) -> Optional[bytes]:
|
||
|
|
"""Encode unicodes for process arguments on Windows."""
|
||
|
|
if isinstance(s, str):
|
||
|
|
return s.encode(locale.getpreferredencoding(False))
|
||
|
|
elif isinstance(s, bytes):
|
||
|
|
return s
|
||
|
|
elif s is not None:
|
||
|
|
raise TypeError("Expected bytes or text, but got %r" % (s,))
|
||
|
|
return None
|