summaryrefslogtreecommitdiff
path: root/indoteknik_api/models/res_users.py
diff options
context:
space:
mode:
Diffstat (limited to 'indoteknik_api/models/res_users.py')
-rw-r--r--indoteknik_api/models/res_users.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/indoteknik_api/models/res_users.py b/indoteknik_api/models/res_users.py
index b0807e9f..3993201e 100644
--- a/indoteknik_api/models/res_users.py
+++ b/indoteknik_api/models/res_users.py
@@ -11,6 +11,12 @@ class ResUsers(models.Model):
product_pricelist_tier3 = int(config.get_param('product.pricelist.tier3'))
user_pricelist = res_user.property_product_pricelist
+ user_pricelist_id = user_pricelist.id or False
+ pricelist_tier = False
+ if user_pricelist_id == product_pricelist_tier1: pricelist_tier = 'tier1'
+ if user_pricelist_id == product_pricelist_tier2: pricelist_tier = 'tier2'
+ if user_pricelist_id == product_pricelist_tier3: pricelist_tier = 'tier3'
+
data = {
'id': res_user.id,
'parent_id': res_user.parent_id.id or False,
@@ -21,15 +27,9 @@ class ResUsers(models.Model):
'mobile': res_user.mobile or '',
'external': res_user.share,
'company': res_user.company_type == 'company',
- 'pricelist': False
+ 'pricelist': pricelist_tier
}
- if user_pricelist.id == product_pricelist_tier1:
- data['pricelist'] = 'tier1'
- if user_pricelist.id == product_pricelist_tier2:
- data['pricelist'] = 'tier2'
- if user_pricelist.id == product_pricelist_tier3:
- data['pricelist'] = 'tier3'
if res_user.parent_id:
data.update({ 'company': res_user.parent_id.company_type == 'company' })