#! /usr/bin/python
"""
A simple demo program that grown beyond, see locateCoordv1.py for the simpler version

This should really be scale back infavor of a better backend implementation
"""
import wikipedia
import MySQLdb, cgi, math
import cgitb; cgitb.enable(logdir='./logs/')

def main():
	f = cgi.FieldStorage()
	dbname	= f.getfirst('dbname', 'coord_enwiki')
	globe   = f.getfirst('globe', None)
	lat1	= f.getfirst('lat1', '')
	lat2	= f.getfirst('lat2', '')
	lon1	= f.getfirst('lon1', '')
	lon2	= f.getfirst('lon2', '')
	range_km = f.getfirst('range_km', '')
	offset = f.getfirst('offset', '0')
	globe   = f.getfirst('globe', None)
	if range_km and not any((lat1, lat2, lon1, lon2)):
		# based on http://toolserver.org/~kolossos/wp-world/umkreis-source.php
		# by http://de.wikipedia.org/wiki/Benutzer:Kolossos
		range_km = float(range_km)/111.3
		lat      = ((float(f.getfirst('lat'))+270) % 180) - 90
		lat1     = lat - (range_km/2)
		lat2     = lat + (range_km/2)
		lon      = ((float(f.getfirst('lon'))+540) % 360) - 180
		lon1     = lon - range_km/2/math.cos(lat/57.27)
		lon2     = lon + range_km/2/math.cos(lat/57.27)
	
	tables = ('coord_commonswiki', 'coord_cawiki', 'coord_cswiki', 'coord_dawiki', 'coord_dewiki', 'coord_enwiki', 'coord_eswiki', 'coord_eowiki', 'coord_frwiki', 'coord_idwiki', 'coord_itwiki', 'coord_huwiki', 'coord_jawiki', 'coord_nlwiki', 'coord_nowiki', 'coord_plwiki', 'coord_ptwiki', 'coord_ruwiki', 'coord_rowiki', 'coord_skwiki', 'coord_fiwiki', 'coord_svwiki', 'coord_trwiki', 'coord_ukwiki', 'coord_vowiki', 'coord_zhwiki',)

	print """<form action="locateCoord.py"><fieldset>
<legend>Located articles in geographic region</legend>
<div  style="white-space:pre;"
>Database:	%s
<!-- Globe:  	<select name='globe'>%s</select> -->
Latitude:	BETWEEN <input name='lat1' value="%s" size="6" /> AND   <input name='lat2' value="%s" size="6" />
Longitude:	BETWEEN <input name='lon1' value="%s" size="6" /> AND   <input name='lon2' value="%s" size="6" />
<input type="submit" /></div>
</fieldset>
</form>
"""% (dbname in tables and "<select name='dbname'>%s</select>" % '\n'.join('<option%s>%s</option>'%(dbname==s and ' selected="selected"' or '',s) for s in tables) or "<input name='dbname' value=\"%s\"/>"%dbname, 
# TODO this should be dynamically generated
'\n'.join('<option%s>%s</option>'%(globe==s and ' selected="selected"' or '',s) for s in ('', 'mercury', 'venus', 'earth', 'moon', 'mars', 'titan', 'ganymend',)),
lat1, lat2, lon1, lon2, 
)
	
	if not (dbname and dbname.replace('_','').isalnum()):
		return
	db = MySQLdb.connect(db='u_dispenser_p', host=dbname[6:].replace('_', '-') + '-p.db.toolserver.org', read_default_file="/home/dispenser/.my.cnf")
	c = db.cursor()
	c.execute("SET SESSION TRANSACTION ISOLATION LEVEL READ UNCOMMITTED")
	c.execute((
"""
SELECT *, 0
FROM """
+dbname+""" 
WHERE MBRWithIn(gc_location, GeomFromText(%s))
/*AND gc_globe = %s*/
LIMIT """+(offset.isdigit() and offset or '0')+""", 501
"""),
	(
	'Polygon((%s %s, %s %s, %s %s, %s %s, %s %s))' % (lat1,lon1, lat1,lon2, lat2,lon2, lat2,lon1, lat1,lon1),
	globe,
	)
)

	print """<!--
* Do not query faster than 10 times per second.
* If you plan to use this more than once please leave 
  a note on my talk page so I know what's being used
* Likewise just ask if something you need is missing

 - [[User:Dispenser]]

-->"""
	print '<table border="1" width="100%" class="wikitable">'
	print '<tr><th>lat</th><th>lon</th><th>alt</th><th>dim</th><th>type</th><th>typesize</th><th>region</th><th>globe</th><th>title</th></tr>'
	print '<!--RESULTS-->'
	for (id, lat, lon, altitude, heading, dim, type, typesize, region, globe, primary, title, rectll, distance) in c.fetchmany(500):
		print '<tr%s><td>%g</td><td>%g</td><td>%s</td><td>%s</td><td>%s</td><td>%s</td><td>%s</td><td>%s</td><td><a href="http://%s.wikipedia.org/w/index.php?curid=%d">%s</a></td><td class="dist">%s</td></tr>' % (
		(not primary and ' class="nonprimary"' or ''), 
		lat,
		lon,
		altitude or '',
		dim or '',
		type or '',
		typesize or '',
		region or '',
		globe or '',
		dbname[6:-4],
		id,
		title.replace('_', ' ') or '[empty string]',
		distance)
	print '<!--/RESULTS-->'
	print '</table>'
	if c.fetchone():
		print '<p>Results were limited to 500, <a href="%s">next 500</a></p>' % '?dbname=%s&amp;lat1=%s&amp;lat2=%s&amp;lon1=%s&amp;lon2=%s&amp;offset=%s' % (dbname, lat1, lat2, lon1, lon2, parseInt(offset, 0)+500)


def parseInt(s, default=0):
	try: return int(s)
	except: return default

if __name__ == "__main__" and wikipedia.handleUrlAndHeader():
	try:
		wikipedia.startContent(form=False, head="""<style type="text/css">
.nonprimary{
	color:gray;
	line-height:95%;
	font-style: italic;
}
.dist {
	display:none;
}
</style>""")
		main()
	finally:
		wikipedia.endContent()
		wikipedia.stopme()
		wikipedia.time.sleep(1)