#! /usr/bin/python
import wikipedia, pagegenerators
import datetime, MySQLdb

import cgitb; cgitb.enable(logdir='/home/dispenser/public_html/cgi-bin/logs/', context=20)

def colorValue(subset, total, revert = False):
    v = min(255 * subset / total, 255)
    if (revert):
        v = 255 - v;
    blue = 00
    if (v < 128):
        # Red to Yellow
        red = 255
        green = 2 * v
    else:
        # Yellow to Green
        red = 2 * (255 - v)
        green = 255
    return "#%02X%02X%02X"  % (red, green, blue)

class UserActivity:
	def __init__(self):
		self.page = None
		self.days = 240
	def main(self):
		site = wikipedia.getSite()
		action = 0
		for arg in wikipedia.handleArgs():
			if arg.startswith('-page:'):
				self.page = wikipedia.Page(site, arg[6:])
				site = self.page.site()
			elif arg.startswith('-inactivetime:'):
				pass
			elif arg.startswith('-action:edit'):
				action = 2
			elif arg.startswith('-days:'):
				self.days = int(arg[6:])

	#	print '<script language="JavaScript" type="text/javascript">wgBreakFrames=false; stylpath=""</script>'
	#	print '<script language="JavaScript" type="text/javascript" src="http://en.wikipedia.org/skins-1.5/common/wikibits.js"></script>'
		print '<form action="../cgi-bin/useractivity.py" style="text-align:left;">'
		print '<fieldset><legend>Member activity check</legend>'
		print '<p><label for="page">Member list page</label>:  <input type="text" name="page" id="page" size="50" value="%s" onchange="fixTitle(this)" /><br/>' % (self.page and self.page.title() or '')
		print '<label for="days">Inactive for </label><input type="text" name="days" id="days" size="3" value="%s" /> days<br/>'%self.days
		print '<select name="action"><option value="table">Display table</option><option value="edit">Prepare update</option></select><br/>'
		print '<input type="submit" value="Check list of users" /><br/>'
		#print 'Example: <a href="/~dispenser/cgi-bin/useractivity.py/Project:WikiProject_Trains/Members">WikiPorject Trains/Members</a></p>'
		print '</fieldset></form>'

		if not self.page:
			return
	
		page_title	= self.page.title(underscore=True)
		if page_title.lower().startswith('wikipedia:'):
			page_title=page_title[10:]
		elif page_title.lower().startswith('project:'):
			page_title=page_title[8:]
		
		db = MySQLdb.connect(db=site.dbName()+'_p', host=site.dbName().replace('_', '-') + '-p.db.toolserver.org', read_default_file="/home/dispenser/.my.cnf")

		self.conn = db.cursor()
		self.conn.execute("SET SESSION TRANSACTION ISOLATION LEVEL READ UNCOMMITTED")
		self.conn.execute("""
SELECT user_id, user_name, user_editcount, TIMESTAMP(user_registration)
FROM page 
RIGHT JOIN pagelinks ON page_id=pl_from 
RIGHT JOIN user ON user_name=replace(pl_title,"_"," ") 
WHERE pl_namespace=2 
AND page_namespace=4 AND page_title=%s;""", (page_title,) )
		if action == 0:
			self.infoTable()
		else:
			self.updatePage()
		self.conn.close()

	def updatePage(self):
		import re
		text = self.page.get()
		msg_commented = 0
		msg_removed = 0
		msg_renamed = 0


		for (user_id, user_name, user_editcount, user_registration) in self.conn.fetchall():
			self.conn.execute("SELECT rd_title FROM page JOIN redirect ON page_id=rd_from WHERE rd_namespace=2 AND page_namespace=2 AND page_title=replace(%s,' ','_')", (user_name))
			user_rd = self.conn.fetchone()
			self.conn.execute("SELECT TIMESTAMP(rev_timestamp), TIMESTAMPDIFF(DAY,rev_timestamp,now()) FROM revision WHERE rev_user=%s ORDER BY rev_timestamp DESC LIMIT 1", (user_id,))
			lasteditdate = self.conn.fetchone()
			p = re.compile(r'(?mi)(^[#*].*(?:\[\[|\{\{)(?:user)[:|] *)(%s)( *(?=[]{|}[]).*)' % re.escape(user_name).replace('\ ', '[_ ]'))

			# List update logic
			if   user_editcount <= 20 and user_rd:
					wikipedia.output('Renaming [[User:%s]] to [[User:%s]]'%(user_name, user_rd[0]))
					msg_renamed += 1
					text = p.sub(r'\g<1>%s\g<3>'%user_rd[0], text)
			elif user_editcount == 1 and lasteditdate and lasteditdate[1] > 180:
					wikipedia.output('Commenting [[User:%s]] single edit account (%d months inactive)'%(user_name, lasteditdate[1]/30))
					msg_commented += 1
					text = p.sub(r'<!-- \g<0> :: single edit account -->', text)
			elif user_editcount == 0:
				if not lasteditdate:
					wikipedia.output('Removing [[User:%s]] - [[wp:vanish]]ed'%user_name)
					msg_removed += 1
					text = p.sub(r'', text)
				else:
					wikipedia.output('Bad edit info for [[User:%s]]'%user_name)
			elif lasteditdate and lasteditdate[1] > self.days and lasteditdate[1]/user_editcount >= 7:
				wikipedia.output('Commenting [[User:%s]] (%d edits, %d months inactive)' % (user_name, user_editcount, lasteditdate and lasteditdate[1]/30, ))
				msg_commented += 1
				text = p.sub(lasteditdate[0].strftime(r"<!-- \g<0> :: Inactive since %b %Y -->"), text)
			else:
				pass

		wikipedia.showDiff(self.page.get(), text)
		wikipedia.setAction(self.summary(parts={'removed':msg_removed, 'renamed':msg_renamed, 'commented out':msg_commented}))
		self.page.put(text)
	
	def summary(self, parts):
		msg = []

		wordform = ('user ', 'users ')
		for (key, value) in parts.iteritems():
			if key and value:
				if value == 1:
					msg += ["%s%s"%(wordform[0], key)]
				else:
					msg += ["%d %s%s"%(value, wordform[1], key)]
		if len(msg) <= 2:
			return ' and '.join(msg)
		else:
			return ', '.join(msg[:-1]) + ', and '+msg[-1]

	def infoTable(self):
		print '<p>It is suggested that you used the <a href="https://www.squarefree.com/bookmarklets/pagedata.html" class="external">sort table bookmarklet</a> until I get sorting implemented</p>'
		print '''<p><a href="javascript:function toArray (c){var a, k;a=new Array;for (k=0; k&lt;c.length; ++k)a[k]=c[k];return a;}function insAtTop(par,child){if(par.childNodes.length) par.insertBefore(child, par.childNodes[0]);else par.appendChild(child);}function countCols(tab){var nCols, i;nCols=0;for(i=0;i&lt;tab.rows.length;++i)if(tab.rows[i].cells.length&gt;nCols)nCols=tab.rows[i].cells.length;return nCols;}function makeHeaderLink(tableNo, colNo, ord){var link;link=document.createElement('a');link.href='javascript:sortTable('+tableNo+','+colNo+','+ord+');';link.appendChild(document.createTextNode((ord&gt;0)?'a':'d'));return link;}function makeHeader(tableNo,nCols){var header, headerCell, i;header=document.createElement('tr');for(i=0;i&lt;nCols;++i){headerCell=document.createElement('td');headerCell.appendChild(makeHeaderLink(tableNo,i,1));headerCell.appendChild(document.createTextNode('/'));headerCell.appendChild(makeHeaderLink(tableNo,i,-1));header.appendChild(headerCell);}return header;}g_tables=toArray(document.getElementsByTagName('table'));if(!g_tables.length) alert(&quot;This page doesn't contain any tables.&quot;);(function(){var j, thead;for(j=0;j&lt;g_tables.length;++j){thead=g_tables[j].createTHead();insAtTop(thead, makeHeader(j,countCols(g_tables[j])))}}) ();function compareRows(a,b){if(a.sortKey==b.sortKey)return 0;return (a.sortKey &lt; b.sortKey) ? g_order : -g_order;}function sortTable(tableNo, colNo, ord){var table, rows, nR, bs, i, j, temp;g_order=ord;g_colNo=colNo;table=g_tables[tableNo];rows=new Array();nR=0;bs=table.tBodies;for(i=0; i&lt;bs.length; ++i)for(j=0; j&lt;bs[i].rows.length; ++j){rows[nR]=bs[i].rows[j];temp=rows[nR].cells[g_colNo];if(temp) rows[nR].sortKey=temp.innerHTML;else rows[nR].sortKey=&quot;&quot;;++nR;}rows.sort(compareRows);for (i=0; i &lt; rows.length; ++i)insAtTop(table.tBodies[0], rows[i]);}">sort tables [HACK]</a></p>'''
		print '<table class="wikitable sortable" style="text-align:center; width:100%;">'
		print '<thead><tr>'
		print '<th>User</th><th>Edits *</th><th>Registered</th><th>Last edit</th><th>Days since</th>'
		print '</tr></thead>'
		for (user_id, user_name, user_editcount, user_registration) in self.conn.fetchall():
			self.conn.execute("""
SELECT TIMESTAMP(rev_timestamp), TIMESTAMPDIFF(DAY,rev_timestamp,now()) 
FROM revision
WHERE rev_user=%s
ORDER BY rev_timestamp
DESC LIMIT 1""", (user_id,))
			d = self.conn.fetchone()
			print '<tr>'
			css = d and d[1] > self.days and d[1]/user_editcount >= 7 and ' style="background:lightblue;"' or ''
			print '<td style="text-align:left;"><a href="http://en.wikipedia.org/wiki/User:%(user_name)s" class="extiw">%(user_name)s</a> (<a href="http://en.wikipedia.org/wiki/User_talk:%(user_name)s" class="extiw">talk</a> &bull; <a href="http://en.wikipedia.org/wiki/Special:Contributions/%(user_name)s" class="extiw">contribs</a>)</td><td%(css)s>%(user_editcount)s</td><td>%(user_registration)s</td>' % locals()
			if d:
				# bt: Blue Trigger value
				print '<td>%s</td><td style="background:%s;" title="btv:%s">% 5d</td>' % (d[0].strftime("%d %b %Y"), colorValue(d[1], self.days, revert=True), '%4.2f'%(d[1]/float(user_editcount)), d[1])
			print '</tr>'
		print '</table>'
		wikipedia.output('* this includes deleted edits')

if __name__ == "__main__" and wikipedia.handleUrlAndHeader():
	import time
	startTime = time.time()
	try:
		wikipedia.startContent(form=False)
		ua = UserActivity()
		ua.main()
	finally:
		wikipedia.endContent()
		wikipedia.stopme()