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
  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                     'warning': { # FIXME: ne fonctionne pas
  81                         'title'   : "Warning: not a list",
  82                         'message' : "You cannot have negative number of seats",
  83                         },
  84                     }
  85
  86         self.write(
  87             cr,
  88             uid,
  89             ids, {
  90                 'date_init_deb' : palier['date_plan_deb'],
  91                 'date_init_fin' : palier['date_plan_fin'],
  92                 'state': 'encours'
  93             },
  94             context)
  95         return self
  96
  97
  98     def suspendre(self, cr, uid, ids, context=None):
  99         if type(ids) == list:
 100             if len(ids) != 1:
 101                 return # TODO: message d'avertissement
 102             ids = ids[0]
 103
 104         mesurable = self.read(cr, uid, ids, ['state'], context, context)
 105         if mesurable['state'] != 'encours':
 106             return
 107         self.write(
 108             cr,
 109             uid,
 110             ids,
 111             {'state': 'suspendu'},
 112             context,
 113         )
 114         return self
 115
 116     def terminer(self, cr, uid, ids, context=None):
 117         if type(ids) == list:
 118             if len(ids) != 1:
 119                 return # TODO: message d'avertissement
 120             ids = ids[0]
 121
 122         mesurable = self.read(cr, uid, ids, ['state'], context, context)
 123         if mesurable['state'] != 'encours':
 124             return
 125         self.write(
 126             cr,
 127             uid,
 128             ids,
 129             {'state': 'termine'},
 130             context,
 131         )
 132         return self
 133
 134     def abandonner(self, cr, uid, ids, context=None):
 135         if type(ids) == list:
 136             if len(ids) != 1:
 137                 return # TODO: message d'avertissement
 138             ids = ids[0]
 139
 140         mesurable = self.read(cr, uid, ids, ['state'], context, context)
 141         if not ('encours', 'cree').__contains__(mesurable['state']):
 142             return
 143         self.write(
 144             cr,
 145             uid,
 146             ids,
 147             {'state': 'abandonne'},
 148             context,
 149         )
 150         return self
 151
 152     def reprendre(self, cr, uid, ids, context=None):
 153         if type(ids) == list:
 154             if len(ids) != 1:
 155                 return # TODO: message d'avertissement
 156             ids = ids[0]
 157
 158         mesurable = self.read(cr, uid, ids, ['state'], context, context)
 159         if mesurable['state'] != 'suspendu':
 160             return
 161         self.write(
 162             cr,
 163             uid,
 164             ids,
 165             {'state': 'encours'},
 166             context,
 167         )
 168         return self
 169
 170 #------------ TRAVAIL CALCUL JOURS OUVRES ------------
 171     @staticmethod
 172     def _get_date_paques(annee):
 173         """
 174         Retourne la date du dimanque de pâques pour une année donnée
 175             sous la forme d'un objet date.
 176         """
 177         a = annee % 19
 178         b = annee // 100
 179         c = annee % 100
 180         d = (19 * a + b - b // 4 - ((b - (b + 8) // 25 + 1) // 3) + 15) % 30
 181         e = (32 + 2 * (b % 4) + 2 * (c // 4) - d - (c % 4)) % 7
 182         f = d + e - 7 * ((a + 11 * d + 22 * e) // 451) + 114
 183         mois = f // 31
 184         jours = f % 31 + 1
 185         return date(annee, mois, jours)
 186
 187
 188     @staticmethod
 189     def _get_jours_feries(annee):
 190         """
 191         Retourne une liste contenant les jours fériés d'une année donnée.
 192         """
 193         date_paques = Mesurable._get_date_paques(annee)
 194         return [
 195             date_paques+timedelta(days=1), # Lundi de Pâques
 196             date_paques+timedelta(days=39), # Jeudi de l'Ascension
 197             date_paques+timedelta(days=50), # Lundi de Pentecôte
 198             date(annee, 1, 1), # Jour de l'An
 199             date(annee, 5, 1), # Fête du travail
 200             date(annee, 5, 8), # Armistice 1945
 201             date(annee, 7, 14), # Fête Nationale
 202             date(annee, 8, 15), # Assomption
 203             date(annee, 11, 1), # Toussaint
 204             date(annee, 11, 11), # Armistice 1918
 205             date(annee, 12, 25), # Noël
 206         ]
 207
 208
 209     @staticmethod
 210     def _jour_ouvre(d):
 211         """ Retourne vrai si la date est un jour ouvre, faux sinon."""
 212         if (5,6).__contains__(d.weekday()) or Mesurable._get_jours_feries(d.year).__contains__(d):
 213             return False
 214         else:
 215             return True
 216
 217
 218     @staticmethod
 219     def _nb_jours_ouvre_entre_2_dates(d1, d2):
 220         """
 221         Retourne le nombre de jours ouvres entre deux dates données.
 222
 223         >>> nb_jours_ouvre_entre_2_dates(date(2013, 5, 1), date(2013, 7, 31))
 224         62
 225         >>> nb_jours_ouvre_entre_2_dates(date(2013, 7, 31), date(2013, 5, 1))
 226         62
 227         """
 228         if not d1 or not d2:
 229             return 0
 230         if d1>d2:
 231             d1, d2 = d2, d1 #Switch les 2 dates pour que d1 soit la plus petite
 232         tmp = d1
 233         jour_ouvres = 0
 234         while tmp <= d2:
 235             if Mesurable._jour_ouvre(tmp): # Si tmp est un jour ouvre
 236                 jour_ouvres += 1
 237             tmp += timedelta(days=1)
 238         return jour_ouvres
 239
 240
 241
 242 # ================================= PALIER ================================== #
 243 class Palier(osv.Model):
 244     _name = "cmmi.axes.palier"
 245
 246     _description = "Palier d'un projet."
 247
 248     _inherit = "cmmi.axes.mesurable"
 249
 250     _types_palier = [("normal", "Normal"), ("exceptionnel", "Exceptionnel"),
 251                      ("correctif", "Correctif"), ("autre", "Autre")]
 252
 253     _columns = {
 254         "type_palier": fields.selection(_types_palier, string="Type"),
 255         "projet_id": fields.many2one("cmmi.projet",
 256                                      string="Projet",
 257                                      required=True),
 258         "evolutions": fields.one2many("cmmi.evolution",
 259                                       "palier_id",
 260                                       string="Evolutions"),
 261         "phases": fields.one2many("cmmi.axes.palier.phase",
 262                                   "palier_id",
 263                                   string="Phases"),
 264     }
 265
 266     _defaults = {
 267         "type_palier": "normal",
 268     }
 269
 270
 271     def create(self, cr, uid, vals, context=None):
 272         palier_id = osv.Model.create(self, cr, uid, vals, context=context)
 273
 274         # Récupération des ids de toutes les phases
 275         phase_model = self.pool.get("cmmi.projet.phase")
 276         phases_ids = phase_model.search(cr, uid, [('selectionne', '=', True)])
 277
 278         palier_model = self.pool.get("cmmi.axes.palier")
 279         palier = palier_model.read(cr, uid, palier_id, ['date_plan_deb', 'date_plan_fin'])
 280
 281         # Création des PalierPhase
 282         palier_phase_model = self.pool.get("cmmi.axes.palier.phase")
 283         for phase_id in phases_ids:
 284             palier_phase_model.create(
 285                 cr,
 286                 uid,
 287                 {
 288                     'phase_id': phase_id,
 289                     'palier_id': palier_id,
 290                     'date_plan_deb': palier['date_plan_deb'],
 291                     'date_plan_fin': palier['date_plan_fin'],
 292                 }
 293             )
 294         return palier_id
 295
 296
 297
 298 # =============================== PALIER-PHASE ============================== #
 299 class PalierPhase(osv.Model):
 300     _name = "cmmi.axes.palier.phase"
 301
 302     _description = "Phase d'un palier"
 303
 304     _inherit = "cmmi.axes.mesurable"
 305
 306     def _get_name(self, cr, uid, ids, field_name=None, arg=None, context=None):
 307         if isinstance(ids, (int, long)):
 308             ids = [ids]
 309         return dict([(i, r.phase_id.name) for i, r in
 310                 zip(ids, self.browse(cr, uid, ids, context=context))])
 311
 312
 313     _columns = {
 314         "name": fields.function(_get_name,
 315                                 type='char',
 316                                 store=True,
 317                                 string="Nom de la phase"),
 318         "phase_id": fields.many2one("cmmi.projet.phase",
 319                                     string="Phase du projet"),
 320         "palier_id": fields.many2one("cmmi.axes.palier",
 321                                      string="Palier"),
 322         # backrefs
 323         "charges": fields.one2many("cmmi.evolution.charge",
 324                                    "phase_id",
 325                                    string="Charges"),
 326 #        "evolutions": fields.one2many("cmmi.evolution", #Supprimé !
 327 #                                      "phase_id",
 328 #                                      string="Evolutions"),
 329     }
 330
 331     def create(self, cr, uid, vals, context=None):
 332         # TODO: gérer la création d'une phase de palier.
 333         # Vérifier les valeurs contenues dans vals et les modifier / rajouter si nécessaire selon les cas suivants
 334
 335         # 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).
 336         # Si commentaire est vide, alors par défaut, recopie du commentaire du palier.
 337         # Si version est vide, alors par dégaut, recopie de la version du palier.
 338
 339         return osv.Model.create(self, cr, uid, vals, context=context)
 340
 341
 342 # ================================ CHANTIER ================================= #
 343 class Chantier(osv.Model):
 344     _name = "cmmi.axes.chantier"
 345
 346     _description = "Chantiers d'un projet."
 347
 348     _inherit = "cmmi.axes.mesurable"
 349
 350     _columns = {
 351         "projet_id": fields.many2one("cmmi.projet",
 352                                      string="Projet",
 353                                      required=True),
 354         "module_ids": fields.many2many("cmmi.description.module",
 355                                        "cmmi_module_chantier_rel",
 356                                        "chantier_id",
 357                                        "module_id",
 358                                        "Modules"),
 359         "evolutions": fields.one2many("cmmi.evolution",
 360                                       "chantier_id",
 361                                       string="Evolutions"),
 362     }