console.py

import sys
import os
import ast


def _get_terminal_size():
    def ioctl_GWINSZ(fd):
        try:
            import fcntl, termios, struct
            return struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
        except:
            pass

    cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)

    if not cr:
        try:
            with open(os.ctermid()) as fd:
                cr = ioctl_GWINSZ(fd)
        except:
            cr = (os.getenv('LINES', 25), os.getenv('COLUMNS', 80))

    return int(cr[1]), int(cr[0])


def supported_chars(*tests):
    """
        Takes any number of strings, and returns the first one
            the terminal encoding supports. If none are supported
            it returns '?' the length of the first string.
    """
    for test in tests:
        try:
            test.encode(sys.stdout.encoding)
            return test
        except UnicodeEncodeError:
            pass
    return '?' * len(tests[0])


def log(*args, trunc=True, m=0):
    if LOGGING and (m in LOGGING_MODES or 0 in LOGGING_MODES):
        args = (str(arg)[:100] + '...' if trunc and len(str(arg)) > 100 else arg for arg in args)
        with open(LOG_FILE, 'a') as f:
            print(*args, file=f)


def in_game_log(string, x, y):
    if IN_GAME_LOGGING:
        print(POS_STR(x, y, string))


def getenv_b(opt):
    """ Converts an enviroment variable into a boolean. """
    o = os.getenv(opt)
    if o:
        return o.lower() in ('true', '1', 'on')


getenv = os.getenv

DEBUG = getenv_b('PYCRAFT_DEBUG')
if os.getenv('PYCRAFT_LOGGING_MODES'):
    LOGGING_MODES = ast.literal_eval(os.getenv('PYCRAFT_LOGGING_MODES'))
else:
    # 0 for all modes
    LOGGING_MODES = [0]

IN_GAME_LOGGING = getenv_b('PYCRAFT_IN_GAME_LOGGING')
LOG_FILE = os.getenv('PYCRAFT_LOG_FILE') or 'pycraft.log'
LOGGING = getenv_b('PYCRAFT_LOGGING')

WIDTH, HEIGHT = _get_terminal_size()
CLS = '\033[2J'
CLS_END = '\033[0J'
CLS_END_LN = '\033[0K'
REDRAW = '\033[0;0f'
HIDE_CUR = '\033[?25l'
SHOW_CUR = '\033[?25h'
POS_STR = lambda x, y, s: '\033[{};{}H{}'.format(y+1, x+1, s)

if LOGGING:
    open(LOG_FILE, 'w').close()