]> git.xonotic.org Git - xonotic/xonstat.git/blob - xonstat/util.py
de5153a485069860eb070d09efe022fad6f9e56b
[xonotic/xonstat.git] / 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:#333333'>",
46  "<span style='color:#FF0000'>",
47  "<span style='color:#33FF00'>",
48  "<span style='color:#FFFF00'>",
49  "<span style='color:#3366FF'>",
50  "<span style='color:#33FFFF'>",
51  "<span style='color:#FF3366'>",
52  "<span style='color:#FFFFFF'>",
53  "<span style='color:#999999'>",
54  "<span style='color:#666666'>"
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 hex to 8 bits and to 0.0-1.0 scale
87     r = int(match.group(1) * 2, 16) / 255.
88     g = int(match.group(2) * 2, 16) / 255.
89     b = int(match.group(3) * 2, 16) / 255.
90     hue, light, satur = rgb_to_hls(r, g, b)
91     if light < _contrast_threshold:
92         light = _contrast_threshold
93     # Get new rgb in 0-255 scale
94     r, g, b = tuple([int(round(255 * i)) for i in hls_to_rgb(hue, light, satur)])
95     return '<span style="color:rgb({0},{1},{2})">'.format(r, g, b)
96
97
98 def html_colors(qstr=''):
99     qstr = html_escape(qfont_decode(qstr).replace('^^', '^'))
100     html = _dec_colors.sub(lambda match: _dec_spans[int(match.group(1))], qstr)
101     html = _hex_colors.sub(hex_repl, html)
102     return html + "</span>" * len(_all_colors.findall(qstr))
103
104
105 def page_url(page):
106     return current_route_url(request, page=page, _query=request.GET)
107
108
109 def pretty_date(time=False):
110     """
111     Get a datetime object or a int() Epoch timestamp and return a
112     pretty string like 'an hour ago', 'Yesterday', '3 months ago',
113     'just now', etc
114     """
115     now = datetime.now()
116     if type(time) is int:
117         diff = now - datetime.fromtimestamp(time)
118     elif isinstance(time,datetime):
119         diff = now - time 
120     elif not time:
121         diff = now - now
122     second_diff = diff.seconds
123     day_diff = diff.days
124
125     if day_diff < 0:
126         return ''
127
128     if day_diff == 0:
129         if second_diff < 10:
130             return "just now"
131         if second_diff < 60:
132             return str(second_diff) + " seconds ago"
133         if second_diff < 120:
134             return  "a minute ago"
135         if second_diff < 3600:
136             return str( second_diff / 60 ) + " minutes ago"
137         if second_diff < 7200:
138             return "an hour ago"
139         if second_diff < 86400:
140             return str( second_diff / 3600 ) + " hours ago"
141     if day_diff == 1:
142         return "Yesterday"
143     if day_diff < 7:
144         return str(day_diff) + " days ago"
145     if day_diff < 31:
146         if day_diff/7 == 1:
147             return "a week ago"
148         else:
149             return str(day_diff/7) + " weeks ago"
150     if day_diff < 365:
151         if day_diff/30 == 1:
152             return "a month ago"
153         else:
154             return str(day_diff/30) + " months ago"
155     else:
156         if day_diff/365 == 1:
157             return "a year ago"
158         else:
159             return str(day_diff/365) + " years ago"