Lors de la création d'un palier phase, celui ci prend les dates planifiées du palier
[OpenERP/cmmi.git] / axes.py
1 #-*- coding: utf8 -*-
2 '''
3 '''
4
5 from openerp.osv import osv, fields
6 from datetime import date, timedelta, datetime
7
8 # ================================ MESURABLE ================================ #
9 class Mesurable(osv.Model):
10     _name = "cmmi.axes.mesurable"
11
12     _description = "Table de reference des mesusrables."
13
14     _states = [("cree", "Crée"), ("encours", "En cours"),
15                 ("termine", "Terminé"), ("abandonne", "Abandonné"),
16                 ("suspendu", "Suspendu"), ("generique", "Générique")]
17
18     def _nb_jours_init(self, cr, uid, ids, field, arg, context=None):
19         result = {}
20         for m in self.browse(cr, uid, ids, context=context):
21             if not m.date_init_deb or not m.date_init_fin:
22                 result[m.id] = 0
23                 continue
24             result[m.id] = Mesurable._nb_jours_ouvre_entre_2_dates(
25                         datetime.strptime(m.date_init_deb, "%Y-%m-%d").date(),
26                         datetime.strptime(m.date_init_fin, "%Y-%m-%d").date())
27         return result
28
29     _columns = {
30         "name": fields.char(string="Title", size=64, required=True),
31         "description": fields.text(string="Description"),
32         "commentaire": fields.text(string="Commentaire"),
33         "state": fields.selection(_states, string="State"),
34         "version": fields.char(string="Version", size=16),
35         "date_jalon": fields.date(string="Jalon"),
36         "date_init_deb": fields.date(string="Init début"),
37         "date_init_fin": fields.date(string="Init fin"),
38         "date_plan_deb": fields.date(string="Plan début"),
39         "date_plan_fin": fields.date(string="Plan fin"),
40         "date_reel_deb": fields.date(string="Réel début"),
41         "date_reel_fin": fields.date(string="Réel fin"),
42         "nb_jours_initial": fields.function(_nb_jours_init,
43                                             type="integer",
44                                             string="Nombre de jour"),
45     }
46
47     _defaults = {
48         "state": "cree",
49     }
50
51     _sql_constraints = [
52         (
53             "date_init_deb_before_date_init_fin",
54             "CHECK(date_init_deb <= date_init_fin)",
55             "The date_init_deb should be previous date_init_fin",
56         ),
57         (
58             "date_plan_deb_before_date_plan_fin",
59             "CHECK(date_plan_deb <= date_plan_fin)",
60             "The date_plan_deb should be previous date_plan_fin",
61         ),
62         (
63             "date_reel_deb_before_date_reel_fin",
64             "CHECK(date_reel_deb <= date_reel_fin)",
65             "The date_reel_deb should be previous date_reel_fin",
66         ),
67     ]
68
69
70     def commencer(self, cr, uid, ids, context=None):
71         if type(ids) == list:
72             if len(ids) != 1:
73                 return # TODO: message d'avertissement
74             ids = ids[0]
75
76         palier = self.read(cr, uid, ids, ['date_plan_deb', 'date_plan_fin', 'state'], context)
77
78         if palier['state'] != 'cree':
79             return
80
81         self.write(
82             cr,
83             uid,
84             ids, {
85                 'date_init_deb' : palier['date_plan_deb'],
86                 'date_init_fin' : palier['date_plan_fin'],
87                 'state': 'encours'
88             },
89             context)
90         return self
91
92
93     def suspendre(self, cr, uid, ids, context=None):
94         if type(ids) == list:
95             if len(ids) != 1:
96                 return # TODO: message d'avertissement
97             ids = ids[0]
98
99         mesurable = self.read(cr, uid, ids, ['state'], context, context)
100         if mesurable['state'] != 'encours':
101             return
102         self.write(
103             cr,
104             uid,
105             ids,
106             {'state': 'suspendu'},
107             context,
108         )
109         return self
110
111     def terminer(self, cr, uid, ids, context=None):
112         if type(ids) == list:
113             if len(ids) != 1:
114                 return # TODO: message d'avertissement
115             ids = ids[0]
116
117         mesurable = self.read(cr, uid, ids, ['state'], context, context)
118         if mesurable['state'] != 'encours':
119             return
120         self.write(
121             cr,
122             uid,
123             ids,
124             {'state': 'termine'},
125             context,
126         )
127         return self
128
129     def abandonner(self, cr, uid, ids, context=None):
130         if type(ids) == list:
131             if len(ids) != 1:
132                 return # TODO: message d'avertissement
133             ids = ids[0]
134
135         mesurable = self.read(cr, uid, ids, ['state'], context, context)
136         if not ('encours', 'cree').__contains__(mesurable['state']):
137             return
138         self.write(
139             cr,
140             uid,
141             ids,
142             {'state': 'abandonne'},
143             context,
144         )
145         return self
146
147     def reprendre(self, cr, uid, ids, context=None):
148         if type(ids) == list:
149             if len(ids) != 1:
150                 return # TODO: message d'avertissement
151             ids = ids[0]
152
153         mesurable = self.read(cr, uid, ids, ['state'], context, context)
154         if mesurable['state'] != 'suspendu':
155             return
156         self.write(
157             cr,
158             uid,
159             ids,
160             {'state': 'encours'},
161             context,
162         )
163         return self
164
165 #------------ TRAVAIL CALCUL JOURS OUVRES ------------
166     @staticmethod
167     def _get_date_paques(annee):
168         """
169         Retourne la date du dimanque de pâques pour une année donnée
170             sous la forme d'un objet date.
171         """
172         a = annee % 19
173         b = annee // 100
174         c = annee % 100
175         d = (19 * a + b - b // 4 - ((b - (b + 8) // 25 + 1) // 3) + 15) % 30
176         e = (32 + 2 * (b % 4) + 2 * (c // 4) - d - (c % 4)) % 7
177         f = d + e - 7 * ((a + 11 * d + 22 * e) // 451) + 114
178         mois = f // 31
179         jours = f % 31 + 1
180         return date(annee, mois, jours)
181
182
183     @staticmethod
184     def _get_jours_feries(annee):
185         """
186         Retourne une liste contenant les jours fériés d'une année donnée.
187         """
188         date_paques = Mesurable._get_date_paques(annee)
189         return [
190             date_paques+timedelta(days=1), # Lundi de Pâques
191             date_paques+timedelta(days=39), # Jeudi de l'Ascension
192             date_paques+timedelta(days=50), # Lundi de Pentecôte
193             date(annee, 1, 1), # Jour de l'An
194             date(annee, 5, 1), # Fête du travail
195             date(annee, 5, 8), # Armistice 1945
196             date(annee, 7, 14), # Fête Nationale
197             date(annee, 8, 15), # Assomption
198             date(annee, 11, 1), # Toussaint
199             date(annee, 11, 11), # Armistice 1918
200             date(annee, 12, 25), # Noël
201         ]
202
203
204     @staticmethod
205     def _jour_ouvre(d):
206         """ Retourne vrai si la date est un jour ouvre, faux sinon."""
207         if (5,6).__contains__(d.weekday()) or Mesurable._get_jours_feries(d.year).__contains__(d):
208             return False
209         else:
210             return True
211
212
213     @staticmethod
214     def _nb_jours_ouvre_entre_2_dates(d1, d2):
215         """
216         Retourne le nombre de jours ouvres entre deux dates données.
217
218         >>> nb_jours_ouvre_entre_2_dates(date(2013, 5, 1), date(2013, 7, 31))
219         62
220         >>> nb_jours_ouvre_entre_2_dates(date(2013, 7, 31), date(2013, 5, 1))
221         62
222         """
223         if not d1 or not d2:
224             return 0
225         if d1>d2:
226             d1, d2 = d2, d1 #Switch les 2 dates pour que d1 soit la plus petite
227         tmp = d1
228         jour_ouvres = 0
229         while tmp <= d2:
230             if Mesurable._jour_ouvre(tmp): # Si tmp est un jour ouvre
231                 jour_ouvres += 1
232             tmp += timedelta(days=1)
233         return jour_ouvres
234
235
236
237 # ================================= PALIER ================================== #
238 class Palier(osv.Model):
239     _name = "cmmi.axes.palier"
240
241     _description = "Palier d'un projet."
242
243     _inherit = "cmmi.axes.mesurable"
244
245     _types_palier = [("normal", "Normal"), ("exceptionnel", "Exceptionnel"),
246                      ("correctif", "Correctif"), ("autre", "Autre")]
247
248     _columns = {
249         "type_palier": fields.selection(_types_palier, string="Type"),
250         "projet_id": fields.many2one("cmmi.projet",
251                                      string="Projet",
252                                      required=True),
253         "evolutions": fields.one2many("cmmi.evolution",
254                                       "palier_id",
255                                       string="Evolutions"),
256         "phases": fields.one2many("cmmi.axes.palier.phase",
257                                   "palier_id",
258                                   string="Phases"),
259     }
260
261     _defaults = {
262         "type_palier": "normal",
263     }
264
265
266     def create(self, cr, uid, vals, context=None):
267         palier_id = osv.Model.create(self, cr, uid, vals, context=context)
268
269         # Récupération des ids de toutes les phases
270         phase_model = self.pool.get("cmmi.projet.phase")
271         phases_ids = phase_model.search(cr, uid, [('selectionne', '=', True)])
272
273         palier_model = self.pool.get("cmmi.axes.palier")
274         palier = palier_model.read(cr, uid, palier_id, ['date_plan_deb', 'date_plan_fin'])
275
276         # Création des PalierPhase
277         palier_phase_model = self.pool.get("cmmi.axes.palier.phase")
278         for phase_id in phases_ids:
279             palier_phase_model.create(
280                 cr,
281                 uid,
282                 {
283                     'phase_id': phase_id,
284                     'palier_id': palier_id,
285                     'date_plan_deb': palier['date_plan_deb'],
286                     'date_plan_fin': palier['date_plan_fin'],
287                 }
288             )
289         return palier_id
290
291
292
293 # =============================== PALIER-PHASE ============================== #
294 class PalierPhase(osv.Model):
295     _name = "cmmi.axes.palier.phase"
296
297     _description = "Phase d'un palier"
298
299     _inherit = "cmmi.axes.mesurable"
300
301     def _get_name(self, cr, uid, ids, field_name=None, arg=None, context=None):
302         if isinstance(ids, (int, long)):
303             ids = [ids]
304         return dict([(i, r.phase_id.name) for i, r in
305                 zip(ids, self.browse(cr, uid, ids, context=context))])
306
307
308     _columns = {
309         "name": fields.function(_get_name,
310                                 type='char',
311                                 store=True,
312                                 string="Nom de la phase"),
313         "phase_id": fields.many2one("cmmi.projet.phase",
314                                     string="Phase du projet"),
315         "palier_id": fields.many2one("cmmi.axes.palier",
316                                      string="Palier"),
317         # backrefs
318         "charges": fields.one2many("cmmi.evolution.charge",
319                                    "phase_id",
320                                    string="Charges"),
321 #        "evolutions": fields.one2many("cmmi.evolution", #Supprimé !
322 #                                      "phase_id",
323 #                                      string="Evolutions"),
324     }
325
326     def create(self, cr, uid, vals, context=None):
327         # TODO: gérer la création d'une phase de palier.
328         # Vérifier les valeurs contenues dans vals et les modifier / rajouter si nécessaire selon les cas suivants
329
330         # Si description est vide, alors par défaut, recopie de la description du palier et de la phase (concaténés avec un retour à la ligne entre les deux).
331         # Si commentaire est vide, alors par défaut, recopie du commentaire du palier.
332         # Si version est vide, alors par dégaut, recopie de la version du palier.
333
334         return osv.Model.create(self, cr, uid, vals, context=context)
335
336
337
338 # ================================ CHANTIER ================================= #
339 class Chantier(osv.Model):
340     _name = "cmmi.axes.chantier"
341
342     _description = "Chantiers d'un projet."
343
344     _inherit = "cmmi.axes.mesurable"
345
346     _columns = {
347         "projet_id": fields.many2one("cmmi.projet",
348                                      string="Projet",
349                                      required=True),
350         "module_ids": fields.many2many("cmmi.description.module",
351                                        "cmmi_module_chantier_rel",
352                                        "chantier_id",
353                                        "module_id",
354                                        "Modules"),
355         "evolutions": fields.one2many("cmmi.evolution",
356                                       "chantier_id",
357                                       string="Evolutions"),
358     }