99f53f1573c5680f36fa2c19c5bfac6baa37b671
[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
9 class Mesurable(osv.Model):
10     _name = "cmmi.axes.mesurable"
11
12     _description = "Table de reference des mesusrables."
13
14     _statuts = [("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         "statut": fields.selection(_statuts, string="Statut"),
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_projets": fields.function(_nb_jours_init,
43                                             type="integer",
44                                             string="Nombre de jour"),
45     }
46
47     _sql_constraints = [
48         (
49             "date_init_deb_before_date_init_fin",
50             "CHECK(date_init_deb<> date_init_fin)",
51             "The date_init_deb should be previous date_init_fin",
52         ),
53         (
54             "date_plan_deb_before_date_plan_fin",
55             "CHECK(date_plan_deb <> date_plan_fin)",
56             "The date_plan_deb should be previous date_plan_fin",
57         ),
58         (
59             "date_reel_deb_before_date_reel_fin",
60             "CHECK(date_reel_deb<> date_reel_fin)",
61             "The date_reel_deb should be previous date_reel_fin",
62         ),
63     ]
64
65
66 #==================== TRAVAIL CALCUL JOURS OUVRES =============================
67
68     @staticmethod
69     def _get_date_paques(annee):
70         """
71         Retourne la date du dimanque de pâques pour une année donnée
72             sous la forme d'un objet date.
73         """
74         a = annee % 19
75         b = annee // 100
76         c = annee % 100
77         d = (19 * a + b - b // 4 - ((b - (b + 8) // 25 + 1) // 3) + 15) % 30
78         e = (32 + 2 * (b % 4) + 2 * (c // 4) - d - (c % 4)) % 7
79         f = d + e - 7 * ((a + 11 * d + 22 * e) // 451) + 114
80         mois = f // 31
81         jours = f % 31 + 1
82         return date(annee, mois, jours)
83
84
85     @staticmethod
86     def _get_jours_feries(annee):
87         """
88         Retourne une liste contenant les jours fériés d'une année donnée.
89         """
90         date_paques = Mesurable._get_date_paques(annee)
91         return [
92             date_paques+timedelta(days=1), # Lundi de Pâques
93             date_paques+timedelta(days=39), # Jeudi de l'Ascension
94             date_paques+timedelta(days=50), # Lundi de Pentecôte
95             date(annee, 1, 1), # Jour de l'An
96             date(annee, 5, 1), # Fête du travail
97             date(annee, 5, 8), # Armistice 1945
98             date(annee, 7, 14), # Fête Nationale
99             date(annee, 8, 15), # Assomption
100             date(annee, 11, 1), # Toussaint
101             date(annee, 11, 11), # Armistice 1918
102             date(annee, 12, 25), # Noël
103         ]
104
105
106     @staticmethod
107     def _jour_ouvre(d):
108         """ Retourne vrai si la date est un jour ouvre, faux sinon."""
109         if (5,6).__contains__(d.weekday()) or Mesurable._get_jours_feries(d.year).__contains__(d):
110             return False
111         else:
112             return True
113
114
115     @staticmethod
116     def _nb_jours_ouvre_entre_2_dates(d1, d2):
117         """
118         Retourne le nombre de jours ouvres entre deux dates données.
119
120         >>> nb_jours_ouvre_entre_2_dates(date(2013, 5, 1), date(2013, 7, 31))
121         62
122         >>> nb_jours_ouvre_entre_2_dates(date(2013, 7, 31), date(2013, 5, 1))
123         62
124         """
125         if not d1 or not d2:
126             return 0
127         if d1>d2:
128             d1, d2 = d2, d1 #Switch les 2 dates pour que d1 soit la plus petite
129         tmp = d1
130         jour_ouvres = 0
131         while tmp <= d2:
132             if Mesurable._jour_ouvre(tmp): # Si tmp est un jour ouvre
133                 jour_ouvres += 1
134             tmp += timedelta(days=1)
135         return jour_ouvres
136
137 #==============================================================================
138
139
140 class Palier(osv.Model):
141     _name = "cmmi.axes.palier"
142
143     _description = "Palier d'un projet."
144
145     _inherit = "cmmi.axes.mesurable"
146
147     _types_palier = [("normal", "Normal"), ("exceptionnel", "Exceptionnel"),
148                      ("correctif", "Correctif"), ("autre", "Autre")]
149
150     _columns = {
151         "type_palier": fields.selection(_types_palier, string="Type"),
152         "projet_id": fields.many2one("cmmi.projet",
153                                      string="Projet",
154                                      required=True),
155         "evolutions": fields.one2many("cmmi.evolution",
156                                       "palier_id",
157                                       string="Evolutions"),
158         "phases": fields.one2many("cmmi.axes.palier.phase",
159                                   "palier_id",
160                                   string="Phases"),
161     }
162
163     _defaults = {
164         "type_palier": "normal",
165     }
166
167
168     def create(self, cr, uid, vals, context=None):
169         palier_id = osv.Model.create(self, cr, uid, vals, context=context)
170
171         # Récupération des ids de toutes les phases
172         phase_model = self.pool.get("cmmi.projet.phase")
173         phases_ids = phase_model.search(cr, uid, [('selectionne', '=', True)])
174
175         # Création des PalierPhase
176         palier_phase_model = self.pool.get("cmmi.axes.palier.phase")
177         for phase_id in phases_ids:
178             palier_phase_model.create(
179                 cr,
180                 uid,
181                 {
182                     'phase_id': phase_id,
183                     'palier_id': palier_id,
184                 }
185             )
186
187         return palier_id
188
189
190
191 class PalierPhase(osv.Model):
192     _name = "cmmi.axes.palier.phase"
193
194     _description = "Phase d'un palier"
195
196     _inherit = "cmmi.axes.mesurable"
197
198     def _get_name(self, cr, uid, ids, field_name=None, arg=None, context=None):
199         if isinstance(ids, (int, long)):
200             ids = [ids]
201         return dict([(i, r.phase_id.name) for i, r in
202                 zip(ids, self.browse(cr, uid, ids, context=context))])
203
204
205     _columns = {
206         "name": fields.function(_get_name,
207                                 type='char',
208                                 store=True,
209                                 string="Nom de la phase"),
210         "phase_id": fields.many2one("cmmi.projet.phase",
211                                     string="Phase du projet"),
212         "palier_id": fields.many2one("cmmi.axes.palier",
213                                      string="Palier"),
214         # backrefs
215         "charges": fields.one2many("cmmi.evolution.charge",
216                                    "phase_id",
217                                    string="Charges"),
218 #        "evolutions": fields.one2many("cmmi.evolution", #Supprimé !
219 #                                      "phase_id",
220 #                                      string="Evolutions"),
221     }
222
223     def create(self, cr, uid, vals, context=None):
224         # TODO: gérer la création d'une phase de palier.
225         # Vérifier les valeurs contenues dans vals et les modifier / rajouter si nécessaire selon les cas suivants
226
227         # 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).
228         # Si commentaire est vide, alors par défaut, recopie du commentaire du palier.
229         # Si version est vide, alors par dégaut, recopie de la version du palier.
230
231         return osv.Model.create(self, cr, uid, vals, context=context)
232
233
234 class Chantier(osv.Model):
235     _name = "cmmi.axes.chantier"
236
237     _description = "Chantiers d'un projet."
238
239     _inherit = "cmmi.axes.mesurable"
240
241     _columns = {
242         "projet_id": fields.many2one("cmmi.projet",
243                                      string="Projet",
244                                      required=True),
245         "module_ids": fields.many2many("cmmi.description.module",
246                                        "cmmi_module_chantier_rel",
247                                        "chantier_id",
248                                        "module_id",
249                                        "Modules"),
250         "evolutions": fields.one2many("cmmi.evolution",
251                                       "chantier_id",
252                                       string="Evolutions"),
253     }