339 lines
16 KiB
Python
339 lines
16 KiB
Python
# -*- 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 datetime import datetime
|
|
from openerp.tools.translate import _
|
|
from openerp.osv import fields, osv
|
|
from openerp.addons.resource.faces import task as Task
|
|
|
|
class project_phase(osv.osv):
|
|
_name = "project.phase"
|
|
_description = "Project Phase"
|
|
|
|
def _check_recursion(self, cr, uid, ids, context=None):
|
|
if context is None:
|
|
context = {}
|
|
|
|
data_phase = self.browse(cr, uid, ids[0], context=context)
|
|
prev_ids = data_phase.previous_phase_ids
|
|
next_ids = data_phase.next_phase_ids
|
|
# it should neither be in prev_ids nor in next_ids
|
|
if (data_phase in prev_ids) or (data_phase in next_ids):
|
|
return False
|
|
ids = [id for id in prev_ids if id in next_ids]
|
|
# both prev_ids and next_ids must be unique
|
|
if ids:
|
|
return False
|
|
# unrelated project
|
|
prev_ids = [rec.id for rec in prev_ids]
|
|
next_ids = [rec.id for rec in next_ids]
|
|
# iter prev_ids
|
|
while prev_ids:
|
|
cr.execute('SELECT distinct prv_phase_id FROM project_phase_rel WHERE next_phase_id IN %s', (tuple(prev_ids),))
|
|
prv_phase_ids = filter(None, map(lambda x: x[0], cr.fetchall()))
|
|
if data_phase.id in prv_phase_ids:
|
|
return False
|
|
ids = [id for id in prv_phase_ids if id in next_ids]
|
|
if ids:
|
|
return False
|
|
prev_ids = prv_phase_ids
|
|
# iter next_ids
|
|
while next_ids:
|
|
cr.execute('SELECT distinct next_phase_id FROM project_phase_rel WHERE prv_phase_id IN %s', (tuple(next_ids),))
|
|
next_phase_ids = filter(None, map(lambda x: x[0], cr.fetchall()))
|
|
if data_phase.id in next_phase_ids:
|
|
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
|
|
|
|
def _check_dates(self, cr, uid, ids, context=None):
|
|
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
|
|
return True
|
|
|
|
def _compute_progress(self, cr, uid, ids, field_name, arg, context=None):
|
|
res = {}
|
|
if not ids:
|
|
return res
|
|
for phase in self.browse(cr, uid, ids, context=context):
|
|
if phase.state=='done':
|
|
res[phase.id] = 100.0
|
|
continue
|
|
elif phase.state=="cancelled":
|
|
res[phase.id] = 0.0
|
|
continue
|
|
elif not phase.task_ids:
|
|
res[phase.id] = 0.0
|
|
continue
|
|
|
|
tot = done = 0.0
|
|
for task in phase.task_ids:
|
|
tot += task.total_hours
|
|
done += min(task.effective_hours, task.total_hours)
|
|
|
|
if not tot:
|
|
res[phase.id] = 0.0
|
|
else:
|
|
res[phase.id] = round(100.0 * done / tot, 2)
|
|
return res
|
|
|
|
_columns = {
|
|
'name': fields.char("Name", size=64, required=True),
|
|
'date_start': fields.datetime('Start Date', select=True, help="It's computed by the scheduler according the project date or the end date of the previous phase.", states={'done':[('readonly',True)], 'cancelled':[('readonly',True)]}),
|
|
'date_end': fields.datetime('End Date', help=" It's computed by the scheduler according to the start date and the duration.", states={'done':[('readonly',True)], 'cancelled':[('readonly',True)]}),
|
|
'constraint_date_start': fields.datetime('Minimum Start Date', help='force the phase to start after this date', states={'done':[('readonly',True)], 'cancelled':[('readonly',True)]}),
|
|
'constraint_date_end': fields.datetime('Deadline', help='force the phase to finish before this date', states={'done':[('readonly',True)], 'cancelled':[('readonly',True)]}),
|
|
'project_id': fields.many2one('project.project', 'Project', required=True, select=True),
|
|
'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)]}),
|
|
'sequence': fields.integer('Sequence', select=True, help="Gives the sequence order when displaying a list of phases."),
|
|
'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 Unit of Measure', required=True, help="Unit of Measure (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)]}),
|
|
'user_force_ids': fields.many2many('res.users', string='Force Assigned Users'),
|
|
'user_ids': fields.one2many('project.user.allocation', 'phase_id', "Assigned Users",states={'done':[('readonly',True)], 'cancelled':[('readonly',True)]},
|
|
help="The resources on the project can be computed automatically by the scheduler."),
|
|
'state': fields.selection([('draft', 'New'), ('cancelled', 'Cancelled'),('open', 'In Progress'), ('pending', 'Pending'), ('done', 'Done')], 'Status', readonly=True, required=True,
|
|
help='If the phase is created the status \'Draft\'.\n If the phase is started, the status becomes \'In Progress\'.\n If review is needed the phase is in \'Pending\' status.\
|
|
\n If the phase is over, the status is set to \'Done\'.'),
|
|
'progress': fields.function(_compute_progress, string='Progress', help="Computed based on related tasks"),
|
|
}
|
|
_defaults = {
|
|
'state': 'draft',
|
|
'sequence': 10,
|
|
}
|
|
_order = "project_id, date_start, sequence"
|
|
_constraints = [
|
|
(_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']),
|
|
]
|
|
|
|
def onchange_project(self, cr, uid, ids, project, context=None):
|
|
return {}
|
|
|
|
def copy(self, cr, uid, id, default=None, context=None):
|
|
if default is None:
|
|
default = {}
|
|
if not default.get('name', False):
|
|
default.update(name=_('%s (copy)') % (self.browse(cr, uid, id, context=context).name))
|
|
return super(project_phase, self).copy(cr, uid, id, default, context)
|
|
|
|
def set_draft(self, cr, uid, ids, *args):
|
|
self.write(cr, uid, ids, {'state': 'draft'})
|
|
return True
|
|
|
|
def set_open(self, cr, uid, ids, *args):
|
|
self.write(cr, uid, ids, {'state': 'open'})
|
|
return True
|
|
|
|
def set_pending(self, cr, uid, ids, *args):
|
|
self.write(cr, uid, ids, {'state': 'pending'})
|
|
return True
|
|
|
|
def set_cancel(self, cr, uid, ids, *args):
|
|
self.write(cr, uid, ids, {'state': 'cancelled'})
|
|
return True
|
|
|
|
def set_done(self, cr, uid, ids, *args):
|
|
self.write(cr, uid, ids, {'state': 'done'})
|
|
return True
|
|
|
|
def generate_phase(self, cr, uid, phases, context=None):
|
|
context = context or {}
|
|
result = ""
|
|
|
|
task_pool = self.pool.get('project.task')
|
|
for phase in phases:
|
|
if phase.state in ('done','cancelled'):
|
|
continue
|
|
# FIXME: brittle and not working if context['lang'] != 'en_US'
|
|
duration_uom = {
|
|
'day(s)': 'd', 'days': 'd', 'day': 'd', 'd':'d',
|
|
'month(s)': 'm', 'months': 'm', 'month':'month', 'm':'m',
|
|
'week(s)': 'w', 'weeks': 'w', 'week': 'w', 'w':'w',
|
|
'hour(s)': 'H', 'hours': 'H', 'hour': 'H', 'h':'H',
|
|
}.get(phase.product_uom.name.lower(), "H")
|
|
duration = str(phase.duration) + duration_uom
|
|
result += '''
|
|
def Phase_%s():
|
|
effort = \"%s\"''' % (phase.id, duration)
|
|
start = []
|
|
if phase.constraint_date_start:
|
|
start.append('datetime.datetime.strptime("'+str(phase.constraint_date_start)+'", "%Y-%m-%d %H:%M:%S")')
|
|
for previous_phase in phase.previous_phase_ids:
|
|
start.append("up.Phase_%s.end" % (previous_phase.id,))
|
|
if start:
|
|
result += '''
|
|
start = max(%s)
|
|
''' % (','.join(start))
|
|
|
|
if phase.user_force_ids:
|
|
result += '''
|
|
resource = %s
|
|
''' % '|'.join(map(lambda x: 'User_'+str(x.id), phase.user_force_ids))
|
|
|
|
result += task_pool._generate_task(cr, uid, phase.task_ids, ident=8, context=context)
|
|
result += "\n"
|
|
|
|
return result
|
|
project_phase()
|
|
|
|
class project_user_allocation(osv.osv):
|
|
_name = 'project.user.allocation'
|
|
_description = 'Phase User Allocation'
|
|
_rec_name = 'user_id'
|
|
_columns = {
|
|
'user_id': fields.many2one('res.users', 'User', required=True),
|
|
'phase_id': fields.many2one('project.phase', 'Project Phase', ondelete='cascade', required=True),
|
|
'project_id': fields.related('phase_id', 'project_id', type='many2one', relation="project.project", string='Project', store=True),
|
|
'date_start': fields.datetime('Start Date', help="Starting Date"),
|
|
'date_end': fields.datetime('End Date', help="Ending Date"),
|
|
}
|
|
project_user_allocation()
|
|
|
|
class project(osv.osv):
|
|
_inherit = "project.project"
|
|
|
|
def _phase_count(self, cr, uid, ids, field_name, arg, context=None):
|
|
res = dict.fromkeys(ids, 0)
|
|
phase_ids = self.pool.get('project.phase').search(cr, uid, [('project_id', 'in', ids)])
|
|
for phase in self.pool.get('project.phase').browse(cr, uid, phase_ids, context):
|
|
res[phase.project_id.id] += 1
|
|
return res
|
|
|
|
_columns = {
|
|
'phase_ids': fields.one2many('project.phase', 'project_id', "Project Phases"),
|
|
'phase_count': fields.function(_phase_count, type='integer', string="Open Phases"),
|
|
}
|
|
|
|
def schedule_phases(self, cr, uid, ids, context=None):
|
|
context = context or {}
|
|
if type(ids) in (long, int,):
|
|
ids = [ids]
|
|
projects = self.browse(cr, uid, ids, context=context)
|
|
result = self._schedule_header(cr, uid, ids, context=context)
|
|
for project in projects:
|
|
result += self._schedule_project(cr, uid, project, context=context)
|
|
result += self.pool.get('project.phase').generate_phase(cr, uid, project.phase_ids, context=context)
|
|
|
|
local_dict = {}
|
|
exec result in local_dict
|
|
projects_gantt = Task.BalancedProject(local_dict['Project'])
|
|
|
|
for project in projects:
|
|
project_gantt = getattr(projects_gantt, 'Project_%d' % (project.id,))
|
|
for phase in project.phase_ids:
|
|
if phase.state in ('done','cancelled'):
|
|
continue
|
|
# Maybe it's better to update than unlink/create if it already exists ?
|
|
p = getattr(project_gantt, 'Phase_%d' % (phase.id,))
|
|
|
|
self.pool.get('project.user.allocation').unlink(cr, uid,
|
|
[x.id for x in phase.user_ids],
|
|
context=context
|
|
)
|
|
|
|
for r in p.booked_resource:
|
|
self.pool.get('project.user.allocation').create(cr, uid, {
|
|
'user_id': int(r.name[5:]),
|
|
'phase_id': phase.id,
|
|
'date_start': p.start.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'date_end': p.end.strftime('%Y-%m-%d %H:%M:%S')
|
|
}, context=context)
|
|
self.pool.get('project.phase').write(cr, uid, [phase.id], {
|
|
'date_start': p.start.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'date_end': p.end.strftime('%Y-%m-%d %H:%M:%S')
|
|
}, context=context)
|
|
return True
|
|
|
|
def map_phase(self, cr, uid, old_project_id, new_project_id, context=None):
|
|
""" copy and map phases from old to new project while keeping order relations """
|
|
project_phase = self.pool['project.phase']
|
|
project_task = self.pool['project.task']
|
|
# mapping source and copy ids to recreate m2m links
|
|
phase_ids_mapping = {}
|
|
project = self.browse(cr, uid, old_project_id, context=context)
|
|
if project.phase_ids:
|
|
for phase in project.phase_ids:
|
|
phase_default = {
|
|
'project_id': new_project_id,
|
|
'previous_phase_ids': [],
|
|
'next_phase_ids': [],
|
|
'task_ids': [],
|
|
}
|
|
# adding relationships with already copied phases
|
|
for previous_phase in phase.previous_phase_ids:
|
|
if previous_phase.id in phase_ids_mapping:
|
|
phase_default['previous_phase_ids'].append((4, phase_ids_mapping[previous_phase.id]))
|
|
for next_phase in phase.next_phase_ids:
|
|
if next_phase.id in phase_ids_mapping:
|
|
phase_default['previous_phase_ids'].append((4, phase_ids_mapping[next_phase.id]))
|
|
phase_ids_mapping[phase.id] = project_phase.copy(cr, uid, phase.id, phase_default, context=context)
|
|
if project.tasks:
|
|
# if has copied tasks, need to update phase_id
|
|
for task in self.browse(cr, uid, new_project_id, context=context).tasks:
|
|
if task.phase_id and task.phase_id.id in phase_ids_mapping:
|
|
project_task.write(cr, uid, task.id, {'phase_id': phase_ids_mapping[phase.id]}, context=context)
|
|
return True
|
|
|
|
def copy(self, cr, uid, id, default=None, context=None):
|
|
if default is None:
|
|
default = {}
|
|
default.update(phase_ids=[])
|
|
new_project_id = super(project, self).copy(cr, uid, id, default, context)
|
|
self.map_phase(cr, uid, id, new_project_id, context=context)
|
|
return new_project_id
|
|
|
|
class account_analytic_account(osv.osv):
|
|
_inherit = 'account.analytic.account'
|
|
_description = 'Analytic Account'
|
|
_columns = {
|
|
'use_phases': fields.boolean('Phases', help="Check this field if you plan to use phase-based scheduling"),
|
|
}
|
|
|
|
def on_change_template(self, cr, uid, ids, template_id, context=None):
|
|
res = super(account_analytic_account, self).on_change_template(cr, uid, ids, template_id, context=context)
|
|
if template_id and 'value' in res:
|
|
template = self.browse(cr, uid, template_id, context=context)
|
|
res['value']['use_phases'] = template.use_phases
|
|
return res
|
|
|
|
|
|
def _trigger_project_creation(self, cr, uid, vals, context=None):
|
|
if context is None: context = {}
|
|
res = super(account_analytic_account, self)._trigger_project_creation(cr, uid, vals, context=context)
|
|
return res or (vals.get('use_phases') and not 'project_creation_in_progress' in context)
|
|
|
|
account_analytic_account()
|
|
|
|
class project_task(osv.osv):
|
|
_inherit = "project.task"
|
|
_columns = {
|
|
'phase_id': fields.many2one('project.phase', 'Project Phase', domain="[('project_id', '=', project_id)]"),
|
|
}
|
|
project_task()
|
|
|
|
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|