forked from Yaltik/golem
108 lines
4.2 KiB
Python
108 lines
4.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
|
|
# Copyright 2018 Fabien Bourgeois <fabien@yaltik.com>
|
|
#
|
|
# 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/>.
|
|
""" Golem Reservation Report """
|
|
import time
|
|
|
|
from random import randint
|
|
from odoo import models, api
|
|
|
|
|
|
|
|
class GolemResevationReport(models.AbstractModel):
|
|
"Golem Reservation Report"
|
|
_name = 'report.golem_resource_report.golem_reservation_report'
|
|
|
|
def get_total_reservation(self, data):
|
|
"Get Reservation Count"
|
|
domain = [('date_start', '>', data['date_start']),
|
|
('date_stop', '<', data['date_stop']),
|
|
('resource_id', 'in', data['resource_ids'])]
|
|
return self.env['golem.resource.reservation'].search_count(domain)
|
|
|
|
def get_resource(self, data):
|
|
"Get Resource List"
|
|
lst = []
|
|
domain = [('date_start', '>', data['date_start']),
|
|
('date_stop', '<', data['date_stop']),
|
|
('resource_id', 'in', data['resource_ids'])]
|
|
reservations = self.env['golem.resource.reservation'].search(domain, order='date_start')
|
|
lst = reservations.mapped('resource_id.name')
|
|
return lst
|
|
|
|
def get_client_color(self, partner_number):
|
|
"Get Client Color"
|
|
colors = ['#FFFF5B', '#81EC54', '#47C8C8', '#FB5A66', '#E8E750',
|
|
'#CF4ACF', '#9655D2', '#FFA15B', '#5F68D5', '#60E652']
|
|
color = "#000000"
|
|
if partner_number < 10:
|
|
color = colors[partner_number]
|
|
else:
|
|
red = randint(128, 255)
|
|
green = randint(128, 255)
|
|
blue = randint(128, 255)
|
|
color = "#" +hex(red)[2:]+hex(green)[2:]+hex(blue)[2:]
|
|
return color
|
|
|
|
def get_data(self, data):
|
|
"Get Resevation Data"
|
|
lst = []
|
|
domain = [('date_start', '>', data['date_start']),
|
|
('date_stop', '<', data['date_stop']),
|
|
('resource_id', 'in', data['resource_ids'])]
|
|
reservations = self.env['golem.resource.reservation'].search(domain, order='date_start')
|
|
partner_ids = reservations.mapped('partner_id.id')
|
|
partner_colors = {}
|
|
partner_number = 0
|
|
for partner_id in partner_ids:
|
|
partner_colors[str(partner_id)] = self.get_client_color(partner_number)
|
|
partner_number += 1
|
|
|
|
res = {}
|
|
for reservation in reservations:
|
|
res = {
|
|
'name': reservation.name,
|
|
'resource_name': reservation.resource_id.name,
|
|
'client': reservation.partner_id.name,
|
|
'date_start': reservation.date_start,
|
|
'date_stop': reservation.date_stop,
|
|
'day_start': reservation.day_start,
|
|
'bgcolor': partner_colors[str(reservation.partner_id.id)]
|
|
}
|
|
lst.append(res)
|
|
return lst
|
|
|
|
@api.model
|
|
def render_html(self, docids, data=None):
|
|
"Render HTML"
|
|
model = self.env.context.get('active_model')
|
|
docs = self.env[model].browse(self.env.context.get('active_id'))
|
|
docargs = {
|
|
'doc_ids': self.ids,
|
|
'doc_model': model,
|
|
'docs': docs,
|
|
'time': time,
|
|
'data': data,
|
|
'date_start': data['date_start'],
|
|
'date_stop': data['date_stop'],
|
|
'get_total_reservation': self.get_total_reservation(data),
|
|
'get_data': self.get_data(data),
|
|
'get_resource': self.get_resource(data),
|
|
}
|
|
return self.env['report'] \
|
|
.render('golem_resource_report.golem_reservation_report', docargs)
|