display the XML messages set to the site administrators

[osqa.git] / forum_modules / updater / views.py

diff --git a/forum_modules/updater/views.py b/forum_modules/updater/views.py
index 33b13b23c06898441ad8d4779a4877ab93ef083c..cc469148d3c2b55f2a3ce7887f73aebe61a5e0f6 100644 (file)
--- a/forum_modules/updater/views.py
+++ b/forum_modules/updater/views.py
@@ -1,6 +1,99 @@
+import os
+import sys
+import bz2
+import urllib2, urllib
+import binascii
+
+from xml.dom.minidom import parse, parseString
+
+from django import VERSION as DJANGO_VERSION
 from django.http import HttpResponse
-from base import generate_installation_key
-from settings import SITE_KEY
+from django.utils.translation import ugettext as _
+from django.utils import simplejson
+from django.conf import settings
 
+from base import get_site_views, get_server_name, get_admin_emails
+import settings as updater_settings
+from forum.settings import APP_URL, SVN_REVISION
+from forum.views.admin import admin_tools_page, admin_page
+
+@admin_tools_page(_('updater'), _('Update Checker'))
 def updater_index(request):
-    return HttpResponse(str(SITE_KEY))
+    return (
+        'modules/updater/index.html',
+        {
+
+        },
+    )
+
+def updater_check(request):
+    # Get the SVN Revision
+    try:
+        svn_revision = int(SVN_REVISION.replace('SVN-', ''))
+    except ValueError:
+        # Here we'll have to find another way of getting the SVN revision
+        svn_revision = 0
+
+    admin_emails_xml = '<emails>'
+    for email in get_admin_emails():
+        admin_emails_xml += '<email value="%s" />' % email
+    admin_emails_xml += '</emails>'
+
+    statistics = """<check>
+    <key value="%(site_key)s" />
+    <app_url value="%(app_url)s" />
+    <svn_revision value="%(svn_revision)d" />
+    <views value="%(site_views)d" />
+    <active_users value="11959" />
+    <server value="%(server_name)s" />
+    <python_version value="%(python_version)s" />
+    <django_version value="%(django_version)s" />
+    <database value="%(database)s" />
+    <os value="%(os)s" />
+    %(emails)s
+</check> """ % {
+        'site_key' : updater_settings.SITE_KEY,
+        'app_url' : APP_URL,
+        'svn_revision' : svn_revision,
+        'site_views' : get_site_views(),
+        'server_name' : get_server_name(),
+        'python_version' : ''.join(sys.version.splitlines()),
+        'django_version' : str(DJANGO_VERSION),
+        'database' : settings.DATABASE_ENGINE,
+        'os' : str(os.uname()),
+        'emails' : admin_emails_xml,
+    }
+
+    # Compress the statistics XML dump
+    statistics_compressed = bz2.compress(statistics)
+
+    # Pass the compressed statistics to the update server
+    post_data = {
+        'statistics' : binascii.b2a_base64(statistics_compressed),
+    }
+    data = urllib.urlencode(post_data)
+
+    # We simulate some browser, otherwise the server can return 403 response
+    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'
+    headers={ 'User-Agent' : user_agent,}
+
+    try:
+        check_request = urllib2.Request('%s%s' % (updater_settings.UPDATE_SERVER_URL, '/site_check/'), data, headers=headers)
+        check_response = urllib2.urlopen(check_request)
+        content = check_response.read()
+    except urllib2.HTTPError, error:
+        content = error.read()
+
+    # Read the messages from the Update Server
+    messages_xml_url = '%s%s' % (updater_settings.UPDATE_SERVER_URL, '/messages/xml/')
+    messages_request = urllib2.Request(messages_xml_url, headers=headers)
+    messages_response = urllib2.urlopen(messages_request)
+    messages_xml = messages_response.read()
+
+    # Store the messages XML in a Setting object
+    updater_settings.UPDATE_MESSAGES_XML.set_value(messages_xml)
+
+    messages_dom = parseString(messages_xml)
+    messages_count = len(messages_dom.getElementsByTagName('message'))
+
+    return HttpResponse(_('%d update messages have been downloaded') % messages_count, mimetype='text/html')
\ No newline at end of file