]> de.git.xonotic.org Git - xonotic/xonstat.git/blobdiff - xonstat/util.py
Initial display of the frag matrix.
[xonotic/xonstat.git] / xonstat / util.py
old mode 100755 (executable)
new mode 100644 (file)
index b0b5481..4ed6aee
@@ -1,8 +1,32 @@
+import sys
+import logging
+import pyramid.httpexceptions
+import pyramid.url
 import re
-from datetime import datetime
+from colorsys import rgb_to_hls, hls_to_rgb
+from cgi import escape as html_escape
+from datetime import datetime, timedelta
+from decimal import Decimal
+from collections import namedtuple
+from xonstat.d0_blind_id import d0_blind_id_verify
 
-# Map of special chars to ascii from Quake's console.c.
-qfont_table = [
+
+log = logging.getLogger(__name__)
+
+
+# Map of old weapons codes to new ones
+weapon_map = {
+  "grenadelauncher": "mortar", 
+  "laser": "blaster", 
+  "minstanex": "vaporizer", 
+  "nex": "vortex", 
+  "rocketlauncher": "devastator", 
+  "uzi": "machinegun", 
+}
+
+
+# Map of special chars to ascii from Darkplace's console.c.
+_qfont_ascii_table = [
  '\0', '#',  '#',  '#',  '#',  '.',  '#',  '#',
  '#',  '\t', '\n', '#',  ' ',  '\r', '.',  '.',
  '[',  ']',  '0',  '1',  '2',  '3',  '4',  '5',
@@ -19,7 +43,7 @@ qfont_table = [
  'h',  'i',  'j',  'k',  'l',  'm',  'n',  'o',
  'p',  'q',  'r',  's',  't',  'u',  'v',  'w',
  'x',  'y',  'z',  '{',  '|',  '}',  '~',  '<',
+
  '<',  '=',  '>',  '#',  '#',  '.',  '#',  '#',
  '#',  '#',  ' ',  '#',  ' ',  '>',  '.',  '.',
  '[',  ']',  '0',  '1',  '2',  '3',  '4',  '5',
@@ -38,104 +62,299 @@ qfont_table = [
  'x',  'y',  'z',  '{',  '|',  '}',  '~',  '<'
 ]
 
+_qfont_unicode_glyphs = [
+   u'\u0020',       u'\u0020',       u'\u2014',       u'\u0020',
+   u'\u005F',       u'\u2747',       u'\u2020',       u'\u00B7',
+   u'\U0001F52B',   u'\u0020',       u'\u0020',       u'\u25A0',
+   u'\u2022',       u'\u2192',       u'\u2748',       u'\u2748',
+   u'\u005B',       u'\u005D',       u'\U0001F47D',   u'\U0001F603',
+   u'\U0001F61E',   u'\U0001F635',   u'\U0001F615',   u'\U0001F60A',
+   u'\u00AB',       u'\u00BB',       u'\u2022',       u'\u203E',
+   u'\u2748',       u'\u25AC',       u'\u25AC',       u'\u25AC',
+   u'\u0020',       u'\u0021',       u'\u0022',       u'\u0023',
+   u'\u0024',       u'\u0025',       u'\u0026',       u'\u0027',
+   u'\u0028',       u'\u0029',       u'\u00D7',       u'\u002B',
+   u'\u002C',       u'\u002D',       u'\u002E',       u'\u002F',
+   u'\u0030',       u'\u0031',       u'\u0032',       u'\u0033',
+   u'\u0034',       u'\u0035',       u'\u0036',       u'\u0037',
+   u'\u0038',       u'\u0039',       u'\u003A',       u'\u003B',
+   u'\u003C',       u'\u003D',       u'\u003E',       u'\u003F',
+   u'\u0040',       u'\u0041',       u'\u0042',       u'\u0043',
+   u'\u0044',       u'\u0045',       u'\u0046',       u'\u0047',
+   u'\u0048',       u'\u0049',       u'\u004A',       u'\u004B',
+   u'\u004C',       u'\u004D',       u'\u004E',       u'\u004F',
+   u'\u0050',       u'\u0051',       u'\u0052',       u'\u0053',
+   u'\u0054',       u'\u0055',       u'\u0056',       u'\u0057',
+   u'\u0058',       u'\u0059',       u'\u005A',       u'\u005B',
+   u'\u005C',       u'\u005D',       u'\u005E',       u'\u005F',
+   u'\u0027',       u'\u0061',       u'\u0062',       u'\u0063',
+   u'\u0064',       u'\u0065',       u'\u0066',       u'\u0067',
+   u'\u0068',       u'\u0069',       u'\u006A',       u'\u006B',
+   u'\u006C',       u'\u006D',       u'\u006E',       u'\u006F',
+   u'\u0070',       u'\u0071',       u'\u0072',       u'\u0073',
+   u'\u0074',       u'\u0075',       u'\u0076',       u'\u0077',
+   u'\u0078',       u'\u0079',       u'\u007A',       u'\u007B',
+   u'\u007C',       u'\u007D',       u'\u007E',       u'\u2190',
+   u'\u003C',       u'\u003D',       u'\u003E',       u'\U0001F680',
+   u'\u00A1',       u'\u004F',       u'\u0055',       u'\u0049',
+   u'\u0043',       u'\u00A9',       u'\u00AE',       u'\u25A0',
+   u'\u00BF',       u'\u25B6',       u'\u2748',       u'\u2748',
+   u'\u2772',       u'\u2773',       u'\U0001F47D',   u'\U0001F603',
+   u'\U0001F61E',   u'\U0001F635',   u'\U0001F615',   u'\U0001F60A',
+   u'\u00AB',       u'\u00BB',       u'\u2747',       u'\u0078',
+   u'\u2748',       u'\u2014',       u'\u2014',       u'\u2014',
+   u'\u0020',       u'\u0021',       u'\u0022',       u'\u0023',
+   u'\u0024',       u'\u0025',       u'\u0026',       u'\u0027',
+   u'\u0028',       u'\u0029',       u'\u002A',       u'\u002B',
+   u'\u002C',       u'\u002D',       u'\u002E',       u'\u002F',
+   u'\u0030',       u'\u0031',       u'\u0032',       u'\u0033',
+   u'\u0034',       u'\u0035',       u'\u0036',       u'\u0037',
+   u'\u0038',       u'\u0039',       u'\u003A',       u'\u003B',
+   u'\u003C',       u'\u003D',       u'\u003E',       u'\u003F',
+   u'\u0040',       u'\u0041',       u'\u0042',       u'\u0043',
+   u'\u0044',       u'\u0045',       u'\u0046',       u'\u0047',
+   u'\u0048',       u'\u0049',       u'\u004A',       u'\u004B',
+   u'\u004C',       u'\u004D',       u'\u004E',       u'\u004F',
+   u'\u0050',       u'\u0051',       u'\u0052',       u'\u0053',
+   u'\u0054',       u'\u0055',       u'\u0056',       u'\u0057',
+   u'\u0058',       u'\u0059',       u'\u005A',       u'\u005B',
+   u'\u005C',       u'\u005D',       u'\u005E',       u'\u005F',
+   u'\u0027',       u'\u0041',       u'\u0042',       u'\u0043',
+   u'\u0044',       u'\u0045',       u'\u0046',       u'\u0047',
+   u'\u0048',       u'\u0049',       u'\u004A',       u'\u004B',
+   u'\u004C',       u'\u004D',       u'\u004E',       u'\u004F',
+   u'\u0050',       u'\u0051',       u'\u0052',       u'\u0053',
+   u'\u0054',       u'\u0055',       u'\u0056',       u'\u0057',
+   u'\u0058',       u'\u0059',       u'\u005A',       u'\u007B',
+   u'\u007C',       u'\u007D',       u'\u007E',       u'\u25C0',
+]
 
-def qfont_decode(qstr=''):
-    """
-    Convert the qfont characters in a string to ascii.
+# Hex-colored spans for decimal color codes ^0 - ^9
+_dec_spans = [
+ "<span style='color:rgb(128,128,128)'>",
+ "<span style='color:rgb(255,0,0)'>",
+ "<span style='color:rgb(51,255,0)'>",
+ "<span style='color:rgb(255,255,0)'>",
+ "<span style='color:rgb(51,102,255)'>",
+ "<span style='color:rgb(51,255,255)'>",
+ "<span style='color:rgb(255,51,102)'>",
+ "<span style='color:rgb(255,255,255)'>",
+ "<span style='color:rgb(153,153,153)'>",
+ "<span style='color:rgb(128,128,128)'>"
+]
+
+# Color code patterns
+_all_colors = re.compile(r'\^(\d|x[\dA-Fa-f]{3})')
+_dec_colors = re.compile(r'\^(\d)')
+_hex_colors = re.compile(r'\^x([\dA-Fa-f])([\dA-Fa-f])([\dA-Fa-f])')
+
+# On a light scale of 0 (black) to 1.0 (white)
+_contrast_threshold = 0.5
+
+
+def qfont_decode(qstr='', glyph_translation=False):
+    """ Convert the qfont characters in a string to ascii.
+
+        glyph_translation - determines whether to convert the unicode characters to
+                   their ascii counterparts (if False, the default) or to
+                   the mapped glyph in the Xolonium font (if True).
     """
+    if qstr == None:
+        qstr = ''
     chars = []
     for c in qstr:
-        if c >= u'\ue000' and c <= u'\ue0ff':
-            c = qfont_table[ord(c) - 0xe000]
+        if u'\ue000' <= c <= u'\ue0ff':
+            if glyph_translation:
+                c = _qfont_unicode_glyphs[ord(c) - 0xe000]
+            else:
+                c = _qfont_ascii_table[ord(c) - 0xe000]
         chars.append(c)
     return ''.join(chars)
 
 
-def strip_colors(str=''):
-    if str is None:
-        str = ''
+def strip_colors(qstr=''):
+    if qstr == None:
+        qstr = ''
+    return _all_colors.sub('', qstr)
+
+
+def hex_repl(match):
+    """Convert Darkplaces hex color codes to CSS rgb.
+    Brighten colors with HSL light value less than 50%"""
+
+    # Extend hex char to 8 bits and to 0.0-1.0 scale
+    r = int(match.group(1) * 2, 16) / 255.0
+    g = int(match.group(2) * 2, 16) / 255.0
+    b = int(match.group(3) * 2, 16) / 255.0
+
+    # Check if color is too dark
+    hue, light, satur = rgb_to_hls(r, g, b)
+    if light < _contrast_threshold:
+        light = _contrast_threshold
+        r, g, b = hls_to_rgb(hue, light, satur)
 
-    str = re.sub(r'\^x\w\w\w', '', str)
-    str = re.sub(r'\^\d', '', str)
-    return str
+    # Convert back to 0-255 scale for css
+    return '<span style="color:rgb(%d,%d,%d)">' % (255 * r, 255 * g, 255 * b)
 
 
-def html_colors(str=''):
-    if str is None:
-        str = ''
+def html_colors(qstr='', limit=None):
+    qstr = html_escape(qfont_decode(qstr, glyph_translation=True))
+    qstr = qstr.replace('^^', '^')
 
-    orig = str
-    str = re.sub(r'\^x(\w)(\w)(\w)', 
-            "<span style='color:#\g<1>\g<1>\g<2>\g<2>\g<3>\g<3>'>", str)
-    str = re.sub(r'\^1', "<span style='color:#FF9900'>", str)
-    str = re.sub(r'\^2', "<span style='color:#33FF00'>", str)
-    str = re.sub(r'\^3', "<span style='color:#FFFF00'>", str)
-    str = re.sub(r'\^4', "<span style='color:#3366FF'>", str)
-    str = re.sub(r'\^5', "<span style='color:#33FFFF'>", str)
-    str = re.sub(r'\^6', "<span style='color:#FF3366'>", str)
-    str = re.sub(r'\^7', "<span style='color:#FFFFFF'>", str)
-    str = re.sub(r'\^8', "<span style='color:#999999'>", str)
-    str = re.sub(r'\^9', "<span style='color:#666666'>", str)
-    str = re.sub(r'\^0', "<span style='color:#333333'>", str)
+    if limit is not None and limit > 0:
+        qstr = limit_printable_characters(qstr, limit)
 
-    for span in range(len(re.findall(r'\^x\w\w\w|\^\d', orig))):
-        str += "</span>"
+    html = _dec_colors.sub(lambda match: _dec_spans[int(match.group(1))], qstr)
+    html = _hex_colors.sub(hex_repl, html)
+    return html + "</span>" * len(_all_colors.findall(qstr))
 
-    return str
+
+def limit_printable_characters(qstr, limit):
+    # initialize assuming all printable characters
+    pc = [1 for i in range(len(qstr))]
+
+    groups = _all_colors.finditer(qstr)
+    for g in groups:
+        pc[g.start():g.end()] = [0 for i in range(g.end() - g.start())]
+
+    # printable characters in the string is less than or equal to what was requested
+    if limit >= len(qstr) or sum(pc) <= limit:
+        return qstr
+    else:
+        sumpc = 0
+        for i,v in enumerate(pc):
+            sumpc += v
+            if sumpc == limit:
+                return qstr[0:i+1]
 
 
 def page_url(page):
-    return current_route_url(request, page=page, _query=request.GET)
+    return pyramid.url.current_route_url(request, page=page, _query=request.GET)
 
 
 def pretty_date(time=False):
-    """
-    Get a datetime object or a int() Epoch timestamp and return a
-    pretty string like 'an hour ago', 'Yesterday', '3 months ago',
-    'just now', etc
-    """
-    now = datetime.now()
+    '''Returns a human-readable relative date.'''
+    now = datetime.utcnow()
     if type(time) is int:
         diff = now - datetime.fromtimestamp(time)
     elif isinstance(time,datetime):
-        diff = now - time 
+        diff = now - time
     elif not time:
+        print "not a time value"
         diff = now - now
-    second_diff = diff.seconds
-    day_diff = diff.days
-
-    if day_diff < 0:
-        return ''
-
-    if day_diff == 0:
-        if second_diff < 10:
-            return "just now"
-        if second_diff < 60:
-            return str(second_diff) + " seconds ago"
-        if second_diff < 120:
-            return  "a minute ago"
-        if second_diff < 3600:
-            return str( second_diff / 60 ) + " minutes ago"
-        if second_diff < 7200:
-            return "an hour ago"
-        if second_diff < 86400:
-            return str( second_diff / 3600 ) + " hours ago"
-    if day_diff == 1:
-        return "Yesterday"
-    if day_diff < 7:
-        return str(day_diff) + " days ago"
-    if day_diff < 31:
-        if day_diff/7 == 1:
-            return "a week ago"
-        else:
-            return str(day_diff/7) + " weeks ago"
-    if day_diff < 365:
-        if day_diff/30 == 1:
-            return "a month ago"
-        else:
-            return str(day_diff/30) + " months ago"
+
+    dim = round(diff.seconds/60.0 + diff.days*1440.0)
+
+    if dim == 0:
+        return "less than a minute ago"
+    elif dim == 1:
+        return "1 minute ago"
+    elif dim >= 2 and dim <= 44:
+        return "{0} minutes ago".format(int(dim))
+    elif dim >= 45 and dim <= 89:
+        return "about 1 hour ago"
+    elif dim >= 90 and dim <= 1439:
+        return "about {0} hours ago".format(int(round(dim/60.0)))
+    elif dim >= 1440 and dim <= 2519:
+        return "1 day ago"
+    elif dim >= 2520 and dim <= 43199:
+        return "{0} days ago".format(int(round(dim/1440.0)))
+    elif dim >= 43200 and dim <= 86399:
+        return "about 1 month ago"
+    elif dim >= 86400 and dim <= 525599:
+        return "{0} months ago".format(int(round(dim/43200.0)))
+    elif dim >= 525600 and dim <= 655199:
+        return "about 1 year ago"
+    elif dim >= 655200 and dim <= 914399:
+        return "over 1 year ago"
+    elif dim >= 914400 and dim <= 1051199:
+        return "almost 2 years ago"
     else:
-        if day_diff/365 == 1:
-            return "a year ago"
+        return "about {0} years ago".format(int(round(dim/525600.0)))
+
+def datetime_seconds(td):
+    return float(td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
+
+def to_json(data):
+    if not type(data) == dict:
+        # assume it's a named tuple
+        data = data._asdict()
+    result = {}
+    for key,value in data.items():
+        if value == None:
+            result[key] = None
+        elif type(value) in [bool,int,long,float,complex,str]:
+            result[key] = value
+        elif type(value) == unicode:
+            result[key] = value.encode('utf-8')
+        elif type(value) == Decimal:
+            result[key] = float(value)
+        elif type(value) == datetime:
+            result[key] = value.strftime('%Y-%m-%dT%H:%M:%SZ')
+        elif type(value) == timedelta:
+            result[key] = datetime_seconds(value)
         else:
-            return str(day_diff/365) + " years ago"
+            result[key] = to_json(value.to_dict())
+    return result
+
+
+def is_leap_year(today_dt=None):
+    if today_dt is None:
+        today_dt = datetime.utcnow()
+
+    if today_dt.year % 400 == 0:
+       leap_year = True
+    elif today_dt.year % 100 == 0:
+       leap_year = False
+    elif today_dt.year % 4 == 0:
+       leap_year = True
+    else:
+       leap_year = False
+
+    return leap_year
+
+
+def is_cake_day(create_dt, today_dt=None):
+    cake_day = False
+
+    if today_dt is None:
+        today_dt = datetime.utcnow()
+
+    # cakes are given on the first anniversary, not the actual create date!
+    if datetime.date(today_dt) != datetime.date(create_dt):
+        if today_dt.day == create_dt.day and today_dt.month == create_dt.month:
+            cake_day = True
+
+        # leap year people get their cakes on March 1
+        if not is_leap_year(today_dt) and create_dt.month == 2 and create_dt.day == 29:
+            if today_dt.month == 3 and today_dt.day == 1:
+                cake_day = True
+
+    return cake_day
+
+
+def verify_request(request):
+    """Verify requests using the d0_blind_id library"""
+
+    # first determine if we should be verifying or not
+    val_verify_requests = request.registry.settings.get('xonstat.verify_requests', 'true')
+    if val_verify_requests == "true":
+        flg_verify_requests = True
+    else:
+        flg_verify_requests = False
+
+    try:
+        (idfp, status) = d0_blind_id_verify(
+                sig=request.headers['X-D0-Blind-Id-Detached-Signature'],
+                querystring='',
+                postdata=request.body)
+    except:
+        log.debug('ERROR: Could not verify request: {0}'.format(sys.exc_info()))
+        idfp = None
+        status = None
+
+    if flg_verify_requests and not idfp:
+        log.debug("ERROR: Unverified request")
+        raise pyramid.httpexceptions.HTTPUnauthorized("Unverified request")
+
+    return (idfp, status)