notebook

ow_vm_management/controllers/__init__.py

@@ -0,0 +1 @@
+from . import portal

ow_vm_management/controllers/portal.py

@@ -0,0 +1,80 @@
+from odoo import http, _
+from odoo.exceptions import AccessError, MissingError
+from odoo.http import request
+from odoo.addons.portal.controllers.portal import CustomerPortal, pager as portal_pager
+from odoo.osv.expression import OR
+
+class CustomerPortalVPS(CustomerPortal):
+
+    def _prepare_home_portal_values(self, counters):
+        values = super()._prepare_home_portal_values(counters)
+        if 'vps_server_count' in counters:
+            values['vps_server_count'] = request.env['vps.server'].search_count([('customer_id', '=', request.env.user.partner_id.id)])
+        return values
+
+    @http.route(['/my/vps-servers', '/my/vps-servers/page/<int:page>'], type='http', auth="user", website=True)
+    def portal_my_vps_servers(self, page=1, date_begin=None, date_end=None, sortby=None, **kw):
+        values = self._prepare_portal_layout_values()
+        VPSServer = request.env['vps.server']
+
+        domain = [('customer_id', '=', request.env.user.partner_id.id)]
+
+        searchbar_sortings = {
+            'name': {'label': _('Name'), 'order': 'name'},
+            'ip_address': {'label': _('IP Address'), 'order': 'ip_address'},
+        }
+
+        if not sortby:
+            sortby = 'name'
+        sort_order = searchbar_sortings[sortby]['order']
+
+        if date_begin and date_end:
+            domain += [('create_date', '>', date_begin), ('create_date', '<=', date_end)]
+
+        # count for pager
+        vps_server_count = VPSServer.search_count(domain)
+
+        # make pager
+        pager = portal_pager(
+            url="/my/vps-servers",
+            url_args={'date_begin': date_begin, 'date_end': date_end, 'sortby': sortby},
+            total=vps_server_count,
+            page=page,
+            step=self._items_per_page
+        )
+
+        # search the count to display, according to the pager data
+        vps_servers = VPSServer.search(
+            domain,
+            order=sort_order,
+            limit=self._items_per_page,
+            offset=pager['offset']
+        )
+
+        values.update({
+            'date': date_begin,
+            'vps_servers': vps_servers,
+            'page_name': 'vps_server',
+            'pager': pager,
+            'default_url': '/my/vps-servers',
+            'searchbar_sortings': searchbar_sortings,
+            'sortby': sortby,
+        })
+        return request.render("ow_vm_management.portal_my_vps_servers", values)
+
+    @http.route(['/my/vps-servers/<int:vps_server_id>'], type='http', auth="user", website=True)
+    def portal_my_vps_server(self, vps_server_id=None, access_token=None, **kw):
+        try:
+            vps_server_sudo = self._document_check_access('vps.server', vps_server_id, access_token)
+        except (AccessError, MissingError):
+            return request.redirect('/my')
+
+        values = self._vps_server_get_page_view_values(vps_server_sudo, access_token, **kw)
+        return request.render("ow_vm_management.portal_vps_server_page", values)
+
+    def _vps_server_get_page_view_values(self, vps_server, access_token, **kwargs):
+        values = {
+            'page_name': 'vps_server',
+            'vps_server': vps_server,
+        }
+        return self._get_page_view_values(vps_server, access_token, values, 'my_vps_servers_history', False, **kwargs)