xonstat/util.py

   1 import re
   2 from colorsys import rgb_to_hls, hls_to_rgb
   3 from cgi import escape as html_escape
   4 from datetime import datetime
   5
   6 # Map of special chars to ascii from Darkplace's console.c.
   7 _qfont_table = [
   8  '\0', '#',  '#',  '#',  '#',  '.',  '#',  '#',
   9  '#',  '\t', '\n', '#',  ' ',  '\r', '.',  '.',
  10  '[',  ']',  '0',  '1',  '2',  '3',  '4',  '5',
  11  '6',  '7',  '8',  '9',  '.',  '<',  '=',  '>',
  12  ' ',  '!',  '"',  '#',  '$',  '%',  '&',  '\'',
  13  '(',  ')',  '*',  '+',  ',',  '-',  '.',  '/',
  14  '0',  '1',  '2',  '3',  '4',  '5',  '6',  '7',
  15  '8',  '9',  ':',  ';',  '<',  '=',  '>',  '?',
  16  '@',  'A',  'B',  'C',  'D',  'E',  'F',  'G',
  17  'H',  'I',  'J',  'K',  'L',  'M',  'N',  'O',
  18  'P',  'Q',  'R',  'S',  'T',  'U',  'V',  'W',
  19  'X',  'Y',  'Z',  '[',  '\\', ']',  '^',  '_',
  20  '`',  'a',  'b',  'c',  'd',  'e',  'f',  'g',
  21  'h',  'i',  'j',  'k',  'l',  'm',  'n',  'o',
  22  'p',  'q',  'r',  's',  't',  'u',  'v',  'w',
  23  'x',  'y',  'z',  '{',  '|',  '}',  '~',  '<',
  24
  25  '<',  '=',  '>',  '#',  '#',  '.',  '#',  '#',
  26  '#',  '#',  ' ',  '#',  ' ',  '>',  '.',  '.',
  27  '[',  ']',  '0',  '1',  '2',  '3',  '4',  '5',
  28  '6',  '7',  '8',  '9',  '.',  '<',  '=',  '>',
  29  ' ',  '!',  '"',  '#',  '$',  '%',  '&',  '\'',
  30  '(',  ')',  '*',  '+',  ',',  '-',  '.',  '/',
  31  '0',  '1',  '2',  '3',  '4',  '5',  '6',  '7',
  32  '8',  '9',  ':',  ';',  '<',  '=',  '>',  '?',
  33  '@',  'A',  'B',  'C',  'D',  'E',  'F',  'G',
  34  'H',  'I',  'J',  'K',  'L',  'M',  'N',  'O',
  35  'P',  'Q',  'R',  'S',  'T',  'U',  'V',  'W',
  36  'X',  'Y',  'Z',  '[',  '\\', ']',  '^',  '_',
  37  '`',  'a',  'b',  'c',  'd',  'e',  'f',  'g',
  38  'h',  'i',  'j',  'k',  'l',  'm',  'n',  'o',
  39  'p',  'q',  'r',  's',  't',  'u',  'v',  'w',
  40  'x',  'y',  'z',  '{',  '|',  '}',  '~',  '<'
  41 ]
  42
  43 # Hex-colored spans for decimal color codes ^0 - ^9
  44 _dec_spans = [
  45  "<span style='color:rgb(128,128,128)'>",
  46  "<span style='color:rgb(255,0,0)'>",
  47  "<span style='color:rgb(51,255,0)'>",
  48  "<span style='color:rgb(255,255,0)'>",
  49  "<span style='color:rgb(51,102,255)'>",
  50  "<span style='color:rgb(51,255,255)'>",
  51  "<span style='color:rgb(255,51,102)'>",
  52  "<span style='color:rgb(255,255,255)'>",
  53  "<span style='color:rgb(153,153,153)'>",
  54  "<span style='color:rgb(128,128,128)'>"
  55 ]
  56
  57 # Color code patterns
  58 _all_colors = re.compile(r'\^(\d|x[\dA-Fa-f]{3})')
  59 _dec_colors = re.compile(r'\^(\d)')
  60 _hex_colors = re.compile(r'\^x([\dA-Fa-f])([\dA-Fa-f])([\dA-Fa-f])')
  61
  62 # On a light scale of 0 (black) to 1.0 (white)
  63 _contrast_threshold = 0.5
  64
  65
  66 def qfont_decode(qstr=''):
  67     """ Convert the qfont characters in a string to ascii.
  68     """
  69     if qstr == None:
  70         qstr = ''
  71     chars = []
  72     for c in qstr:
  73         if u'\ue000' <= c <= u'\ue0ff':
  74             c = _qfont_table[ord(c) - 0xe000]
  75         chars.append(c)
  76     return ''.join(chars)
  77
  78
  79 def strip_colors(qstr=''):
  80     if qstr == None:
  81         qstr = ''
  82     return _all_colors.sub('', qstr)
  83
  84
  85 def hex_repl(match):
  86     """Convert Darkplaces hex color codes to CSS rgb.
  87     Brighten colors with HSL light value less than 50%"""
  88
  89     # Extend hex char to 8 bits and to 0.0-1.0 scale
  90     r = int(match.group(1) * 2, 16) / 255.0
  91     g = int(match.group(2) * 2, 16) / 255.0
  92     b = int(match.group(3) * 2, 16) / 255.0
  93
  94     # Check if color is too dark
  95     hue, light, satur = rgb_to_hls(r, g, b)
  96     if light < _contrast_threshold:
  97         light = _contrast_threshold
  98         r, g, b = hls_to_rgb(hue, light, satur)
  99
 100     # Convert back to 0-255 scale for css
 101     return '<span style="color:rgb(%d,%d,%d)">' % (255 * r, 255 * g, 255 * b)
 102
 103
 104 def html_colors(qstr=''):
 105     qstr = html_escape(qfont_decode(qstr).replace('^^', '^'))
 106     html = _dec_colors.sub(lambda match: _dec_spans[int(match.group(1))], qstr)
 107     html = _hex_colors.sub(hex_repl, html)
 108     return html + "</span>" * len(_all_colors.findall(qstr))
 109
 110
 111 def page_url(page):
 112     return current_route_url(request, page=page, _query=request.GET)
 113
 114
 115 def pretty_date(time=False):
 116     """
 117     Get a datetime object or a int() Epoch timestamp and return a
 118     pretty string like 'an hour ago', 'Yesterday', '3 months ago',
 119     'just now', etc
 120     """
 121     now = datetime.utcnow()
 122     if type(time) is int:
 123         diff = now - datetime.fromtimestamp(time)
 124     elif isinstance(time,datetime):
 125         diff = now - time
 126     elif not time:
 127         diff = now - now
 128     second_diff = diff.seconds
 129     day_diff = diff.days
 130
 131     if day_diff < 0:
 132         return ''
 133
 134     if day_diff == 0:
 135         if second_diff < 10:
 136             return "just now"
 137         if second_diff < 60:
 138             return str(second_diff) + " seconds ago"
 139         if second_diff < 120:
 140             return  "a minute ago"
 141         if second_diff < 3600:
 142             return str( second_diff / 60 ) + " minutes ago"
 143         if second_diff < 7200:
 144             return "an hour ago"
 145         if second_diff < 86400:
 146             return str( second_diff / 3600 ) + " hours ago"
 147     if day_diff == 1:
 148         return "Yesterday"
 149     if day_diff < 7:
 150         return str(day_diff) + " days ago"
 151     if day_diff < 31:
 152         if day_diff/7 == 1:
 153             return "a week ago"
 154         else:
 155             return str(day_diff/7) + " weeks ago"
 156     if day_diff < 365:
 157         if day_diff/30 == 1:
 158             return "a month ago"
 159         else:
 160             return str(day_diff/30) + " months ago"
 161     else:
 162         if day_diff/365 == 1:
 163             return "a year ago"
 164         else:
 165             return str(day_diff/365) + " years ago"