# -*- 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)