Reports - user stats update

sitescripts/reports/utils.py

 # coding: utf-8
 
 # This Source Code is subject to the terms of the Mozilla Public License
 # version 2.0 (the "License"). You can obtain a copy of the License at
 # http://mozilla.org/MPL/2.0/.
 
 import hashlib, hmac, base64, MySQLdb, os, re, marshal, subprocess
 from sitescripts.utils import get_config, cached, get_template, sendMail
 
 def getReportSubscriptions(guid):
@@ skipping 81 matching lines @@
 
   reportData['guid'] = guid
   file = os.path.join(get_config().get('reports', 'dataPath'), guid[0], guid[1], guid[2], guid[3], guid + '.html')
   dir = os.path.dirname(file)
   if not os.path.exists(dir):
     os.makedirs(dir)
   template = get_template(get_config().get('reports', 'webTemplate'))
   template.stream(reportData).dump(file, encoding='utf-8')
 
 def removeReport(guid):
+  reportData = getReport(guid)    
+  if (reportData.get('email', None)):
+    utility = reportData['utility']
+    contact = getUserId(reportData['email'])
+    updateUserUtility(contact, utility)

[Wladimir Palant, 2012/10/18 15:45:52]

So you are updating user's score only after 30 days? I see how this approach is
simpler - but that's quite a bit of delay. I think that updating immediately is
preferable.

+
   cursor = get_db().cursor()
   executeQuery(cursor,
          '''DELETE FROM #PFX#reports WHERE guid = %s''',
          (guid))
   get_db().commit()
   file = os.path.join(get_config().get('reports', 'dataPath'), guid[0], guid[1], guid[2], guid[3], guid + '.html')
   if os.path.isfile(file):
     os.remove(file)
   file = os.path.join(get_config().get('reports', 'dataPath'), guid[0], guid[1], guid[2], guid[3], guid + '.png')
   if os.path.isfile(file):
     os.remove(file)
 
+def updateUserUtility(contact, utility):
+  if utility == 0:
+    return
+  cursor = get_db().cursor()
+  if utility > 0:
+    executeQuery(cursor,
+                '''UPDATE #PFX#users SET positive = positive + 1 WHERE id = %s''',
+                (contact))
+  else:
+    executeQuery(cursor,
+                '''UPDATE #PFX#users SET negative = negative + 1 WHERE id = %s''',
+                (contact))
+  get_db().commit()
+
 def saveScreenshot(guid, screenshot):
   prefix = 'data:image/png;base64,'
   if not screenshot.startswith(prefix):
     raise TypeError('Screenshot is not a PNG image')
   data = base64.b64decode(screenshot[len(prefix):])
   file = os.path.join(get_config().get('reports', 'dataPath'), guid[0], guid[1], guid[2], guid[3], guid + '.png')
   dir = os.path.dirname(file)
   if not os.path.exists(dir):
     os.makedirs(dir)
   f = open(file, 'wb')
@@ skipping 51 matching lines @@
   dbpasswd = get_config().get('reports', 'dbpassword')
   if os.name == 'nt':
     return MySQLdb.connect(user=dbuser, passwd=dbpasswd, db=database, use_unicode=True, charset='utf8', named_pipe=True)
   else:
     return MySQLdb.connect(user=dbuser, passwd=dbpasswd, db=database, use_unicode=True, charset='utf8')
     
 def executeQuery(cursor, query, args=None):
   tablePrefix = get_config().get('reports', 'dbprefix')
   query = re.sub(r'#PFX#', tablePrefix, query)
   cursor.execute(query, args)