mirror of
https://github.com/nvbn/thefuck.git
synced 2025-03-27 13:09:00 +00:00
28 lines
620 B
Python
28 lines
620 B
Python
import os
|
|
import sys
|
|
import msvcrt
|
|
import win_unicode_console
|
|
from .. import const
|
|
|
|
|
|
def init_output():
|
|
import colorama
|
|
win_unicode_console.enable()
|
|
colorama.init()
|
|
|
|
|
|
def get_key():
|
|
ch = msvcrt.getch()
|
|
if ch in (b'\x00', b'\xe0'): # arrow or function key prefix?
|
|
ch = msvcrt.getch() # second call returns the actual key code
|
|
|
|
if ch == b'\x03':
|
|
return const.KEY_CTRL_C
|
|
if ch == b'H':
|
|
return const.KEY_UP
|
|
if ch == b'P':
|
|
return const.KEY_DOWN
|
|
|
|
encoding = sys.stdout.encoding or os.environ.get('PYTHONIOENCODING', 'utf-8')
|
|
return ch.decode(encoding)
|