vertical-cooperative/easy_my_coop_website_portal/controllers/main.py

# -*- coding: utf-8 -*-

# Copyright 2015-2016 Odoo S.A.
# Copyright 2016 Jairo Llopis <jairo.llopis@tecnativa.com>
# Copyright 2017-2018 Rémy Taymans <remytaymans@gmail.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).


from werkzeug.exceptions import Forbidden, NotFound

from openerp import http
from openerp.exceptions import AccessError, MissingError
from openerp.fields import Date
from openerp.http import request

from openerp.addons.website_portal_v10.controllers.main import WebsiteAccount


class CooperatorWebsiteAccount(WebsiteAccount):

    def _prepare_portal_layout_values(self):
        values = super(CooperatorWebsiteAccount,
                       self)._prepare_portal_layout_values()
        # We assume that commercial_partner_id always point to the
        # partner itself or to the linked partner. So there is no
        # need to check if the partner is a "contact" or not.
        coop = request.env.user.partner_id.commercial_partner_id
        coop_bank = request.env['res.partner.bank'].sudo().search(
            [('partner_id', 'in', [coop.id])],
            limit=1
        )
        values.update({
            'coop': coop,
            'coop_bank': coop_bank,
        })
        return values

    @http.route()
    def account(self):
        """ Add Release Capital Request to main account page """
        response = super(CooperatorWebsiteAccount, self).account()
        partner = request.env.user.partner_id

        invoice_mgr = request.env['account.invoice']
        capital_request_count = invoice_mgr.sudo().search_count([
            ('partner_id', 'in',
             [partner.commercial_partner_id.id]),
            ('state', 'in', ['open', 'paid', 'cancelled']),
            # Get only the release capital request
            ('release_capital_request', '=', True),
        ])

        response.qcontext.update({
            'capital_request_count': capital_request_count,
        })
        return response

    @http.route(
        ['/my/release_capital_request',
         '/my/release_capital_request/page/<int:page>'],
        type='http', auth="user", website=True)
    def portal_my_release_capital_request(self, page=1, date_begin=None,
                                          date_end=None, **kw):
        """Render a page with the list of release capital request.
        A release capital request is an invoice with a flag that tell
        if it's a capital request or not.
        """
        values = self._prepare_portal_layout_values()
        partner = request.env.user.partner_id
        invoice_mgr = request.env['account.invoice']

        domain = [
            ('partner_id', 'in',
             [partner.commercial_partner_id.id]),
            ('state', 'in', ['open', 'paid', 'cancelled']),
            # Get only the release capital request
            ('release_capital_request', '=', True),
        ]
        archive_groups = self._get_archive_groups_sudo('account.invoice',
                                                       domain)
        if date_begin and date_end:
            domain += [('create_date', '>=', date_begin),
                       ('create_date', '<', date_end)]

        # count for pager
        capital_request_count = invoice_mgr.sudo().search_count(domain)
        # pager
        pager = request.website.pager(
            url="/my/release_capital_request",
            url_args={'date_begin': date_begin, 'date_end': date_end},
            total=capital_request_count,
            page=page,
            step=self._items_per_page
        )
        # content according to pager and archive selected
        invoices = invoice_mgr.sudo().search(
            domain, limit=self._items_per_page, offset=pager['offset'])
        values.update({
            'date': date_begin,
            'capital_requests': invoices,
            'page_name': 'invoice',
            'pager': pager,
            'archive_groups': archive_groups,
            'default_url': '/my/release_capital_request',
        })
        return request.website.render(
            "easy_my_coop_website_portal.portal_my_capital_releases",
            values
        )

    @http.route(['/my/release_capital_request/pdf/<int:oid>'],
                type='http', auth="user", website=True)
    def get_release_capital_request(self, oid=-1, **kw):
        """Render the pdf of the given release capital request"""
        # Get the release capital request and raise an error if the user
        # is not allowed to access to it or if the object is not found.
        partner = request.env.user.partner_id
        invoice_mgr = request.env['account.invoice']
        capital_request = invoice_mgr.sudo().browse(oid)
        try:
            if capital_request.partner_id != partner:
                raise Forbidden()
        except AccessError:
            raise Forbidden()
        except MissingError:
            raise NotFound()
        # Get the pdf
        report_mgr = request.env['report']
        pdf = report_mgr.sudo().get_pdf(
            capital_request,
            'easy_my_coop.theme_invoice_G002'
        )
        filename = "Release Capital Request - %d" % capital_request.id
        return self._render_pdf(pdf, filename)

    @http.route(['/my/cooperator_certificate/pdf'],
                type='http', auth="user", website=True)
    def get_cooperator_certificat(self, **kw):
        """Render the cooperator certificate pdf of the current user"""
        partner = request.env.user.partner_id
        report_mgr = request.env['report']
        pdf = report_mgr.sudo().get_pdf(
            partner,
            'easy_my_coop.cooperator_certificat_G001'
        )
        filename = "Cooperator Certificate - %s" % partner.name
        return self._render_pdf(pdf, filename)

    def _render_pdf(self, pdf, filename):
        """Render a http response for a pdf"""
        pdfhttpheaders = [
            ('Content-Disposition', 'inline; filename="%s.pdf"' % filename),
            ('Content-Type', 'application/pdf'),
            ('Content-Length', len(pdf))
        ]
        return request.make_response(pdf, headers=pdfhttpheaders)

    def _get_archive_groups_sudo(self, model, domain=None, fields=None,
                                 groupby="create_date",
                                 order="create_date desc"):
        """Same as the one from website_portal_v10 except that it runs
        in root.
        """
        if not model:
            return []
        if domain is None:
            domain = []
        if fields is None:
            fields = ['name', 'create_date']
        groups = []
        for group in request.env[model].sudo().read_group(
                domain, fields=fields, groupby=groupby, orderby=order):
            label = group[groupby]
            date_begin = date_end = None
            for leaf in group["__domain"]:
                if leaf[0] == groupby:
                    if leaf[1] == ">=":
                        date_begin = leaf[2]
                    elif leaf[1] == "<":
                        date_end = leaf[2]
            groups.append({
                'date_begin': Date.to_string(Date.from_string(date_begin)),
                'date_end': Date.to_string(Date.from_string(date_end)),
                'name': label,
                'item_count': group[groupby + '_count']
            })
        return groups