purchase_custom_visibility/models/product.py

126 lines
4.2 KiB
Python

# -*- coding: utf-8 -*-
from odoo import models, api
from lxml import etree
class ProductSupplierinfo(models.Model):
_inherit = 'product.supplierinfo'
@api.model
def get_views(self, views, options=None):
res = super().get_views(views, options)
if not self.env.user.hide_purchase_prices:
return res
for view_type, view_data in res.get('views', {}).items():
arch_str = view_data.get('arch')
if not arch_str:
continue
try:
doc = etree.fromstring(arch_str.encode('utf-8'))
except Exception:
continue
modified = False
price_fields = {'price', 'discount'}
for field in doc.xpath("//field"):
if field.get('name') in price_fields:
is_column = False
p = field.getparent()
while p is not None:
if p.tag in ('tree', 'list'):
is_column = True
break
p = p.getparent()
if is_column:
field.set('column_invisible', 'True')
else:
field.set('invisible', '1')
modified = True
if modified:
view_data['arch'] = etree.tostring(doc, encoding='utf-8').decode('utf-8')
return res
class ProductTemplate(models.Model):
_inherit = 'product.template'
@api.model
def get_views(self, views, options=None):
res = super().get_views(views, options)
if not self.env.user.hide_purchase_prices:
return res
for view_type, view_data in res.get('views', {}).items():
arch_str = view_data.get('arch')
if not arch_str:
continue
try:
doc = etree.fromstring(arch_str.encode('utf-8'))
except Exception:
continue
modified = False
for field in doc.xpath("//field[@name='standard_price']"):
is_column = False
p = field.getparent()
while p is not None:
if p.tag in ('tree', 'list'):
is_column = True
break
p = p.getparent()
if is_column:
field.set('column_invisible', 'True')
else:
field.set('invisible', '1')
modified = True
if modified:
view_data['arch'] = etree.tostring(doc, encoding='utf-8').decode('utf-8')
return res
class ProductProduct(models.Model):
_inherit = 'product.product'
@api.model
def get_views(self, views, options=None):
res = super().get_views(views, options)
if not self.env.user.hide_purchase_prices:
return res
for view_type, view_data in res.get('views', {}).items():
arch_str = view_data.get('arch')
if not arch_str:
continue
try:
doc = etree.fromstring(arch_str.encode('utf-8'))
except Exception:
continue
modified = False
for field in doc.xpath("//field[@name='standard_price']"):
is_column = False
p = field.getparent()
while p is not None:
if p.tag in ('tree', 'list'):
is_column = True
break
p = p.getparent()
if is_column:
field.set('column_invisible', 'True')
else:
field.set('invisible', '1')
modified = True
if modified:
view_data['arch'] = etree.tostring(doc, encoding='utf-8').decode('utf-8')
return res