creating some views for the update checker and creating the functions that get the...

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

diff --git a/forum_modules/updater/base.py b/forum_modules/updater/base.py
index 74d37e53287f8e47900981859941a88222e815ce..995c70ce2929ce0d0660faa6693f44671943022c 100644 (file)
--- a/forum_modules/updater/base.py
+++ b/forum_modules/updater/base.py
@@ -1,6 +1,40 @@
 import string
 import random
+import re
+
+import urllib2
+
+from forum.models import Question
+from forum.settings import APP_URL
 
 def generate_installation_key():
     gen = lambda length: "".join( [random.choice(string.digits+string.letters) for i in xrange(length)])
-    return '%s-%s-%s-%s' % (gen(4), gen(4), gen(4), gen(4))
\ No newline at end of file
+    return '%s-%s-%s-%s' % (gen(4), gen(4), gen(4), gen(4))
+
+# To get the site views count we get the SUM of all questions views.
+def get_site_views():
+    views = 0
+
+    # Go through all questions and increase the views count
+    for question in Question.objects.all():
+        views += question.view_count
+
+    return views
+
+def get_server_name():
+    url = '%s/' % APP_URL
+
+    try:
+        # Make the request
+        request = urllib2.Request(url)
+        response = urllib2.urlopen(request)
+
+        # Get the response information
+        response_info = response.info()
+
+        server_name = re.findall("Server: (?P<server_name>.*)$", str(response_info))[0]
+        server_name = ''.join(server_name.splitlines())
+
+        return server_name
+    except:
+        return 'Unknown'