forum_modules/updater/base.py

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