forum_modules/updates/base.py

   1 import os
   2 import sys
   3 import platform
   4 import bz2
   5 import urllib2, urllib
   6 import binascii
   7 import string
   8 import random
   9 import re
  10 import urllib2
  11 import settings
  12 import datetime
  13 import logging
  14
  15
  16 from xml.dom.minidom import parse, parseString
  17 from forum.base import get_database_engine
  18 from forum.models import Question, Answer, Comment, User
  19 from forum.settings import APP_URL, SVN_REVISION, APP_TITLE, APP_DESCRIPTION
  20 from django import VERSION as DJANGO_VERSION
  21 from django.utils import simplejson
  22 from django.utils.html import escape
  23 from django.utils.encoding import smart_unicode
  24 from django.conf import settings as django_settings
  25 from django.utils.translation import ugettext as _
  26
  27
  28 def generate_installation_key():
  29     gen = lambda length: "".join( [random.choice(string.digits+string.letters) for i in xrange(length)])
  30     return '%s-%s-%s-%s' % (gen(4), gen(4), gen(4), gen(4))
  31
  32 # To get the site views count we get the SUM of all questions views.
  33 def get_site_views():
  34     views = 0
  35
  36     # Go through all questions and increase the views count
  37     for question in Question.objects.all():
  38         views += question.view_count
  39
  40     return views
  41
  42 # Gets the active users count since the last visit
  43 def get_active_users():
  44     users_count = 0
  45
  46     try:
  47         if settings.LATEST_UPDATE_DATETIME:
  48             users_count = User.objects.filter(last_login__gt=settings.LATEST_UPDATE_DATETIME).count()
  49     except:
  50         pass
  51
  52     return users_count
  53
  54 def get_server_name():
  55     url = '%s/' % APP_URL
  56
  57     try:
  58         # Make the request
  59         request = urllib2.Request(url)
  60         response = urllib2.urlopen(request)
  61
  62         # Get the response information
  63         response_info = response.info()
  64
  65         server_name = re.findall("Server: (?P<server_name>.*)$", str(response_info))[0]
  66         server_name = ''.join(server_name.splitlines())
  67
  68         return server_name
  69     except:
  70         return 'Unknown'
  71
  72 def get_admin_emails():
  73     emails = []
  74
  75     for user in User.objects.filter(is_superuser=True):
  76         emails.append(user.email)
  77
  78     return emails
  79
  80 def check_for_updates():
  81     # Get the SVN Revision
  82     try:
  83         svn_revision = int(SVN_REVISION.replace('SVN-', ''))
  84     except ValueError:
  85         # Here we'll have to find another way of getting the SVN revision
  86         svn_revision = 0
  87
  88     admin_emails_xml = '<emails>'
  89     for email in get_admin_emails():
  90         admin_emails_xml += '<email value="%s" />' % email
  91     admin_emails_xml += '</emails>'
  92
  93     database_type = get_database_engine()
  94
  95     statistics = """<check>
  96     <key value="%(site_key)s" />
  97     <app_url value="%(app_url)s" />
  98     <app_title value="%(app_title)s" />
  99     <app_description value="%(app_description)s" />
 100     <svn_revision value="%(svn_revision)d" />
 101     <views value="%(site_views)d" />
 102     <questions_count value="%(questions_count)d" />
 103     <answers_count value="%(answers_count)d" />
 104     <comments_count value="%(comments_count)d" />
 105     <active_users value="%(active_users)d" />
 106     <server value="%(server_name)s" />
 107     <python_version value="%(python_version)s" />
 108     <django_version value="%(django_version)s" />
 109     <database value="%(database)s" />
 110     <os value="%(os)s" />
 111     %(emails)s
 112 </check> """ % {
 113         'site_key' : settings.SITE_KEY,
 114         'app_url' : APP_URL,
 115         'app_title' : escape(APP_TITLE.value),
 116         'app_description' : escape(APP_DESCRIPTION.value),
 117         'svn_revision' : svn_revision,
 118         'site_views' : get_site_views(),
 119         'server_name' : get_server_name(),
 120         'questions_count' : Question.objects.filter_state(deleted=False).count(),
 121         'answers_count' : Answer.objects.filter_state(deleted=False).count(),
 122         'comments_count' : Comment.objects.filter_state(deleted=False).count(),
 123         'active_users' : get_active_users(),
 124         'python_version' : ''.join(sys.version.splitlines()),
 125         'django_version' : str(DJANGO_VERSION),
 126         'database' : database_type,
 127         'os' : str(platform.uname()),
 128         'emails' : admin_emails_xml,
 129     }
 130
 131     # Compress the statistics XML dump
 132     statistics_compressed = bz2.compress(statistics)
 133
 134     # Pass the compressed statistics to the update server
 135     post_data = {
 136         'statistics' : binascii.b2a_base64(statistics_compressed),
 137     }
 138     data = urllib.urlencode(post_data)
 139
 140     # We simulate some browser, otherwise the server can return 403 response
 141     user_agent = 'Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_5_8; en-us) AppleWebKit/531.21.8 (KHTML, like Gecko) Version/4.0.4 Safari/5'
 142     headers={ 'User-Agent' : user_agent,}
 143
 144     try:
 145         check_request = urllib2.Request('%s%s' % (settings.UPDATE_SERVER_URL, '/site_check/'), data, headers=headers)
 146         check_response = urllib2.urlopen(check_request)
 147         content = check_response.read()
 148     except urllib2.HTTPError, error:
 149         content = error.read()
 150     except:
 151         return _("Wasn't able to check to the update server.")
 152
 153     # Read the messages from the Update Server
 154     try:
 155         messages_xml_url = '%s%s' % (settings.UPDATE_SERVER_URL, '/messages/xml/')
 156         messages_request = urllib2.Request(messages_xml_url, headers=headers)
 157         messages_response = urllib2.urlopen(messages_request)
 158         messages_xml = messages_response.read()
 159     except:
 160         return _("Wasn't able to retreive the update messages.")
 161
 162     # Store the messages XML in a Setting object
 163     settings.UPDATE_MESSAGES_XML.set_value(messages_xml)
 164
 165     messages_dom = parseString(messages_xml)
 166     messages_count = len(messages_dom.getElementsByTagName('message'))
 167
 168     # Set the latest update datetime to now.
 169     now = datetime.datetime.now()
 170     settings.LATEST_UPDATE_DATETIME.set_value(now)
 171
 172     return _('%d update messages have been downloaded.') % messages_count
 173
 174 def update_trigger():
 175     # Trigger the update process
 176     now = datetime.datetime.now()
 177     if (now - settings.LATEST_UPDATE_DATETIME) > datetime.timedelta(days=1):
 178         update_status = check_for_updates()
 179
 180         logging.error(smart_unicode("Update process has been triggered: %s" % update_status))
 181         settings.LATEST_UPDATE_DATETIME.set_value(now)