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