93aa5654f4eecd6b7aae2e843e29d5d8daaa2321
[restcoda.git] / document.py
1 #!/usr/bin/python
2 # -*- coding: utf-8 -*-
3
4 #
5 # Classes pour les documents DEM-* et COM-*
6 #
7 # Ces documents ont tous les même format. On créé donc une classe "Document"
8 # générique, qui sera hérité. Chaque classe fille précisera le code du document.
9 #
10
11 from objet import Objet, ObjetInconnu
12 from utils import dict_fetchone, coda2rest
13 # pour l'annee du budget
14 from datetime import datetime
15
16 class Document(Objet):
17 """objet document CODA (demlog, comlog, demdep... ils ont tous le même format)"""
18 accept_like_filters = [ 'code', 'demandeur', 'approbateur', 'code_service', 'statut', 'fournisseur', 'statut' ]
19 def __init__(self, environ, code_document='%', prefix_template='document'):
20 super(Document, self).__init__(environ)
21 self.code_document = code_document
22 self.prefix_template = prefix_template
23 # filtrage : si aucun statut n'est spécifié, on prend "statut like non%"
24 if self.sqlwhere != '': self.sqlwhere = ' and ' + self.sqlwhere
25
26 def _get_index(self, code):
27 """renvoie une liste de documents"""
28 # connexion a la base de données
29 cursor = self.cursor()
30 # extraction des données
31 # statut par defaut : non-autorisé
32 if not 'statut' in self.filters:
33 self.sqlwhere = " and (statut = 'non-autorise') " + self.sqlwhere
34 self.filters['status'] = ['non-autorise']
35 request = "select top %d * from auf_v_acces_demcom where (code like '%s') %s order by date_modif desc" % ( self.sqllimit, code, self.sqlwhere )
36 cursor.execute( request )
37 items = []
38 while 1:
39 item = dict_fetchone( cursor )
40 if item == None: break
41 item['code_rest'] = coda2rest(item['code'])
42 items.append(item)
43 index = {}
44 index['code'] = code
45 index['documents'] = items
46 index['nombre_documents'] = len(items)
47 index['limite_documents'] = self.sqllimit
48 index['filtres'] = self.filters
49 index['requete_sql'] = request
50 return index
51
52 def _get_details(self,code,numero):
53 """renvoie la liste des détails pour un document"""
54 details = []
55 cursor = self.cursor()
56 request = "select * from auf_v_acces_dtls_demcom where (code = '%s') and (numero = '%s')" % (code, numero)
57 cursor.execute( request )
58 while 1:
59 detail = dict_fetchone(cursor)
60 if detail == None: break
61 details.append(detail)
62 return details
63
64 def _get_liaisons(self,code,numero):
65 """renvoie la liste des demandes sources d'une commande ou des commandes issues d'une demande"""
66 liaisons = []
67 cursor = self.cursor()
68 if code[:3] == 'DEM':
69 request = "select code,numero from auf_v_acces_dtls_demcom where source_code = '%s' and source_numero = '%s' group by code,numero"
70 else:
71 request = "select source_code,source_numero from auf_v_acces_dtls_demcom where code = '%s' and numero = '%s' group by source_code,source_numero"
72 cursor.execute( request % (code, numero ) )
73 while 1:
74 liaison = dict_fetchone(cursor)
75 if liaison == None: break
76 liaisons.append(liaison)
77 return liaisons
78
79 def _get_budget(self,code_service,annee):
80 """renvoie les données du budget du code_service"""
81 cursor = self.cursor()
82 request = "select top 1 * from auf_budget where projet_poste = '%s' and annee = %d" % ( code_service[:6] , annee )
83 cursor.execute( request )
84 budget = dict_fetchone(cursor)
85 if budget == None:
86 budget = { 'projet_poste': '0000XX', 'projet': 'Inconnu', 'poste':'Inconnu', 'annee': annee }
87 budget['date_interrogation'] = datetime.today()
88 return budget
89
90 def _get_document(self,code,numero):
91 """renvoie un document"""
92 cursor = self.cursor()
93 request = "select top 1 * from auf_v_acces_demcom where (code like '%s') and (numero = %d) %s" % (code, numero, self.sqlwhere)
94 cursor.execute( request )
95 document = dict_fetchone(cursor)
96 if document == None:
97 raise ObjetInconnu('document')
98 # TODO : declencher une autre exception ObjetNonUnique si plusieurs résultats ?
99 document['requete_sql'] = request
100 document['code_rest'] = coda2rest( document['code'] )
101 document['details'] = self._get_details( document['code'], document['numero'] )
102 document['liaisons'] = self._get_liaisons( document['code'], document['numero'] )
103 if isinstance( document['date_modif'], datetime ):
104 annee = document['date_modif'].year
105 else:
106 annee = datetime.today().year
107 document['budget'] = self._get_budget( document['code_service'], annee )
108 # TODO : ajouter budget (a l'heure de la demande)
109 return document
110
111 def index(self):
112 """renvoie une liste de documents formatée"""
113 template = self.template('%s-index' % self.prefix_template) # préparation du modèle
114 index = self._get_index( self.code_document ) # extraction des documents concernés
115 output = template.render( index ) # formattage via le modèle
116 return self.outputformat, output
117
118 def show(self):
119 """renvoie un document formaté"""
120 # return 'txt', '%s' % self.environ # pour debug
121 numero = int(self.environ['wsgiorg.routing_args'][1]['id'])
122 template = self.template( self.prefix_template )
123 document = self._get_document( self.code_document, numero )
124 output = template.render( document )
125 return self.outputformat, output
126
127 #
128 # les classes filles précisent le code_document
129 #
130
131 class dem(Document):
132 def __init__(self, environ):
133 super(dem, self).__init__(environ, code_document = 'DEM-%')
134
135 class com(Document):
136 def __init__(self, environ):
137 super(com, self).__init__(environ, code_document = 'COM-%')
138
139 class demlog(Document):
140 def __init__(self, environ):
141 super(demlog, self).__init__(environ, code_document = 'DEM-LOG-AUF')
142
143 class comlog(Document):
144 def __init__(self, environ):
145 super(comlog, self).__init__(environ, code_document = 'COM-LOG-AUF')
146
147 class demdep(Document):
148 def __init__(self, environ):
149 super(demdep, self).__init__(environ, code_document = 'DEM-DEP-AUF')
150
151 class comdep(Document):
152 def __init__(self, environ):
153 super(comdep, self).__init__(environ, code_document = 'COM-DEP-AUF')
154
155 class dempub(Document):
156 def __init__(self, environ):
157 super(dempub, self).__init__(environ, code_document = 'DEM-PUB-AUF')
158
159 class compub(Document):
160 def __init__(self, environ):
161 super(compub, self).__init__(environ, code_document = 'COM-PUB-AUF')
162
163 class comsre(Document):
164 def __init__(self, environ):
165 # on distingue comsre et comsrexxx (avec xxx = implantation)
166 impl = environ['wsgiorg.routing_args'][1].get('impl','%')
167 super(comsre, self).__init__(environ, code_document = 'COM-SRE-%s' % impl)
168
169 class comare(Document):
170 def __init__(self, environ):
171 impl = environ['wsgiorg.routing_args'][1].get('impl','%')
172 super(comare, self).__init__(environ, code_document = 'COM-ARE-%s' % impl)
173
174 class comxre(Document):
175 def __init__(self, environ):
176 impl = environ['wsgiorg.routing_args'][1].get('impl','%')
177 super(comxre, self).__init__(environ, code_document = 'COM-%%RE-%s' % impl)
178