xonstat/util.py

   1 import re
   2 from datetime import datetime
   3
   4 def strip_colors(str=''):
   5     if str is None:
   6         str = ''
   7
   8     str = re.sub(r'\^x\w\w\w', '', str)
   9     str = re.sub(r'\^\d', '', str)
  10     return str
  11
  12
  13 def html_colors(str=''):
  14     if str is None:
  15         str = ''
  16
  17     orig = str
  18     str = re.sub(r'\^x(\w)(\w)(\w)',
  19             "<span style='color:#\g<1>\g<1>\g<2>\g<2>\g<3>\g<3>'>", str)
  20     str = re.sub(r'\^1', "<span style='color:#FF9900'>", str)
  21     str = re.sub(r'\^2', "<span style='color:#33FF00'>", str)
  22     str = re.sub(r'\^3', "<span style='color:#FFFF00'>", str)
  23     str = re.sub(r'\^4', "<span style='color:#3366FF'>", str)
  24     str = re.sub(r'\^5', "<span style='color:#33FFFF'>", str)
  25     str = re.sub(r'\^6', "<span style='color:#FF3366'>", str)
  26     str = re.sub(r'\^7', "<span style='color:#FFFFFF'>", str)
  27     str = re.sub(r'\^8', "<span style='color:#999999'>", str)
  28     str = re.sub(r'\^9', "<span style='color:#666666'>", str)
  29     str = re.sub(r'\^0', "<span style='color:#333333'>", str)
  30
  31     for span in range(len(re.findall(r'\^x\w\w\w|\^\d', orig))):
  32         str += "</span>"
  33
  34     return str
  35
  36
  37 def page_url(page):
  38     return current_route_url(request, page=page, _query=request.GET)
  39
  40
  41 def pretty_date(time=False):
  42     """
  43     Get a datetime object or a int() Epoch timestamp and return a
  44     pretty string like 'an hour ago', 'Yesterday', '3 months ago',
  45     'just now', etc
  46     """
  47     now = datetime.now()
  48     if type(time) is int:
  49         diff = now - datetime.fromtimestamp(time)
  50     elif isinstance(time,datetime):
  51         diff = now - time
  52     elif not time:
  53         diff = now - now
  54     second_diff = diff.seconds
  55     day_diff = diff.days
  56
  57     if day_diff < 0:
  58         return ''
  59
  60     if day_diff == 0:
  61         if second_diff < 10:
  62             return "just now"
  63         if second_diff < 60:
  64             return str(second_diff) + " seconds ago"
  65         if second_diff < 120:
  66             return  "a minute ago"
  67         if second_diff < 3600:
  68             return str( second_diff / 60 ) + " minutes ago"
  69         if second_diff < 7200:
  70             return "an hour ago"
  71         if second_diff < 86400:
  72             return str( second_diff / 3600 ) + " hours ago"
  73     if day_diff == 1:
  74         return "Yesterday"
  75     if day_diff < 7:
  76         return str(day_diff) + " days ago"
  77     if day_diff < 31:
  78         if day_diff/7 == 1:
  79             return "a week ago"
  80         else:
  81             return str(day_diff/7) + " weeks ago"
  82     if day_diff < 365:
  83         if day_diff/30 == 1:
  84             return "a month ago"
  85         else:
  86             return str(day_diff/30) + " months ago"
  87     else:
  88         if day_diff/365 == 1:
  89             return "a year ago"
  90         else:
  91             return str(day_diff/365) + " years ago"