You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
244 lines
8.6 KiB
244 lines
8.6 KiB
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# Odoo, Open Source Management Solution
|
|
#
|
|
# Copyright (c) All rights reserved:
|
|
# (c) 2012 Agile Business Group sagl (<http://www.agilebg.com>)
|
|
# (c) 2012 Domsense srl (<http://www.domsense.com>)
|
|
# (c) 2015 Anubía, soluciones en la nube,SL (http://www.anubia.es)
|
|
# Alejandro Santana <alejandrosantana@anubia.es>
|
|
#
|
|
# 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 openerp import models, fields, api
|
|
from openerp.tools.translate import _
|
|
from mako.template import Template
|
|
from datetime import datetime
|
|
from openerp import tools
|
|
from openerp.tools.safe_eval import safe_eval
|
|
import logging
|
|
|
|
|
|
def _models_get(self):
|
|
model_obj = self.env['ir.model']
|
|
model_ids = model_obj.search([])
|
|
return [(model.model, model.name) for model in model_ids]
|
|
|
|
|
|
class SuperCalendarConfigurator(models.Model):
|
|
_logger = logging.getLogger(__name__)
|
|
_name = 'super.calendar.configurator'
|
|
|
|
name = fields.Char(
|
|
string='Name',
|
|
size=64,
|
|
required=True,
|
|
)
|
|
line_ids = fields.One2many(
|
|
comodel_name='super.calendar.configurator.line',
|
|
inverse_name='configurator_id',
|
|
string='Lines',
|
|
)
|
|
|
|
def _clear_super_calendar_records(self):
|
|
""" Remove old super_calendar records """
|
|
super_calendar_pool = self.env['super.calendar']
|
|
super_calendar_ids = super_calendar_pool.search([])
|
|
super_calendar_ids.unlink()
|
|
|
|
@api.multi
|
|
def generate_calendar_records(self):
|
|
""" At every CRON execution, every 'super calendar' data is deleted
|
|
and regenerated again. """
|
|
|
|
# Remove old records
|
|
self._clear_super_calendar_records()
|
|
|
|
# Rebuild all calendar records
|
|
configurator_ids = self.search([])
|
|
for configurator in configurator_ids:
|
|
for line in configurator.line_ids:
|
|
self._generate_record_from_line(configurator, line)
|
|
self._logger.info('Calendar generated')
|
|
return True
|
|
|
|
@api.multi
|
|
def _generate_record_from_line(self, configurator, line):
|
|
super_calendar_pool = self.env['super.calendar']
|
|
values = self._get_record_values_from_line(configurator, line)
|
|
for record in values:
|
|
super_calendar_pool.create(values[record])
|
|
|
|
@api.multi
|
|
def _get_record_values_from_line(self, configurator, line):
|
|
res = {}
|
|
current_pool = self.env[line.name.model]
|
|
domain = line.domain and safe_eval(line.domain) or []
|
|
current_record_ids = current_pool.search(domain)
|
|
for cur_rec in current_record_ids:
|
|
f_user = line.user_field_id and line.user_field_id.name
|
|
f_descr = (line.description_field_id and
|
|
line.description_field_id.name)
|
|
f_date_start = (line.date_start_field_id and
|
|
line.date_start_field_id.name)
|
|
f_date_stop = (line.date_stop_field_id and
|
|
line.date_stop_field_id.name)
|
|
f_duration = (line.duration_field_id and
|
|
line.duration_field_id.name)
|
|
if (f_user and
|
|
cur_rec[f_user] and
|
|
cur_rec[f_user]._model._name != 'res.users'):
|
|
raise Exception(
|
|
_('Error'),
|
|
_("The 'User' field of record %s (%s) "
|
|
"does not refer to res.users")
|
|
% (cur_rec[f_descr], line.name.model))
|
|
|
|
if (((f_descr and cur_rec[f_descr]) or
|
|
line.description_code) and
|
|
cur_rec[f_date_start]):
|
|
duration = False
|
|
if (not line.duration_field_id and
|
|
line.date_stop_field_id and
|
|
cur_rec[f_date_start] and
|
|
cur_rec[f_date_stop]):
|
|
date_start = datetime.strptime(
|
|
cur_rec[f_date_start],
|
|
tools.DEFAULT_SERVER_DATETIME_FORMAT
|
|
)
|
|
date_stop = datetime.strptime(
|
|
cur_rec[f_date_stop],
|
|
tools.DEFAULT_SERVER_DATETIME_FORMAT
|
|
)
|
|
date_diff = (date_stop - date_start)
|
|
duration = date_diff.total_seconds() / 3600
|
|
elif line.duration_field_id:
|
|
duration = cur_rec[f_duration]
|
|
if line.description_type != 'code':
|
|
name = cur_rec[f_descr]
|
|
else:
|
|
parse_dict = {'o': cur_rec}
|
|
mytemplate = Template(line.description_code)
|
|
name = mytemplate.render(**parse_dict)
|
|
|
|
super_calendar_values = {
|
|
'name': name,
|
|
'date_start': cur_rec[f_date_start],
|
|
'duration': duration,
|
|
'user_id': (
|
|
f_user and
|
|
cur_rec[f_user] and
|
|
cur_rec[f_user].id or
|
|
False
|
|
),
|
|
'configurator_id': configurator.id,
|
|
'res_id': line.name.model+','+str(cur_rec['id']),
|
|
'model_id': line.name.id,
|
|
}
|
|
res[cur_rec] = super_calendar_values
|
|
return res
|
|
|
|
|
|
class SuperCalendarConfiguratorLine(models.Model):
|
|
_name = 'super.calendar.configurator.line'
|
|
|
|
name = fields.Many2one(
|
|
comodel_name='ir.model',
|
|
string='Model',
|
|
required=True,
|
|
)
|
|
domain = fields.Char(
|
|
string='Domain',
|
|
size=512,
|
|
)
|
|
configurator_id = fields.Many2one(
|
|
comodel_name='super.calendar.configurator',
|
|
string='Configurator',
|
|
)
|
|
description_type = fields.Selection(
|
|
[('field', 'Field'),
|
|
('code', 'Code')],
|
|
string="Description Type",
|
|
default='field',
|
|
)
|
|
description_field_id = fields.Many2one(
|
|
comodel_name='ir.model.fields',
|
|
string='Description field',
|
|
domain=("['&','|',('ttype', '=', 'char'),('ttype', '=', 'text'),"
|
|
"('model_id', '=', name)]"),
|
|
)
|
|
description_code = fields.Text(
|
|
string='Description field',
|
|
help=("Use '${o}' to refer to the involved object. "
|
|
"E.g.: '${o.project_id.name}'"),
|
|
)
|
|
date_start_field_id = fields.Many2one(
|
|
comodel_name='ir.model.fields',
|
|
string='Start date field',
|
|
domain=("['&','|',('ttype', '=', 'datetime'),('ttype', '=', 'date'),"
|
|
"('model_id', '=', name)]"),
|
|
required=True,
|
|
)
|
|
date_stop_field_id = fields.Many2one(
|
|
comodel_name='ir.model.fields',
|
|
string='End date field',
|
|
domain="['&',('ttype', '=', 'datetime'),('model_id', '=', name)]"
|
|
)
|
|
duration_field_id = fields.Many2one(
|
|
comodel_name='ir.model.fields',
|
|
string='Duration field',
|
|
domain="['&',('ttype', '=', 'float'), ('model_id', '=', name)]",
|
|
)
|
|
user_field_id = fields.Many2one(
|
|
comodel_name='ir.model.fields',
|
|
string='User field',
|
|
domain="['&', ('ttype', '=', 'many2one'), ('model_id', '=', name)]",
|
|
)
|
|
|
|
|
|
class SuperCalendar(models.Model):
|
|
_name = 'super.calendar'
|
|
|
|
name = fields.Char(
|
|
string='Description',
|
|
size=512,
|
|
required=True,
|
|
)
|
|
date_start = fields.Datetime(
|
|
string='Start date',
|
|
required=True,
|
|
)
|
|
duration = fields.Float(
|
|
string='Duration'
|
|
)
|
|
user_id = fields.Many2one(
|
|
comodel_name='res.users',
|
|
string='User',
|
|
)
|
|
configurator_id = fields.Many2one(
|
|
comodel_name='super.calendar.configurator',
|
|
string='Configurator',
|
|
)
|
|
res_id = fields.Reference(
|
|
selection=_models_get,
|
|
string='Resource',
|
|
size=128,
|
|
)
|
|
model_id = fields.Many2one(
|
|
comodel_name='ir.model',
|
|
string='Model',
|
|
)
|