Refonte de la notion de phase 3/3
[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     _statuts = [("cree", "Crée"), ("encours", "En cours"),
13                 ("termine", "Terminé"), ("abandonne", "Abandonné"),
14                 ("suspendu", "Suspendu"), ("generique", "Générique")]
15
16     def _nb_jours_init(self, cr, uid, ids, field, arg, context=None):
17         result = {}
18         for m in self.browse(cr, uid, ids, context=context):
19             result[m.id] = Mesurable._nb_jours_ouvre_entre_2_dates(
20                         datetime.strptime(m.date_init_deb, "%Y-%m-%d").date(),
21                         datetime.strptime(m.date_init_fin, "%Y-%m-%d").date())
22         return result
23
24     _columns = {
25         "name": fields.char(string="Title", size=64, required=True),
26         "description": fields.text(string="Description"),
27         "commentaire": fields.text(string="Commentaire"),
28         "statut": fields.selection(_statuts, string="Statut"),
29         "version": fields.char(string="Version", size=16),
30         "date_jalon": fields.date(string="Jalon"),
31         "date_init_deb": fields.date(string="Date initiale début"),
32         "date_init_fin": fields.date(string="Date initiale de fin"),
33         "date_plan_deb": fields.date(string="Date plannifiée début"),
34         "date_plan_fin": fields.date(string="Date plannifiée de fin"),
35         "date_reel_deb": fields.date(string="Data réelle début"),
36         "date_reel_fin": fields.date(string="Data réelle fin"),
37         "nb_jours_projets": fields.function(_nb_jours_init,
38                                             type="integer",
39                                             string="Nombre de jour"),
40     }
41
42     _sql_constraints = [
43         (
44             "date_init_deb_before_date_init_fin",
45             "CHECK(date_init_deb<> date_init_fin)",
46             "The date_init_deb should be previous date_init_fin",
47         ),
48         (
49             "date_plan_deb_before_date_plan_fin",
50             "CHECK(date_plan_deb <> date_plan_fin)",
51             "The date_plan_deb should be previous date_plan_fin",
52         ),
53         (
54             "date_reel_deb_before_date_reel_fin",
55             "CHECK(date_reel_deb<> date_reel_fin)",
56             "The date_reel_deb should be previous date_reel_fin",
57         ),
58     ]
59
60
61 #=============== TODO TRAVAIL CALCUL JOURS OUVRES =============================
62
63     @staticmethod
64     def _get_date_paques(annee): # TODO vérifier le besoin de self
65         """
66         Retourne la date du dimanque de pâques pour une année donnée
67             sous la forme d'un objet date.
68         """
69         a = annee % 19
70         b = annee // 100
71         c = annee % 100
72         d = (19 * a + b - b // 4 - ((b - (b + 8) // 25 + 1) // 3) + 15) % 30
73         e = (32 + 2 * (b % 4) + 2 * (c // 4) - d - (c % 4)) % 7
74         f = d + e - 7 * ((a + 11 * d + 22 * e) // 451) + 114
75         mois = f // 31
76         jours = f % 31 + 1
77         return date(annee, mois, jours)
78
79
80     @staticmethod
81     def _get_jours_feries(annee):
82         """
83         Retourne une liste contenant les jours fériés d'une année donnée.
84         """
85         date_paques = Mesurable._get_date_paques(annee)
86         return [
87             date_paques+timedelta(days=1), # Lundi de Pâques
88             date_paques+timedelta(days=39), # Jeudi de l'Ascension
89             date_paques+timedelta(days=50), # Lundi de Pentecôte
90             date(annee, 1, 1), # Jour de l'An
91             date(annee, 5, 1), # Fête du travail
92             date(annee, 5, 8), # Armistice 1945
93             date(annee, 7, 14), # Fête Nationale
94             date(annee, 8, 15), # Assomption
95             date(annee, 11, 1), # Toussaint
96             date(annee, 11, 11), # Armistice 1918
97             date(annee, 12, 25), # Noël
98         ]
99
100
101     @staticmethod
102     def _jour_ouvre(d):
103         """ Retourne vrai si la date est un jour ouvre, faux sinon."""
104         if (5,6).__contains__(d.weekday()) or Mesurable._get_jours_feries(d.year).__contains__(d):
105             return False
106         else:
107             return True
108
109
110     @staticmethod
111     def _nb_jours_ouvre_entre_2_dates(d1, d2):
112         """
113         Retourne le nombre de jours ouvres entre deux dates données.
114
115         >>> nb_jours_ouvre_entre_2_dates(date(2013, 5, 1), date(2013, 7, 31))
116         62
117         >>> nb_jours_ouvre_entre_2_dates(date(2013, 7, 31), date(2013, 5, 1))
118         62
119         """
120         if d1>d2:
121             d1, d2 = d2, d1 #Switch les 2 dates pour que d1 soit la plus petite
122         tmp = d1
123         jour_ouvres = 0
124         while tmp <= d2:
125             if Mesurable._jour_ouvre(tmp): # Si tmp est un jour ouvre
126                 jour_ouvres += 1
127             tmp += timedelta(days=1)
128         return jour_ouvres
129
130 #==============================================================================
131
132
133 class Palier(osv.Model):
134     _name = "cmmi.axes.palier"
135
136     _description = "Palier d'un projet."
137
138     _inherit = "cmmi.axes.mesurable"
139
140     _types_palier = [("normal", "Normal"), ("exceptionnel", "Exceptionnel"),
141                      ("correctif", "Correctif"), ("autre", "Autre")]
142
143     _columns = {
144         "type_palier": fields.selection(_types_palier, string="Type"),
145         "projet_id": fields.many2one("cmmi.projet",
146                                      string="Projet",
147                                      required=True),
148         "evolutions": fields.one2many("cmmi.evolution",
149                                       "palier_id",
150                                       string="Evolutions"),
151         "phases": fields.one2many("cmmi.projet.phase",
152                                   "palier_id",
153                                   string="Phases"),
154     }
155
156
157 class Phase(osv.Model):
158     _name = "cmmi.axes.palier.phase"
159
160     _description = "Step's phase / Phase d'un palier"
161
162     _inherit = "cmmi.axes.mesurable"
163
164     def _get_name(self, cr, uid, ids, field_name=None, arg=None, context=None):
165         if isinstance(ids, (int, long)):
166             ids = [ids]
167         return dict([(i, r.phase_id.name) for i, r in
168                 zip(ids, self.browse(cr, uid, ids, context=context))])
169
170
171     _columns = {
172         "name": fields.function(_get_name,
173                                 type='char',
174                                 store=True,
175                                 string="Nom de la phase"),
176         "phase_id": fields.many2one("cmmi.projet.phase",
177                                     string="Type phase"),
178         "palier_id": fields.many2one("cmmi.axes.palier",
179                                      string="Palier"),
180         # backrefs
181         "charges": fields.one2many("cmmi.evolution.charge",
182                                    "phase_id",
183                                    string="Charges"),
184         "evolutions": fields.one2many("cmmi.evolution",
185                                       "phase_id",
186                                       string="Evolutions"),
187     }
188
189     def create(self, cr, uid, vals, context=None):
190         # TODO: gérer la création d'une phase de palier.
191         # Vérifier les valeurs contenues dans vals et les modifier / rajouter si nécessaire selon les cas suivants
192
193         # 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).
194         # Si commentaire est vide, alors par défaut, recopie du commentaire du palier.
195         # Si version est vide, alors par dégaut, recopie de la version du palier.
196
197         return osv.Model.create(self, cr, uid, vals, context=context)
198
199
200 class Chantier(osv.Model):
201     _name = "cmmi.axes.chantier"
202
203     _inherit = "cmmi.axes.mesurable"
204
205     _columns = {
206         "projet_id": fields.many2one("cmmi.projet",
207                                      string="Projet",
208                                      required=True),
209         "evolutions": fields.one2many("cmmi.evolution",
210                                       "chantier_id",
211                                       string="Evolutions"),
212     }