Files
odoo_vm_proxmox/ow_vm_management/controllers/portal.py
2024-08-05 23:45:20 +02:00

105 lines
4.2 KiB
Python

from odoo import http, _
from odoo.exceptions import AccessError
from odoo.http import request
from odoo.addons.portal.controllers.portal import CustomerPortal, pager as portal_pager
from odoo.osv.expression import OR
class VPSPortal(CustomerPortal):
def _prepare_home_portal_values(self, counters):
values = super()._prepare_home_portal_values(counters)
partner = request.env.user.partner_id
VPSServer = request.env['vps.server']
if 'vps_server_count' in counters:
values['vps_server_count'] = VPSServer.search_count([('customer_id', '=', partner.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()
partner = request.env.user.partner_id
VPSServer = request.env['vps.server']
domain = [('customer_id', '=', partner.id)]
searchbar_sortings = {
'name': {'label': _('Name'), 'order': 'name'},
'state': {'label': _('State'), 'order': 'state'},
}
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_id>'], type='http', auth="user", website=True)
def portal_my_vps_server(self, vps_id, **kw):
try:
vps_sudo = self._document_check_access('vps.server', vps_id)
if vps_sudo.customer_id != request.env.user.partner_id:
return request.redirect('/my/vps-servers')
values = self._vps_server_get_page_view_values(vps_sudo, **kw)
return request.render("ow_vm_management.portal_vps_server_page", values)
except AccessError:
return request.redirect('/my/vps-servers')
def _vps_server_get_page_view_values(self, vps_server, access_token=None, **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)
@http.route(['/my/vps/<int:vps_id>/restart'], type='http', auth="user", website=True)
def portal_restart_vps(self, vps_id, **kw):
try:
vps_sudo = self._document_check_access('vps.server', vps_id)
if vps_sudo.customer_id != request.env.user.partner_id:
return request.redirect('/my/vps-servers')
vps_sudo.action_restart_from_portal()
return request.redirect(f'/my/vps-servers/{vps_id}')
except AccessError:
return request.redirect('/my/vps-servers')
def _document_check_access(self, model_name, document_id, access_token=None):
document = request.env[model_name].sudo().browse(document_id)
document_sudo = document.with_user(request.env.user).sudo()
try:
document_sudo.check_access_rights('read')
document_sudo.check_access_rule('read')
except AccessError:
raise
return document_sudo