Использование GetPass с Python 2 и 3 совместимости

Question

Я знаю, что getpass возвращает строку Юникода в Python 3 и некоторый str в Python 2. Тем не менее, я не уверен насчет кодирования на Python 2.

Как могу ли я преобразовать значение, которое он возвращает в кодированный UTF-8 объект bytes таким образом, который работает как в Python 2, так и в 3?

(я не использую six и я пытаюсь написать код, который просто работает на обоих, а не if Fing на номер версии.)

Answer 1

Я бы следовать EAFP политика в этом случае. Попробуйте декодировать() и закодировать() строку, предполагающую ее среду python 2, и ожидать TypeError, а если вы получите TypeError, просто сделайте encode(). Пример -

import sys 
import getpass 

s = getpass.getpass() 

try: 
    u = s.decode(sys.stdin.encoding).encode('UTF-8') 
except TypeError: 
    u = s.encode('UTF-8') 

Answer 2

В Python 2, getpass() возвращает нормальный Python 2 строки, т.е. байтовая строка. Он использует кодировку терминала, поэтому, если терминал настроен на использование UTF-8, вам не нужно ничего делать - возвращаемая строка уже будет кодированной байтовой строкой UTF-8. Но, чтобы быть надежным, вы можете сделать это:

import sys 
from getpass import getpass 

p = getpass() 
if isinstance(p, bytes): 
    p = p.decode(sys.stdin.encoding) 
p = p.encode('utf-8') # :D 

Мы используем if isinstance(p, bytes): остановить Python 3 от прикосновения строки, возвращаемой getpass(), а затем закодировать Unicode в UTF-8