2010-01-12 12:45:02 +00:00
# -*- coding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
from lxml import etree
2010-02-18 10:07:31 +00:00
import mx . DateTime
2010-01-12 12:45:02 +00:00
import time
2010-03-11 07:10:06 +00:00
2010-01-12 12:45:02 +00:00
from tools . translate import _
from osv import fields , osv
2010-02-18 10:07:31 +00:00
2010-01-12 12:45:02 +00:00
class project_phase ( osv . osv ) :
_name = " project.phase "
_description = " Project Phase "
2010-07-02 14:15:50 +00:00
def _check_recursion ( self , cr , uid , ids , context = None ) :
if context is None :
context = { }
2010-03-11 07:10:06 +00:00
data_phase = self . browse ( cr , uid , ids [ 0 ] , context = context )
prev_ids = data_phase . previous_phase_ids
next_ids = data_phase . next_phase_ids
2010-05-27 10:00:31 +00:00
# it should neither be in prev_ids nor in next_ids
2010-03-11 07:10:06 +00:00
if ( data_phase in prev_ids ) or ( data_phase in next_ids ) :
2010-01-18 13:44:53 +00:00
return False
ids = [ id for id in prev_ids if id in next_ids ]
2010-03-09 11:52:35 +00:00
# both prev_ids and next_ids must be unique
2010-01-18 13:44:53 +00:00
if ids :
return False
2010-03-09 11:52:35 +00:00
# unrelated project
2010-01-18 13:44:53 +00:00
prev_ids = [ rec . id for rec in prev_ids ]
next_ids = [ rec . id for rec in next_ids ]
2010-03-09 11:52:35 +00:00
# iter prev_ids
2010-01-18 13:44:53 +00:00
while prev_ids :
2010-07-02 14:15:50 +00:00
cr . execute ( ' SELECT distinct prv_phase_id FROM project_phase_rel WHERE next_phase_id IN %s ' , ( tuple ( prev_ids ) , ) )
2010-01-18 13:44:53 +00:00
prv_phase_ids = filter ( None , map ( lambda x : x [ 0 ] , cr . fetchall ( ) ) )
2010-03-11 07:10:06 +00:00
if data_phase . id in prv_phase_ids :
2010-01-18 13:44:53 +00:00
return False
ids = [ id for id in prv_phase_ids if id in next_ids ]
if ids :
return False
prev_ids = prv_phase_ids
2010-03-09 11:52:35 +00:00
# iter next_ids
2010-01-18 13:44:53 +00:00
while next_ids :
2010-07-02 14:15:50 +00:00
cr . execute ( ' SELECT distinct next_phase_id FROM project_phase_rel WHERE prv_phase_id IN %s ' , ( tuple ( next_ids ) , ) )
2010-01-18 13:44:53 +00:00
next_phase_ids = filter ( None , map ( lambda x : x [ 0 ] , cr . fetchall ( ) ) )
2010-03-11 07:10:06 +00:00
if data_phase . id in next_phase_ids :
2010-01-18 13:44:53 +00:00
return False
ids = [ id for id in next_phase_ids if id in prev_ids ]
if ids :
return False
next_ids = next_phase_ids
return True
2010-01-12 12:45:02 +00:00
2010-07-02 14:15:50 +00:00
def _check_dates ( self , cr , uid , ids , context = None ) :
2010-04-06 16:15:10 +00:00
for phase in self . read ( cr , uid , ids , [ ' date_start ' , ' date_end ' ] , context = context ) :
if phase [ ' date_start ' ] and phase [ ' date_end ' ] and phase [ ' date_start ' ] > phase [ ' date_end ' ] :
return False
2010-02-26 11:18:14 +00:00
return True
2010-07-02 14:15:50 +00:00
def _check_constraint_start ( self , cr , uid , ids , context = None ) :
2010-03-11 07:10:06 +00:00
phase = self . read ( cr , uid , ids [ 0 ] , [ ' date_start ' , ' constraint_date_start ' ] , context = context )
if phase [ ' date_start ' ] and phase [ ' constraint_date_start ' ] and phase [ ' date_start ' ] < phase [ ' constraint_date_start ' ] :
return False
2010-02-26 14:00:28 +00:00
return True
2010-07-02 14:15:50 +00:00
def _check_constraint_end ( self , cr , uid , ids , context = None ) :
2010-03-11 07:10:06 +00:00
phase = self . read ( cr , uid , ids [ 0 ] , [ ' date_end ' , ' constraint_date_end ' ] , context = context )
if phase [ ' date_end ' ] and phase [ ' constraint_date_end ' ] and phase [ ' date_end ' ] > phase [ ' constraint_date_end ' ] :
return False
2010-02-26 14:00:28 +00:00
return True
2010-02-18 10:07:31 +00:00
2010-07-08 12:53:59 +00:00
def _get_default_uom_id ( self , cr , uid ) :
model_data_obj = self . pool . get ( ' ir.model.data ' )
model_data_id = model_data_obj . _get_id ( cr , uid , ' product ' , ' uom_hour ' )
return model_data_obj . read ( cr , uid , [ model_data_id ] , [ ' res_id ' ] ) [ 0 ] [ ' res_id ' ]
2010-01-12 12:45:02 +00:00
_columns = {
2010-05-21 07:14:29 +00:00
' name ' : fields . char ( " Name " , size = 64 , required = True ) ,
2010-07-14 09:04:50 +00:00
' date_start ' : fields . date ( ' Start Date ' , help = " Starting Date of the phase " ) ,
' date_end ' : fields . date ( ' End Date ' , help = " Ending Date of the phase " ) ,
2010-03-11 11:17:29 +00:00
' constraint_date_start ' : fields . datetime ( ' Start Date ' , help = ' force the phase to start after this date ' ) ,
' constraint_date_end ' : fields . datetime ( ' End Date ' , help = ' force the phase to finish before this date ' ) ,
2010-01-12 12:45:02 +00:00
' project_id ' : fields . many2one ( ' project.project ' , ' Project ' , required = True ) ,
2010-02-25 06:01:27 +00:00
' next_phase_ids ' : fields . many2many ( ' project.phase ' , ' project_phase_rel ' , ' prv_phase_id ' , ' next_phase_id ' , ' Next Phases ' ) ,
' previous_phase_ids ' : fields . many2many ( ' project.phase ' , ' project_phase_rel ' , ' next_phase_id ' , ' prv_phase_id ' , ' Previous Phases ' ) ,
' sequence ' : fields . integer ( ' Sequence ' , help = " Gives the sequence order when displaying a list of phases. " ) ,
2010-05-27 13:18:57 +00:00
' duration ' : fields . float ( ' Duration ' , required = True , help = " By default in days " ) ,
2010-03-09 11:52:35 +00:00
' product_uom ' : fields . many2one ( ' product.uom ' , ' Duration UoM ' , required = True , help = " UoM (Unit of Measure) is the unit of measurement for Duration " ) ,
2010-01-12 12:45:02 +00:00
' task_ids ' : fields . one2many ( ' project.task ' , ' phase_id ' , " Project Tasks " ) ,
' resource_ids ' : fields . one2many ( ' project.resource.allocation ' , ' phase_id ' , " Project Resources " ) ,
2010-07-02 14:15:50 +00:00
' responsible_id ' : fields . many2one ( ' res.users ' , ' Responsible ' ) ,
2010-03-09 11:52:35 +00:00
' state ' : fields . selection ( [ ( ' draft ' , ' Draft ' ) , ( ' open ' , ' In Progress ' ) , ( ' pending ' , ' Pending ' ) , ( ' cancelled ' , ' Cancelled ' ) , ( ' done ' , ' Done ' ) ] , ' State ' , readonly = True , required = True ,
2010-02-25 06:01:27 +00:00
help = ' If the phase is created the state \' Draft \' . \n If the phase is started, the state becomes \' In Progress \' . \n If review is needed the phase is in \' Pending \' state. \
\n If the phase is over , the states is set to \' Done \' . ' )
2010-01-12 12:45:02 +00:00
}
_defaults = {
2010-02-25 06:01:27 +00:00
' responsible_id ' : lambda obj , cr , uid , context : uid ,
2010-05-14 09:15:02 +00:00
' state ' : ' draft ' ,
' sequence ' : 10 ,
2010-07-14 09:04:50 +00:00
' product_uom ' : lambda self , cr , uid , c : self . pool . get ( ' product.uom ' ) . search ( cr , uid , [ ( ' name ' , ' = ' , _ ( ' Day ' ) ) ] , context = c ) [ 0 ]
2010-01-12 12:45:02 +00:00
}
_order = " name "
2010-02-26 11:18:14 +00:00
_constraints = [
2010-04-06 16:15:10 +00:00
( _check_recursion , ' Loops in phases not allowed ' , [ ' next_phase_ids ' , ' previous_phase_ids ' ] ) ,
( _check_dates , ' Phase start-date must be lower than phase end-date. ' , [ ' date_start ' , ' date_end ' ] ) ,
#(_check_constraint_start, 'Phase must start-after constraint start Date.', ['date_start', 'constraint_date_start']),
#(_check_constraint_end, 'Phase must end-before constraint end Date.', ['date_end', 'constraint_date_end']),
2010-02-26 11:18:14 +00:00
]
2010-07-02 14:15:50 +00:00
def onchange_project ( self , cr , uid , ids , project , context = None ) :
2010-02-26 11:18:14 +00:00
result = { }
2010-03-15 08:49:13 +00:00
project_obj = self . pool . get ( ' project.project ' )
2010-02-26 11:18:14 +00:00
if project :
2010-03-15 08:49:13 +00:00
project_id = project_obj . browse ( cr , uid , project , context = context )
2010-03-11 07:10:06 +00:00
if project_id . date_start :
result [ ' date_start ' ] = mx . DateTime . strptime ( project_id . date_start , " % Y- % m- %d " ) . strftime ( ' % Y- % m- %d % H: % M: % S ' )
return { ' value ' : result }
2010-03-09 11:52:35 +00:00
return { ' value ' : { ' date_start ' : [ ] } }
2010-07-02 14:15:50 +00:00
def _check_date_start ( self , cr , uid , phase , date_end , context = None ) :
if context is None :
context = { }
2010-03-09 11:52:35 +00:00
"""
Check And Compute date_end of phase if change in date_start < older time .
"""
uom_obj = self . pool . get ( ' product.uom ' )
resource_obj = self . pool . get ( ' resource.resource ' )
2010-03-15 08:49:13 +00:00
cal_obj = self . pool . get ( ' resource.calendar ' )
2010-03-11 07:10:06 +00:00
calendar_id = phase . project_id . resource_calendar_id and phase . project_id . resource_calendar_id . id or False
2010-03-09 11:52:35 +00:00
resource_id = resource_obj . search ( cr , uid , [ ( ' user_id ' , ' = ' , phase . responsible_id . id ) ] )
2010-03-02 13:27:02 +00:00
if resource_id :
2010-03-16 09:30:14 +00:00
# cal_id = resource_obj.browse(cr, uid, resource_id[0], context=context).calendar_id.id
res = resource_obj . read ( cr , uid , resource_id [ 0 ] , [ ' calendar_id ' ] , context = context ) [ 0 ]
cal_id = res . get ( ' calendar_id ' , False ) and res . get ( ' calendar_id ' ) [ 0 ] or False
2010-03-09 11:52:35 +00:00
if cal_id :
calendar_id = cal_id
2010-07-08 12:53:59 +00:00
default_uom_id = self . _get_default_uom_id ( cr , uid )
2010-03-09 11:52:35 +00:00
avg_hours = uom_obj . _compute_qty ( cr , uid , phase . product_uom . id , phase . duration , default_uom_id )
2010-03-15 08:49:13 +00:00
work_times = cal_obj . interval_min_get ( cr , uid , calendar_id , date_end , avg_hours or 0.0 , resource_id and resource_id [ 0 ] or False )
2010-03-09 11:52:35 +00:00
dt_start = work_times [ 0 ] [ 0 ] . strftime ( ' % Y- % m- %d % H: % M: % S ' )
2010-03-11 07:10:06 +00:00
self . write ( cr , uid , [ phase . id ] , { ' date_start ' : dt_start , ' date_end ' : date_end . strftime ( ' % Y- % m- %d % H: % M: % S ' ) } , context = context )
2010-03-09 11:52:35 +00:00
2010-07-02 14:15:50 +00:00
def _check_date_end ( self , cr , uid , phase , date_start , context = None ) :
if context is None :
context = { }
2010-03-09 11:52:35 +00:00
"""
Check And Compute date_end of phase if change in date_end > older time .
"""
uom_obj = self . pool . get ( ' product.uom ' )
resource_obj = self . pool . get ( ' resource.resource ' )
2010-03-15 08:49:13 +00:00
cal_obj = self . pool . get ( ' resource.calendar ' )
2010-03-11 07:10:06 +00:00
calendar_id = phase . project_id . resource_calendar_id and phase . project_id . resource_calendar_id . id or False
resource_id = resource_obj . search ( cr , uid , [ ( ' user_id ' , ' = ' , phase . responsible_id . id ) ] , context = context )
2010-03-02 13:27:02 +00:00
if resource_id :
2010-03-16 09:30:14 +00:00
# cal_id = resource_obj.browse(cr, uid, resource_id[0], context=context).calendar_id.id
res = resource_obj . read ( cr , uid , resource_id [ 0 ] , [ ' calendar_id ' ] , context = context ) [ 0 ]
cal_id = res . get ( ' calendar_id ' , False ) and res . get ( ' calendar_id ' ) [ 0 ] or False
2010-03-09 11:52:35 +00:00
if cal_id :
calendar_id = cal_id
2010-07-08 12:53:59 +00:00
default_uom_id = self . _get_default_uom_id ( cr , uid )
2010-03-09 11:52:35 +00:00
avg_hours = uom_obj . _compute_qty ( cr , uid , phase . product_uom . id , phase . duration , default_uom_id )
2010-03-15 08:49:13 +00:00
work_times = cal_obj . interval_get ( cr , uid , calendar_id , date_start , avg_hours or 0.0 , resource_id and resource_id [ 0 ] or False )
2010-03-09 11:52:35 +00:00
dt_end = work_times [ - 1 ] [ 1 ] . strftime ( ' % Y- % m- %d % H: % M: % S ' )
2010-03-11 07:10:06 +00:00
self . write ( cr , uid , [ phase . id ] , { ' date_start ' : date_start . strftime ( ' % Y- % m- %d % H: % M: % S ' ) , ' date_end ' : dt_end } , context = context )
2010-02-26 11:18:14 +00:00
2010-07-02 14:15:50 +00:00
def write ( self , cr , uid , ids , vals , context = None ) :
2010-03-15 08:49:13 +00:00
resource_calendar_obj = self . pool . get ( ' resource.calendar ' )
resource_obj = self . pool . get ( ' resource.resource ' )
uom_obj = self . pool . get ( ' product.uom ' )
2010-07-02 14:15:50 +00:00
if context is None :
2010-02-26 11:18:14 +00:00
context = { }
if context . get ( ' scheduler ' , False ) :
return super ( project_phase , self ) . write ( cr , uid , ids , vals , context = context )
2010-03-09 11:52:35 +00:00
# Consider calendar and efficiency if the phase is performed by a resource
# otherwise consider the project's working calendar
2010-03-11 07:10:06 +00:00
phase = self . browse ( cr , uid , ids [ 0 ] , context = context )
2010-03-16 09:41:40 +00:00
calendar_id = phase . project_id . resource_calendar_id and phase . project_id . resource_calendar_id . id or False
2010-03-11 07:10:06 +00:00
resource_id = resource_obj . search ( cr , uid , [ ( ' user_id ' , ' = ' , phase . responsible_id . id ) ] , context = context )
2010-03-02 13:27:02 +00:00
if resource_id :
2010-03-11 07:10:06 +00:00
cal_id = resource_obj . browse ( cr , uid , resource_id [ 0 ] , context = context ) . calendar_id . id
2010-03-09 11:52:35 +00:00
if cal_id :
calendar_id = cal_id
2010-07-08 12:53:59 +00:00
default_uom_id = self . _get_default_uom_id ( cr , uid )
2010-03-09 11:52:35 +00:00
avg_hours = uom_obj . _compute_qty ( cr , uid , phase . product_uom . id , phase . duration , default_uom_id )
# Change the date_start and date_end
# for previous and next phases respectively based on valid condition
2010-03-11 07:10:06 +00:00
if vals . get ( ' date_start ' , False ) and vals [ ' date_start ' ] < phase . date_start :
2010-07-02 14:15:50 +00:00
dt_start = mx . DateTime . strptime ( vals [ ' date_start ' ] , ' % Y- % m- %d % H: % M: % S ' )
2010-03-16 09:41:40 +00:00
work_times = resource_calendar_obj . interval_get ( cr , uid , calendar_id , dt_start , avg_hours or 0.0 , resource_id and resource_id [ 0 ] or False )
2010-03-22 13:28:36 +00:00
if work_times :
vals [ ' date_end ' ] = work_times [ - 1 ] [ 1 ] . strftime ( ' % Y- % m- %d % H: % M: % S ' )
2010-02-26 11:18:14 +00:00
for prv_phase in phase . previous_phase_ids :
2010-03-11 07:10:06 +00:00
self . _check_date_start ( cr , uid , prv_phase , dt_start , context = context )
if vals . get ( ' date_end ' , False ) and vals [ ' date_end ' ] > phase . date_end :
2010-02-26 11:18:14 +00:00
dt_end = mx . DateTime . strptime ( vals [ ' date_end ' ] , ' % Y- % m- %d % H: % M: % S ' )
2010-03-16 09:41:40 +00:00
work_times = resource_calendar_obj . interval_min_get ( cr , uid , calendar_id , dt_end , avg_hours or 0.0 , resource_id and resource_id [ 0 ] or False )
2010-03-22 13:28:36 +00:00
if work_times :
vals [ ' date_start ' ] = work_times [ 0 ] [ 0 ] . strftime ( ' % Y- % m- %d % H: % M: % S ' )
2010-02-26 11:18:14 +00:00
for next_phase in phase . next_phase_ids :
2010-03-11 07:10:06 +00:00
self . _check_date_end ( cr , uid , next_phase , dt_end , context = context )
2010-02-26 11:18:14 +00:00
return super ( project_phase , self ) . write ( cr , uid , ids , vals , context = context )
2010-02-25 06:01:27 +00:00
2010-03-09 11:52:35 +00:00
def set_draft ( self , cr , uid , ids , * args ) :
2010-02-25 06:01:27 +00:00
self . write ( cr , uid , ids , { ' state ' : ' draft ' } )
return True
2010-03-09 11:52:35 +00:00
def set_open ( self , cr , uid , ids , * args ) :
2010-02-25 06:01:27 +00:00
self . write ( cr , uid , ids , { ' state ' : ' open ' } )
return True
2010-07-02 14:15:50 +00:00
def set_pending ( self , cr , uid , ids , * args ) :
2010-02-25 06:01:27 +00:00
self . write ( cr , uid , ids , { ' state ' : ' pending ' } )
return True
2010-03-09 11:52:35 +00:00
def set_cancel ( self , cr , uid , ids , * args ) :
2010-02-25 06:01:27 +00:00
self . write ( cr , uid , ids , { ' state ' : ' cancelled ' } )
return True
2010-03-09 11:52:35 +00:00
def set_done ( self , cr , uid , ids , * args ) :
2010-02-25 06:01:27 +00:00
self . write ( cr , uid , ids , { ' state ' : ' done ' } )
return True
2010-02-18 10:07:31 +00:00
2010-01-12 12:45:02 +00:00
project_phase ( )
class project_resource_allocation ( osv . osv ) :
_name = ' project.resource.allocation '
_description = ' Project Resource Allocation '
2010-02-10 06:14:22 +00:00
_rec_name = ' resource_id '
2010-01-12 12:45:02 +00:00
_columns = {
2010-07-23 16:08:54 +00:00
' resource_id ' : fields . many2one ( ' resource.resource ' , ' Resource ' , ondelete = ' cascade ' , required = True ) ,
2010-03-09 11:52:35 +00:00
' phase_id ' : fields . many2one ( ' project.phase ' , ' Project Phase ' , required = True ) ,
2010-07-13 23:21:26 +00:00
' phase_id_date_start ' : fields . related ( ' phase_id ' , ' date_start ' , type = ' date ' , string = ' Starting Date of the phase ' ) ,
' phase_id_date_end ' : fields . related ( ' phase_id ' , ' date_end ' , type = ' date ' , string = ' Ending Date of the phase ' ) ,
2010-05-27 10:00:31 +00:00
' useability ' : fields . float ( ' Usability ' , help = " Usability of this resource for this project phase in percentage (=50 % ) " ) ,
2010-07-09 14:38:43 +00:00
' date_start ' : fields . related ( ' phase_id ' , ' date_start ' , type = ' datetime ' , string = ' Start Date ' ) ,
' date_end ' : fields . related ( ' phase_id ' , ' date_end ' , type = ' datetime ' , string = ' End Date ' ) ,
2010-01-12 12:45:02 +00:00
}
_defaults = {
2010-05-14 09:15:02 +00:00
' useability ' : 100 ,
2010-01-12 12:45:02 +00:00
}
2010-02-10 06:14:22 +00:00
2010-01-12 12:45:02 +00:00
project_resource_allocation ( )
class project ( osv . osv ) :
_inherit = " project.project "
_columns = {
2010-03-03 07:53:14 +00:00
' phase_ids ' : fields . one2many ( ' project.phase ' , ' project_id ' , " Project Phases " ) ,
' resource_calendar_id ' : fields . many2one ( ' resource.calendar ' , ' Working Time ' , help = " Timetable working hours to adjust the gantt diagram report " ) ,
2010-01-12 12:45:02 +00:00
}
project ( )
class task ( osv . osv ) :
_inherit = " project.task "
_columns = {
2010-03-02 13:27:02 +00:00
' phase_id ' : fields . many2one ( ' project.phase ' , ' Project Phase ' ) ,
2010-01-12 12:45:02 +00:00
}
2010-03-02 13:27:02 +00:00
2010-01-12 12:45:02 +00:00
task ( )
2010-05-19 18:32:32 +00:00
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: