optimisations et meilleure gestion des erreurs
[restcoda.git] / rest.wsgi
CommitLineData
1795efcd
TN
1#!/usr/bin/python
2# -*- coding: utf-8 -*-
3
4import sys, traceback
5
a6627284
TN
6# configuration (codes d'accès à la base MS-SQL)
7sys.path.append('/home/thomas/public_html/')
8import rest_config
9
0a4c31d4 10# pour savoir quel objet interroger : Routes
1795efcd 11from routes import Mapper
a6627284 12from routes.middleware import RoutesMiddleware
81b8aba3
TN
13# et analyse des paramètres (POST et query_string) pour filtrage
14from paste.request import parse_formvars
1795efcd 15
a6627284 16# pour chercher les données sur MS-SQL
1795efcd 17from pymssql import connect
a6627284
TN
18
19# pour afficher le résultat : jinja
1795efcd
TN
20from jinja import Environment, FileSystemLoader
21from jinja.filters import stringfilter
0ae81148 22import jinja.exceptions
1795efcd 23
0a4c31d4
TN
24# TODO systeme de cache : beaker
25# from beaker.middleware import CacheMiddleware
1795efcd 26
a6627284 27# formats de sortie autorisés, et content-type correspondant
d588d902
TN
28# formats = { 'xml': 'application/xml', 'html': 'text/html', 'txt': 'text/plain', 'json': 'application/json', 'rss': 'application/rss+xml' }
29formats = { 'xml': 'application/xml', 'html': 'text/html', 'txt': 'text/plain', 'json': 'text/plain', 'rss': 'application/rss+xml' }
1795efcd 30
0a4c31d4 31# les routes RESTful (cf http://routes.groovie.org/manual.html#restful-services)
1795efcd 32mapper = Mapper()
0a4c31d4
TN
33mapper.resource('demlog','demlog')
34mapper.resource('comlog','comlog')
35mapper.resource('demdep','demdep')
36mapper.resource('comdep','comdep')
37mapper.resource('dempub','dempub')
38mapper.resource('compub','compub')
0a4c31d4 39mapper.resource('comare','comare')
db1681c0
TN
40# pour les comarexxx où xxx est un code d'implantation
41mapper.resource('comarei','comare:(impl)',controller='comare')
0a4c31d4 42mapper.resource('comsra','comsre')
db1681c0 43mapper.resource('comsrai','comsre:(impl)',controller='comsre')
f6b772ed
TN
44mapper.resource('dem','dem')
45mapper.resource('com','com')
1795efcd
TN
46
47class objetsql(object):
0a4c31d4 48 """objet de base : dispose d'un accès à MS-SQL (lire les données) et d'un accès à jinja (rendu des données)"""
1795efcd 49 def __init__(self, environ):
1795efcd 50 self.environ = environ
0ae81148
TN
51 def template(self,template):
52 self.jinja = Environment(loader=FileSystemLoader('/home/thomas/public_html/'))
53 self.outputformat = self.environ['wsgiorg.routing_args'][1].get('format','xml')
54 return self.jinja.get_template('%s.%s' % (template,self.outputformat))
55 def cursor(self):
56 if not hasattr(self,'bd'):
57 self.bd = connect(host=rest_config.host,user=rest_config.user,password=rest_config.password,database=rest_config.database)
58 return self.bd.cursor()
a6627284 59
1795efcd 60class document(objetsql):
0a4c31d4 61 """objet document CODA (demlog, comlog, demdep... ils ont tous le même format)"""
1795efcd
TN
62 def __init__(self, environ, code_document='%', basename_template='document'):
63 super(document, self).__init__(environ)
64 self.code_document = code_document
65 self.basename_template = basename_template
66
0ae81148
TN
67 def _get_index(self, code):
68 """renvoie une liste de documents"""
69 # connexion a la base de données
70 cursor = self.cursor()
71 # extraction des données
72 cursor.execute("select top 15 * from auf_v_acces_demcom where code like '%s' order by date_modif desc" % code)
73 index = {}
74 index['code'] = code
75 items = []
1795efcd 76 while 1:
0ae81148
TN
77 item = dict_fetchone( cursor )
78 if item == None: break
79 item['code_rest'] = coda2rest(item['code'])
80 items.append(item)
81 index['documents'] = items
82 return index
83
84 def _get_details(self,code,id):
85 """renvoie la liste des détails pour un document"""
1795efcd 86 details = []
0ae81148
TN
87 cursor = self.cursor()
88 cursor.execute("select * from auf_v_acces_dtls_demcom where code like '%s' and numero = %d" % (code, id))
1795efcd 89 while 1:
0ae81148 90 detail = dict_fetchone(cursor)
1795efcd
TN
91 if detail == None: break
92 details.append(detail)
0ae81148
TN
93 return details
94
95 def _get_document(self,code,id):
96 """renvoie un document"""
97 cursor = self.cursor()
98 cursor.execute("select top 1 * from auf_v_acces_demcom where code like '%s' and numero = %d" % (code, id))
99 document = dict_fetchone(cursor)
100 if document == None:
101 raise "document inexistant"
102 document['code_rest'] = coda2rest( document['code'] )
103 document['details'] = self._get_details( code,id )
104 return document
105
106 def index(self):
107 """renvoie une liste de documents formatée"""
108 # preparation du modèle
109 template = self.template('%s-index' % self.basename_template)
110 # recherche des documents concernés
111 index = self._get_index( self.code_document )
112 # envoi des données au template
113 output = template.render( index )
114 return self.outputformat, output
115
116 def show(self):
117 """renvoie un document formaté"""
118 # return 'txt', '%s' % self.environ # pour debug
119 id = int(self.environ['wsgiorg.routing_args'][1]['id'])
120 template = self.template( self.basename_template )
121 document = self._get_document( self.code_document, id )
122 output = template.render( document )
1795efcd
TN
123 return self.outputformat, output
124
f6b772ed
TN
125class dem(document):
126 def __init__(self, environ):
127 super(dem, self).__init__(environ, code_document = 'DEM-%')
128
129class com(document):
130 def __init__(self, environ):
131 super(com, self).__init__(environ, code_document = 'COM-%')
132
1795efcd
TN
133class demlog(document):
134 def __init__(self, environ):
135 super(demlog, self).__init__(environ, code_document = 'DEM-LOG-AUF')
136
137class comlog(document):
138 def __init__(self, environ):
139 super(comlog, self).__init__(environ, code_document = 'COM-LOG-AUF')
140
0a4c31d4
TN
141class demdep(document):
142 def __init__(self, environ):
143 super(demdep, self).__init__(environ, code_document = 'DEM-DEP-AUF')
144
145class comdep(document):
146 def __init__(self, environ):
147 super(comdep, self).__init__(environ, code_document = 'COM-DEP-AUF')
148
149class dempub(document):
150 def __init__(self, environ):
151 super(dempub, self).__init__(environ, code_document = 'DEM-PUB-AUF')
152
153class compub(document):
154 def __init__(self, environ):
155 super(compub, self).__init__(environ, code_document = 'COM-PUB-AUF')
156
0a4c31d4
TN
157class comsre(document):
158 def __init__(self, environ):
db1681c0
TN
159 # est-ce un appel de comsre ou comsrexxx (avec xxx = implantation)
160 impl = environ['wsgiorg.routing_args'][1].get('impl','%')
161 super(comsre, self).__init__(environ, code_document = 'COM-SRE-%s' % impl)
0a4c31d4
TN
162
163class comare(document):
164 def __init__(self, environ):
db1681c0
TN
165 impl = environ['wsgiorg.routing_args'][1].get('impl','%')
166 super(comare, self).__init__(environ, code_document = 'COM-ARE-%s' % impl)
0a4c31d4 167
0ae81148
TN
168#
169# fin des objets accessibles
170#
171
a6627284 172def dispatcher(environ, start_response):
81b8aba3
TN
173 """dispatch vers la bonne methode du bon objet, et retour WSGI"""
174 parse_formvars(environ)
a6627284 175 results = environ['wsgiorg.routing_args'][1]
1795efcd
TN
176 try:
177 target_class = globals()[results['controller']]
0ae81148
TN
178 method = getattr(target_class,results['action'])
179 except:
180 start_response("404 Not Found", [('Content-type', 'text/html')])
181 return '<html><body><h2>404 objet ou action invalide</h2><pre>%s: %s\n%s</pre></body></html>' % ( sys.exc_info()[0] , sys.exc_info()[1] , traceback.format_exc())
182 try:
1795efcd
TN
183 type, output = method(target_class(environ))
184 start_response("200 OK", [('Content-type', formats[type])])
185 return output.encode('utf-8')
0ae81148
TN
186 except jinja.exceptions.TemplateNotFound, template:
187 start_response("415 Unsupported Media Type", [('Content-type', 'text/html')])
188 return '<html><body><h2>415 format non supporté (%s inexistant)</h2></body></html>' % template
1795efcd 189 except:
0ae81148
TN
190 start_response("500 INTERNAL ERROR", [('Content-type', 'text/html')])
191 return '<html><body><h2>500 erreur lors du traitement</h2><pre>%s: %s\n%s</pre></body></html>' % ( sys.exc_info()[0] , sys.exc_info()[1] , traceback.format_exc())
1795efcd 192
81b8aba3 193# application qui sera lancée par mod_wsgi : on route et on dispatche
a6627284 194application = RoutesMiddleware( dispatcher, mapper)
0ae81148 195# TODO : ajouter un middleware de cache (beaker, basé sur REQUEST_URI quand la methode est GET)
1795efcd 196
0a4c31d4
TN
197
198#
199# petits utilitaires
200#
1795efcd
TN
201def dict_fetchone(cursor):
202 """Renvoie le resultat d'un fetchone dans un dictionnaire"""
203 result = cursor.fetchone()
204 if result == None: return None
205 result_dict = {}
206 for i in range(len(result)):
207 if isinstance( result[i], str ):
208 result_dict[cursor.description[i][0]] = result[i].decode('iso-8859-1')
209 else:
210 result_dict[cursor.description[i][0]] = result[i]
211 return result_dict
212
a6627284 213import re
db1681c0 214p = re.compile('(dem|com)-(...)-(...)',re.IGNORECASE)
a6627284 215def coda2rest(value):
db1681c0
TN
216 """Traduit un nom CODA vers la base REST correspodante,
217 par exemple DEM-LOG-AUF en demlog ou COM-ARE-VN3 en comarevn3"""
a6627284
TN
218 m = p.search(value)
219 if m == None: return value
db1681c0
TN
220 if m.group(3).lower() == 'auf':
221 return m.group(1).lower() + m.group(2).lower()
222 else:
223 return m.group(1).lower() + m.group(2).lower() + m.group(3).lower()
a6627284 224