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