main.py

# Copyright 2021- Le Filament (https://le-filament.com)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html)
from odoo import http
from odoo.http import request

from odoo.addons.oacc_portal.controllers.main import CustomerPortal as CustomerPortal


class CustomerPortal(CustomerPortal):
    def _get_data_date(self, is_data_cdc, operation=None, partner_id=None):
        vals = {}
        # Si des données existent, récupérer les dates min/max/début/fin
        # de l'opération
        if is_data_cdc:
            vals["data_values"] = operation.get_values_init_graph(partner_id)

        else:
            vals["data_values"] = {
                "date_day_start": False,
                "date_day_end": False,
                "date_week_start": False,
                "date_week_end": False,
                "date_month_start": False,
                "date_month_end": False,
                "date_year_start": False,
                "date_year_end": False,
            }

        vals["data_values"].update(operation.get_date_min_max(partner_id))

        return vals

    @http.route(
        ['/operation/<model("acc.operation"):operation>/pmo'],
        type="http",
        auth="user",
        website=True,
    )
    def pmo(self, operation, **kw):
        render_values = {}
        render_values.update(self._get_role(operation))
        render_values.update(
            self._get_data_date(render_values.get("isDataCdc"), operation)
        )

        render_values.update(
            {
                "page_name": "pmo-view",
                "page_breadcrumbs": "Vue globale",
                "scale": "month",
            }
        )

        return request.render("oacc_portal_overview_cdc.operation_pmo", render_values)

    @http.route(
        ['/operation/<model("acc.operation"):operation>/consumer'],
        type="http",
        auth="user",
        website=True,
    )
    def consumer(self, operation, **kw):
        render_values = {}
        render_values.update(self._get_role(operation))

        if (
            render_values.get("isAdmin")
            or render_values.get("isSuperAdmin")
            or render_values.get("isPmo")
        ):
            render_values.update(
                self._get_data_date(render_values.get("isDataCdc"), operation)
            )
            partner_ids = operation.acc_delivery_period_ids.mapped("partner_id").sorted(
                key=lambda p: p.name
            )
        elif render_values.get("isConsumer"):
            render_values.update(
                self._get_data_date(
                    render_values.get("isDataCdc"),
                    operation,
                    request.env.user.commercial_partner_id.id,
                )
            )
            partner_ids = request.env.user.commercial_partner_id

        render_values.update(
            {
                "partners": partner_ids,
                "page_name": "conso",
                "page_breadcrumbs": "Suivi consommation",
                "scale": "month",
            }
        )

        return request.render(
            "oacc_portal_overview_cdc.operation_consumer", render_values
        )

    @http.route(
        [
            '/operation/<model("acc.operation"):operation>/productor',
        ],
        type="http",
        auth="user",
        website=True,
    )
    def productor(self, operation, **kw):
        render_values = {}
        render_values.update(self._get_role(operation))

        if (
            render_values.get("isAdmin")
            or render_values.get("isSuperAdmin")
            or render_values.get("isPmo")
        ):
            render_values.update(
                self._get_data_date(render_values.get("isDataCdc"), operation)
            )
            partner_ids = operation.acc_injection_period_ids.mapped(
                "partner_id"
            ).sorted(key=lambda p: p.name)
        elif render_values.get("isProd"):
            render_values.update(
                self._get_data_date(
                    render_values.get("isDataCdc"),
                    operation,
                    request.env.user.commercial_partner_id.id,
                )
            )
            partner_ids = request.env.user.commercial_partner_id

        render_values.update(
            {
                "partners": partner_ids,
                "page_name": "prod",
                "page_breadcrumbs": "Suivi production",
                "scale": "month",
            }
        )

        return request.render(
            "oacc_portal_overview_cdc.operation_productor", render_values
        )

    @http.route(
        ["/chart/update_json"],
        type="json",
        auth="public",
        methods=["POST"],
        website=True,
        csrf=False,
    )
    def chart_update_json(
        self,
        operation_id,
        partner_id=None,
        prm_id=None,
        start_date=None,
        end_date=None,
        data_type=None,
        **kw,
    ):
        """
        This route is called :
            - When click on button period
            - When click on a consumer, a productor or a PRM
            - When click on date range
        """
        # TODO: add check that prm_id and partner_id are allowed for this user
        operation = request.env["acc.operation"].browse(operation_id)
        vals = operation.graph_view_global(
            start_date, end_date, partner_id, prm_id, data_type
        )

        return vals