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/>.
#
##############################################################################
2010-08-12 07:31:45 +00:00
2010-09-27 13:44:03 +00:00
from datetime import datetime
2010-09-30 07:59:42 +00:00
from dateutil . relativedelta import relativedelta
2010-01-12 12:45:02 +00:00
from tools . translate import _
from osv import fields , osv
2010-09-24 09:26:14 +00:00
from resource . faces import task as Task
import operator
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-10-12 13:21:51 +00:00
2010-10-13 10:44:06 +00:00
def _compute ( self , cr , uid , ids , field_name , arg , context = None ) :
2010-09-29 15:07:02 +00:00
res = { }
if not ids :
return res
for phase in self . browse ( cr , uid , ids , context = context ) :
tot = 0.0
for task in phase . task_ids :
tot + = task . planned_hours
2010-10-06 11:12:47 +00:00
res [ phase . id ] = tot
2010-09-29 15:07:02 +00:00
return res
2010-07-08 12:53:59 +00:00
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-29 05:31:04 +00:00
' date_start ' : fields . date ( ' Start Date ' , help = " Starting Date of the phase " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' date_end ' : fields . date ( ' End Date ' , help = " Ending Date of the phase " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
2010-09-30 14:41:19 +00:00
' constraint_date_start ' : fields . date ( ' Minimum Start Date ' , help = ' force the phase to start after this date ' , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' constraint_date_end ' : fields . date ( ' Deadline ' , help = ' force the phase to finish before this date ' , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
2010-01-12 12:45:02 +00:00
' project_id ' : fields . many2one ( ' project.project ' , ' Project ' , required = True ) ,
2010-07-29 05:31:04 +00:00
' next_phase_ids ' : fields . many2many ( ' project.phase ' , ' project_phase_rel ' , ' prv_phase_id ' , ' next_phase_id ' , ' Next Phases ' , states = { ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' previous_phase_ids ' : fields . many2many ( ' project.phase ' , ' project_phase_rel ' , ' next_phase_id ' , ' prv_phase_id ' , ' Previous Phases ' , states = { ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
2010-02-25 06:01:27 +00:00
' sequence ' : fields . integer ( ' Sequence ' , help = " Gives the sequence order when displaying a list of phases. " ) ,
2010-07-29 05:31:04 +00:00
' duration ' : fields . float ( ' Duration ' , required = True , help = " By default in days " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' product_uom ' : fields . many2one ( ' product.uom ' , ' Duration UoM ' , required = True , help = " UoM (Unit of Measure) is the unit of measurement for Duration " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' task_ids ' : fields . one2many ( ' project.task ' , ' phase_id ' , " Project Tasks " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' resource_ids ' : fields . one2many ( ' project.resource.allocation ' , ' phase_id ' , " Project Resources " , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
' responsible_id ' : fields . many2one ( ' res.users ' , ' Responsible ' , states = { ' done ' : [ ( ' readonly ' , True ) ] , ' cancelled ' : [ ( ' readonly ' , True ) ] } ) ,
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. \
2010-09-29 15:07:02 +00:00
\n If the phase is over , the states is set to \' Done \' . ' ) ,
' total_hours ' : fields . function ( _compute , method = True , string = ' Total Hours ' ) ,
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
}
2010-09-14 12:57:44 +00:00
_order = " project_id, date_start, sequence, 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 ' ] ) ,
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-08-12 07:31:45 +00:00
result [ ' date_start ' ] = False
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-08-12 07:31:45 +00:00
result [ ' date_start ' ] = project_id . date_start
return { ' value ' : result }
2010-10-12 13:21:51 +00:00
2010-09-29 15:07:02 +00:00
def onchange_days ( self , cr , uid , ids , project , context = None ) :
result = { }
for id in ids :
project_id = self . browse ( cr , uid , id , context = context )
2010-09-30 07:59:42 +00:00
newdate = datetime . strptime ( project_id . date_start , ' % Y- % m- %d ' ) + relativedelta ( days = project_id . duration or 0.0 )
2010-09-29 15:07:02 +00:00
result [ ' date_end ' ] = newdate . strftime ( ' % Y- % m- %d ' )
return { ' value ' : result }
2010-03-09 11:52:35 +00:00
2010-07-02 14:15:50 +00:00
def _check_date_start ( self , cr , uid , phase , date_end , context = None ) :
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-07-29 05:31:04 +00:00
res = resource_obj . read ( cr , uid , resource_id , [ ' calendar_id ' ] , context = context ) [ 0 ]
2010-03-16 09:30:14 +00:00
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-07-29 05:31:04 +00:00
dt_start = work_times [ 0 ] [ 0 ] . strftime ( ' % Y- % m- %d ' )
self . write ( cr , uid , [ phase . id ] , { ' date_start ' : dt_start , ' date_end ' : date_end . strftime ( ' % Y- % m- %d ' ) } , 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 ) :
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-07-29 05:31:04 +00:00
res = resource_obj . read ( cr , uid , resource_id , [ ' calendar_id ' ] , context = context ) [ 0 ]
2010-03-16 09:30:14 +00:00
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-07-29 05:31:04 +00:00
dt_end = work_times [ - 1 ] [ 1 ] . strftime ( ' % Y- % m- %d ' )
self . write ( cr , uid , [ phase . id ] , { ' date_start ' : date_start . strftime ( ' % Y- % m- %d ' ) , ' 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 = { }
2010-09-24 09:26:14 +00:00
res = super ( project_phase , self ) . write ( cr , uid , ids , vals , context = context )
2010-02-26 11:18:14 +00:00
if context . get ( ' scheduler ' , False ) :
2010-09-24 09:26:14 +00:00
return res
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-09-24 09:26:14 +00:00
#TOCHECK : why need this ?
2010-07-30 12:35:21 +00:00
if isinstance ( ids , ( int , long ) ) :
2010-07-25 15:55:48 +00:00
ids = [ ids ]
2010-09-24 09:26:14 +00:00
default_uom_id = self . _get_default_uom_id ( cr , uid )
for phase in self . browse ( cr , uid , ids , context = context ) :
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 )
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-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +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
if vals . get ( ' date_start ' , False ) and vals [ ' date_start ' ] < phase . date_start :
2010-08-12 07:31:45 +00:00
dt_start = datetime . strptime ( vals [ ' date_start ' ] , ' % Y- % m- %d ' )
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 :
2010-07-29 05:31:04 +00:00
vals [ ' date_end ' ] = work_times [ - 1 ] [ 1 ] . strftime ( ' % Y- % m- %d ' )
2010-02-26 11:18:14 +00:00
for prv_phase in phase . previous_phase_ids :
2010-09-24 09:26:14 +00:00
if prv_phase . id == phase . id :
continue
2010-03-11 07:10:06 +00:00
self . _check_date_start ( cr , uid , prv_phase , dt_start , context = context )
2010-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +00:00
if vals . get ( ' date_end ' , False ) and vals [ ' date_end ' ] > phase . date_end :
2010-08-12 07:31:45 +00:00
dt_end = datetime . strptime ( vals [ ' date_end ' ] , ' % Y- % m- %d ' )
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 :
2010-07-29 05:31:04 +00:00
vals [ ' date_start ' ] = work_times [ 0 ] [ 0 ] . strftime ( ' % Y- % m- %d ' )
2010-02-26 11:18:14 +00:00
for next_phase in phase . next_phase_ids :
2010-09-24 09:26:14 +00:00
if next_phase . id == phase . id :
continue
2010-03-11 07:10:06 +00:00
self . _check_date_end ( cr , uid , next_phase , dt_end , context = context )
2010-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +00:00
return res
2010-09-18 05:52:11 +00:00
def copy ( self , cr , uid , id , default = None , context = None ) :
if default is None :
default = { }
if not default . get ( ' name ' , False ) :
default [ ' name ' ] = self . browse ( cr , uid , id , context = context ) . name + _ ( ' (copy) ' )
return super ( project_phase , self ) . copy ( cr , uid , id , default , 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-09-24 09:26:14 +00:00
def generate_resources ( self , cr , uid , ids , context = None ) :
"""
Return a list of Resource Class objects for the resources allocated to the phase .
"""
res = { }
resource_pool = self . pool . get ( ' resource.resource ' )
for phase in self . browse ( cr , uid , ids , context = context ) :
user_ids = map ( lambda x : x . resource_id . user_id . id , phase . resource_ids )
project = phase . project_id
calendar_id = project . resource_calendar_id and project . resource_calendar_id . id or False
resource_objs = resource_pool . generate_resources ( cr , uid , user_ids , calendar_id , context = context )
res [ phase . id ] = resource_objs
return res
2010-12-30 08:51:44 +00:00
def generate_schedule ( self , cr , uid , ids , start_date = False , calendar_id = False , context = None ) :
2010-09-24 09:26:14 +00:00
"""
Schedule phase with the start date till all the next phases are completed .
2010-12-30 08:51:44 +00:00
@param : start_date ( datetime . datetime ) : start date for the phase . It would be either Start date of phase or start date of project or system current date
2010-09-24 09:26:14 +00:00
@param : calendar_id : working calendar of the project
2010-10-12 13:21:51 +00:00
"""
2010-09-24 09:26:14 +00:00
if context is None :
context = { }
resource_pool = self . pool . get ( ' resource.resource ' )
2010-10-14 10:26:54 +00:00
resource_allocation_pool = self . pool . get ( ' project.resource.allocation ' )
2010-09-24 09:26:14 +00:00
uom_pool = self . pool . get ( ' product.uom ' )
default_uom_id = self . _get_default_uom_id ( cr , uid )
for phase in self . browse ( cr , uid , ids , context = context ) :
if not phase . responsible_id :
raise osv . except_osv ( _ ( ' No responsible person assigned ! ' ) , _ ( " You must assign a responsible person for phase ' %s ' ! " ) % ( phase . name , ) )
2010-10-12 13:21:51 +00:00
2010-12-30 08:51:44 +00:00
if not start_date :
start_date = phase . project_id . date_start or phase . date_start or datetime . now ( ) . strftime ( " % Y- % m- %d " )
start_date = datetime . strftime ( ( datetime . strptime ( start_date , " % Y- % m- %d " ) ) , " % Y- % m- %d " )
2010-09-24 09:26:14 +00:00
phase_resource_obj = resource_pool . generate_resources ( cr , uid , [ phase . responsible_id . id ] , calendar_id , context = context )
avg_hours = uom_pool . _compute_qty ( cr , uid , phase . product_uom . id , phase . duration , default_uom_id )
duration = str ( avg_hours ) + ' H '
# Create a new project for each phase
def Project ( ) :
# If project has working calendar then that
# else the default one would be considered
2010-09-28 13:52:54 +00:00
start = start_date
minimum_time_unit = 1
resource = phase_resource_obj
2010-09-24 09:26:14 +00:00
if calendar_id :
working_days = resource_pool . compute_working_calendar ( cr , uid , calendar_id , context = context )
vacation = tuple ( resource_pool . compute_vacation ( cr , uid , calendar_id ) )
def phase ( ) :
effort = duration
project = Task . BalancedProject ( Project )
s_date = project . phase . start . to_datetime ( )
e_date = project . phase . end . to_datetime ( )
# Recalculate date_start and date_end
# according to constraints on date start and date end on phase
if phase . constraint_date_start and str ( s_date ) < phase . constraint_date_start :
start_date = datetime . strptime ( phase . constraint_date_start , ' % Y- % m- %d ' )
else :
start_date = s_date
if phase . constraint_date_end and str ( e_date ) > phase . constraint_date_end :
end_date = datetime . strptime ( phase . constraint_date_end , ' % Y- % m- %d ' )
date_start = phase . constraint_date_end
else :
end_date = e_date
date_start = end_date
# Write the calculated dates back
ctx = context . copy ( )
ctx . update ( { ' scheduler ' : True } )
self . write ( cr , uid , [ phase . id ] , {
' date_start ' : start_date . strftime ( ' % Y- % m- %d ' ) ,
' date_end ' : end_date . strftime ( ' % Y- % m- %d ' )
} , context = ctx )
2010-10-14 10:26:54 +00:00
# write dates into Resources Allocation
for resource in phase . resource_ids :
resource_allocation_pool . write ( cr , uid , [ resource . id ] , {
' date_start ' : start_date . strftime ( ' % Y- % m- %d ' ) ,
' date_end ' : end_date . strftime ( ' % Y- % m- %d ' )
} , context = ctx )
2010-09-24 09:26:14 +00:00
# Recursive call till all the next phases scheduled
for phase in phase . next_phase_ids :
if phase . state in [ ' draft ' , ' open ' , ' pending ' ] :
id_cal = phase . project_id . resource_calendar_id and phase . project_id . resource_calendar_id . id or False
self . generate_schedule ( cr , uid , [ phase . id ] , date_start , id_cal , context = context )
else :
continue
return True
def schedule_tasks ( self , cr , uid , ids , context = None ) :
"""
Schedule the tasks according to resource available and priority .
"""
task_pool = self . pool . get ( ' project.task ' )
resource_pool = self . pool . get ( ' resource.resource ' )
resources_list = self . generate_resources ( cr , uid , ids , context = context )
return_msg = { }
for phase in self . browse ( cr , uid , ids , context = context ) :
start_date = phase . date_start
if not start_date and phase . project_id . date_start :
start_date = phase . project_id . date_start
if not start_date :
2010-10-12 13:21:51 +00:00
start_date = datetime . now ( ) . strftime ( " % Y- % m- %d " )
2010-09-24 09:26:14 +00:00
resources = resources_list . get ( phase . id , [ ] )
calendar_id = phase . project_id . resource_calendar_id . id
task_ids = map ( lambda x : x . id , ( filter ( lambda x : x . state in [ ' open ' , ' draft ' , ' pending ' ] , phase . task_ids ) ) )
if task_ids :
task_pool . generate_schedule ( cr , uid , task_ids , resources , calendar_id , start_date , context = context )
if not task_ids :
warning_msg = _ ( " No tasks to compute for Phase ' %s ' . " ) % ( phase . name )
if " warning " not in return_msg :
return_msg [ " warning " ] = warning_msg
else :
return_msg [ " warning " ] = return_msg [ " warning " ] + " \n " + warning_msg
return return_msg
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-10-15 15:47:01 +00:00
def get_name ( self , cr , uid , ids , field_name , arg , context = None ) :
res = { }
for allocation in self . browse ( cr , uid , ids , context = context ) :
2010-10-15 22:01:00 +00:00
name = allocation . phase_id . name
2010-10-15 15:47:01 +00:00
name + = ' ( %s %% ) ' % ( allocation . useability )
res [ allocation . id ] = name
return res
2010-01-12 12:45:02 +00:00
_columns = {
2010-10-15 15:47:01 +00:00
' name ' : fields . function ( get_name , method = True , type = ' char ' , size = 256 ) ,
2010-01-12 12:45:02 +00:00
' resource_id ' : fields . many2one ( ' resource.resource ' , ' Resource ' , required = True ) ,
2010-07-25 15:55:48 +00:00
' phase_id ' : fields . many2one ( ' project.phase ' , ' Project Phase ' , ondelete = ' cascade ' , required = True ) ,
2010-09-30 14:41:19 +00:00
' project_id ' : fields . related ( ' phase_id ' , ' project_id ' , type = ' many2one ' , relation = " project.project " , string = ' Project ' , store = True ) ,
' user_id ' : fields . related ( ' resource_id ' , ' user_id ' , type = ' many2one ' , relation = " res.users " , string = ' User ' ) ,
' date_start ' : fields . date ( ' Start Date ' , help = " Starting Date " ) ,
' date_end ' : fields . date ( ' End Date ' , help = " Ending Date " ) ,
2010-10-13 12:25:19 +00:00
' useability ' : fields . float ( ' Availability ' , help = " Availability of this resource for this project phase in percentage (=50 % ) " ) ,
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 " ) ,
2010-07-29 05:31:04 +00:00
' resource_calendar_id ' : fields . many2one ( ' resource.calendar ' , ' Working Time ' , help = " Timetable working hours to adjust the gantt diagram report " , states = { ' close ' : [ ( ' readonly ' , True ) ] } ) ,
2010-01-12 12:45:02 +00:00
}
2010-09-24 09:26:14 +00:00
def generate_members ( self , cr , uid , ids , context = None ) :
"""
Return a list of Resource Class objects for the resources allocated to the phase .
"""
res = { }
resource_pool = self . pool . get ( ' resource.resource ' )
for project in self . browse ( cr , uid , ids , context = context ) :
user_ids = map ( lambda x : x . id , project . members )
calendar_id = project . resource_calendar_id and project . resource_calendar_id . id or False
resource_objs = resource_pool . generate_resources ( cr , uid , user_ids , calendar_id , context = context )
res [ project . id ] = resource_objs
return res
def schedule_phases ( self , cr , uid , ids , context = None ) :
"""
Schedule the phases .
"""
if type ( ids ) in ( long , int , ) :
ids = [ ids ]
phase_pool = self . pool . get ( ' project.phase ' )
for project in self . browse ( cr , uid , ids , context = context ) :
phase_ids = phase_pool . search ( cr , uid , [ ( ' project_id ' , ' = ' , project . id ) ,
( ' state ' , ' in ' , [ ' draft ' , ' open ' , ' pending ' ] ) ,
( ' previous_phase_ids ' , ' = ' , False )
] )
calendar_id = project . resource_calendar_id and project . resource_calendar_id . id or False
2010-12-30 08:51:44 +00:00
start_date = False
phase_pool . generate_schedule ( cr , uid , phase_ids , start_date , calendar_id , context = context )
2010-09-24 09:26:14 +00:00
return True
def schedule_tasks ( self , cr , uid , ids , context = None ) :
"""
Schedule the tasks according to resource available and priority .
"""
if type ( ids ) in ( long , int , ) :
ids = [ ids ]
user_pool = self . pool . get ( ' res.users ' )
task_pool = self . pool . get ( ' project.task ' )
resource_pool = self . pool . get ( ' resource.resource ' )
resources_list = self . generate_members ( cr , uid , ids , context = context )
return_msg = { }
for project in self . browse ( cr , uid , ids , context = context ) :
start_date = project . date_start
if not start_date :
2010-10-12 13:21:51 +00:00
start_date = datetime . now ( ) . strftime ( " % Y- % m- %d " )
2010-09-24 09:26:14 +00:00
resources = resources_list . get ( project . id , [ ] )
calendar_id = project . resource_calendar_id . id
task_ids = task_pool . search ( cr , uid , [ ( ' project_id ' , ' = ' , project . id ) ,
( ' state ' , ' in ' , [ ' draft ' , ' open ' , ' pending ' ] )
] )
2010-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +00:00
if task_ids :
task_pool . generate_schedule ( cr , uid , task_ids , resources , calendar_id , start_date , context = context )
else :
warning_msg = _ ( " No tasks to compute for Project ' %s ' . " ) % ( project . name )
if " warning " not in return_msg :
return_msg [ " warning " ] = warning_msg
else :
return_msg [ " warning " ] = return_msg [ " warning " ] + " \n " + warning_msg
2010-10-12 13:21:51 +00:00
return return_msg
2010-01-12 12:45:02 +00:00
project ( )
2010-09-24 09:26:14 +00:00
class resource_resource ( osv . osv ) :
_inherit = " resource.resource "
def search ( self , cr , uid , args , offset = 0 , limit = None , order = None , context = None , count = False ) :
if context is None :
context = { }
if context . get ( ' project_id ' , False ) :
project_pool = self . pool . get ( ' project.project ' )
2010-11-22 10:37:53 +00:00
project_rec = project_pool . browse ( cr , uid , context [ ' project_id ' ] , context = context )
2010-09-24 09:26:14 +00:00
user_ids = [ user_id . id for user_id in project_rec . members ]
args . append ( ( ' user_id ' , ' in ' , user_ids ) )
return super ( resource_resource , self ) . search ( cr , uid , args , offset , limit , order , context , count )
resource_resource ( )
class project_task ( osv . osv ) :
2010-01-12 12:45:02 +00:00
_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-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +00:00
def generate_schedule ( self , cr , uid , ids , resources , calendar_id , start_date , context = None ) :
"""
Schedule the tasks according to resource available and priority .
"""
if not ids :
return False
if context is None :
context = { }
user_pool = self . pool . get ( ' res.users ' )
project_pool = self . pool . get ( ' project.project ' )
priority_dict = { ' 0 ' : 1000 , ' 1 ' : 800 , ' 2 ' : 500 , ' 3 ' : 300 , ' 4 ' : 100 }
# Create dynamic no of tasks with the resource specified
def create_tasks ( task_number , eff , priorty = 500 , obj = False ) :
def task ( ) :
"""
task is a dynamic method !
"""
effort = eff
if obj :
resource = obj
priority = priorty
task . __doc__ = " TaskNO %d " % task_number
task . __name__ = " task %d " % task_number
return task
# Create a 'Faces' project with all the tasks and resources
def Project ( ) :
title = " Project "
start = datetime . strftime ( datetime . strptime ( start_date , " % Y- % m- %d " ) , " % Y- % m- %d % H: % M " )
2010-09-24 09:47:12 +00:00
try :
resource = reduce ( operator . or_ , resources )
except :
raise osv . except_osv ( _ ( ' Error ' ) , _ ( ' Should have Resources Allocation or Project Members! ' ) )
2010-09-24 09:26:14 +00:00
minimum_time_unit = 1
if calendar_id : # If project has working calendar
working_days = resource_pool . compute_working_calendar ( cr , uid , calendar_id , context = context )
vacation = tuple ( resource_pool . compute_vacation ( cr , uid , calendar_id , context = context ) )
# Dynamic creation of tasks
task_number = 0
for openobect_task in self . browse ( cr , uid , ids , context = context ) :
hours = str ( openobect_task . planned_hours ) + ' H '
if openobect_task . priority in priority_dict . keys ( ) :
priorty = priority_dict [ openobect_task . priority ]
real_resource = False
if openobect_task . user_id :
for task_resource in resources :
if task_resource . __name__ == task_resource :
real_resource = task_resource
break
2010-10-12 13:21:51 +00:00
2010-09-24 09:26:14 +00:00
task = create_tasks ( task_number , hours , priorty , real_resource )
task_number + = 1
2010-10-12 13:21:51 +00:00
2010-03-02 13:27:02 +00:00
2010-09-24 09:26:14 +00:00
face_projects = Task . BalancedProject ( Project )
loop_no = 0
# Write back the computed dates
for face_project in face_projects :
s_date = face_project . start . to_datetime ( )
e_date = face_project . end . to_datetime ( )
if loop_no > 0 :
ctx = context . copy ( )
ctx . update ( { ' scheduler ' : True } )
user_id = user_pool . search ( cr , uid , [ ( ' name ' , ' = ' , face_project . booked_resource [ 0 ] . __name__ ) ] )
self . write ( cr , uid , [ ids [ loop_no - 1 ] ] , {
' date_start ' : s_date . strftime ( ' % Y- % m- %d % H: % M: % S ' ) ,
' date_end ' : e_date . strftime ( ' % Y- % m- %d % H: % M: % S ' ) ,
' user_id ' : user_id [ 0 ]
} , context = ctx )
loop_no + = 1
return True
project_task ( )
2010-05-19 18:32:32 +00:00
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: