mapfilize.py

# -*- coding: utf-8 -*- 

import sys, os, subprocess
from mako.template import Template
from mako import exceptions


try:
    import psycopg2
    import psycopg2.extras
    from psycopg2.extensions import register_type, UNICODE, connection
except ImportError:
    print "You need psycopg2 to run this script. Try to install it with 'easy_install psycopg2'"
    sys.exit()

projdir = os.path.abspath(os.path.join('..',os.path.dirname(__file__)))

class Mapfilize():
	
	layers = []
	
	tplvars = None
        project = ''	
	
	def __init__(self, project):
                self.project = project 
		self.connect()
		self.initLayers()
	        self.project_dir = os.path.abspath(os.path.join(os.curdir,'..','services', project))
		
	def connect(self):
		try:
   			self.conn=psycopg2.connect("dbname='bod' user='www-data' password='www-data' port=5432 host='bgdipg01t.lt.admin.ch'")
   			print "Database connection established"
		except:
   			print "Critical Error: Unable to connect to the database. Exit"
   			sys.exit()
	
	def initLayers(self):
		cur = self.conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
                if self.project == 'wms-swisstopowms':
        		cur.execute("SELECT fk_id_dataset FROM xt_dataset_wms xt left join dataset d on xt.fk_id_dataset = d.id_dataset WHERE fk_map_name LIKE '%" + self.project +"%' AND NOT d.id_dataset IN ( select parent_id from dataset where not parent_id is null ) ORDER BY sort_key ASC")
                elif self.project == 'wms-bgdi':
                # only use layers with staging = prod for wms-bgdi
        		cur.execute("SELECT fk_id_dataset FROM xt_dataset_wms x left join dataset d on x.fk_id_dataset = d.id_dataset WHERE fk_map_name LIKE '%" + self.project + "%' AND d.staging = 'prod'  AND NOT d.id_dataset IN ( select parent_id from dataset where not parent_id is null ) ORDER BY 'ch.'||split_part(fk_id_dataset,'.',2)||'.' DESC ,sort_key DESC")
                else:
        		cur.execute("SELECT fk_id_dataset FROM xt_dataset_wms xt left join dataset d on xt.fk_id_dataset = d.id_dataset WHERE fk_map_name LIKE '%" + self.project + "%' AND NOT d.id_dataset IN ( select parent_id from dataset where not parent_id is null ) ORDER BY 'ch.'||split_part(fk_id_dataset,'.',2)||'.' DESC ,sort_key DESC")
        	for row in  cur.fetchall():
		
			self.layers.append(row['fk_id_dataset'])
		print "Add %d layer(s) to mapfile" % len(self.layers)
		return True
		
	def writeMapfile(self):
		tpl = Template(filename= os.path.join( self.project_dir, self.project +'.map.mako'), input_encoding = 'utf-8')
		self.tplvars = type('tplvars',(object,),{})
		
		self.tplvars.layers = self.layers

		renderedTpl = None
		try:
            		renderedTpl = tpl.render_unicode(c=self.tplvars)
		except:
     			print exceptions.text_error_template().render()
		if renderedTpl is not None:
            		#logging.debug(renderedTpl)
            		filename =  os.path.join( self.project_dir, self.project + '.map')
            		FILE = open(filename,"w")
            		FILE.writelines(renderedTpl.encode('utf-8'))
            		FILE.close()
            		print "Mapfile written successfullyi to %s" %  filename
            		return True
		print "mapfile creation failed!!!"        


if __name__ == '__main__':
    if not sys.argv[1:]:
        sys.stdout.write("Sorry, you must specify one argument, for instance wms-bgdi")
        sys.exit(0)
    else:
        newMapfile = Mapfilize(sys.argv[1])
        newMapfile.writeMapfile()