NewsBlur-viq/utils/log.py

import logging
import re
import string
import time

from django.core.handlers.wsgi import WSGIRequest
from django.conf import settings
from django.utils.encoding import smart_str, smart_str

from .user_functions import extract_user_agent


class NullHandler(logging.Handler):  # exists in python 3.1
    def emit(self, record):
        pass


def getlogger():
    logger = logging.getLogger('newsblur')
    return logger


def user(u, msg, request=None, warn_color=True):
    msg = smart_str(msg)
    if not u:
        return debug(msg)

    platform = '------'
    time_elapsed = ""
    if isinstance(u, WSGIRequest) or request:
        if not request:
            request = u
            u = request.user
        platform = extract_user_agent(request)

        if hasattr(request, 'start_time'):
            seconds = time.time() - request.start_time
            color = '~FB'
            if warn_color:
                if seconds >= 1:
                    color = '~FR'
                elif seconds > .2:
                    color = '~SB~FK'
            time_elapsed = "[%s%.4ss~SB] " % (
                color,
                seconds,
            )
    is_premium = u.is_authenticated and u.profile.is_premium
    premium = '*' if is_premium else ''
    if is_premium and u.profile.is_archive:
        premium = "^"
    username = cipher(str(u)) if settings.CIPHER_USERNAMES else str(u)
    info(' ---> [~FB~SN%-6s~SB] %s[%s%s] %s' % (platform, time_elapsed, username, premium, msg))


def cipher(msg):
    shift = len(msg)
    in_alphabet = str(string.ascii_lowercase)
    out_alphabet = in_alphabet[shift:] + in_alphabet[:shift]
    translation_table = dict((ord(ic), oc) for ic, oc in zip(in_alphabet, out_alphabet))

    return msg.translate(translation_table)


def debug(msg):
    msg = smart_str(msg)
    logger = getlogger()
    logger.debug(colorize(msg))


def info(msg):
    msg = smart_str(msg)
    logger = getlogger()
    logger.info(colorize(msg))


def error(msg):
    msg = smart_str(msg)
    logger = getlogger()
    logger.error(msg)


def colorize(msg):
    params = {
        r'\-\-\->'        : '~FB~SB--->~FW',
        r'\*\*\*>'        : '~FB~SB~BB--->~BT~FW',
        r'\['             : '~SB~FB[~SN~FM',
        r'AnonymousUser'  : '~FBAnonymousUser',
        r'\*\]'           : r'~SN~FR*~FB~SB]',
        r'\^\]'           : r'~SN~FR^~FB~SB]',
        r'\]'             : '~FB~SB]~FW~SN',
    }
    colors = {
        '~SB' : Style.BRIGHT,
        '~SN' : Style.NORMAL,
        '~SK' : Style.BLINK,
        '~SU' : Style.UNDERLINE,
        '~ST' : Style.RESET_ALL,
        '~FK': Fore.BLACK,
        '~FR': Fore.RED,
        '~FG': Fore.GREEN,
        '~FY': Fore.YELLOW,
        '~FB': Fore.BLUE,
        '~FM': Fore.MAGENTA,
        '~FC': Fore.CYAN,
        '~FW': Fore.WHITE,
        '~FT': Fore.RESET,
        '~BK': Back.BLACK,
        '~BR': Back.RED,
        '~BG': Back.GREEN,
        '~BY': Back.YELLOW,
        '~BB': Back.BLUE,
        '~BM': Back.MAGENTA,
        '~BC': Back.CYAN,
        '~BW': Back.WHITE,
        '~BT': Back.RESET,
    }
    for k, v in list(params.items()):
        msg = re.sub(k, v, msg)
    msg = msg + '~ST~FW~BT'
    # msg = re.sub(r'(~[A-Z]{2})', r'%(\1)s', msg)
    for k, v in list(colors.items()):
        msg = msg.replace(k, v)
    return msg
    
'''
This module generates ANSI character codes to printing colors to terminals.
See: http://en.wikipedia.org/wiki/ANSI_escape_code
'''

COLOR_ESC = '\033['

class AnsiCodes(object):
    def __init__(self, codes):
        for name in dir(codes):
            if not name.startswith('_'):
                value = getattr(codes, name)
                setattr(self, name, COLOR_ESC + str(value) + 'm')

class AnsiFore:
    BLACK   = 30
    RED     = 31
    GREEN   = 32
    YELLOW  = 33
    BLUE    = 34
    MAGENTA = 35
    CYAN    = 36
    WHITE   = 37
    RESET   = 39

class AnsiBack:
    BLACK   = 40
    RED     = 41
    GREEN   = 42
    YELLOW  = 43
    BLUE    = 44
    MAGENTA = 45
    CYAN    = 46
    WHITE   = 47
    RESET   = 49

class AnsiStyle:
    BRIGHT    = 1
    DIM       = 2
    UNDERLINE = 4
    BLINK     = 5
    NORMAL    = 22
    RESET_ALL = 0

Fore = AnsiCodes(AnsiFore)
Back = AnsiCodes(AnsiBack)
Style = AnsiCodes(AnsiStyle)