160 lines
6.4 KiB
Python
160 lines
6.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
import logging
|
|
import uuid
|
|
import werkzeug.urls
|
|
|
|
from odoo import api, fields, models
|
|
from odoo.addons.iap.tools import iap_tools
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
DEFAULT_ENDPOINT = 'https://iap.odoo.com'
|
|
|
|
|
|
class IapAccount(models.Model):
|
|
_name = 'iap.account'
|
|
_rec_name = 'service_name'
|
|
_description = 'IAP Account'
|
|
|
|
service_name = fields.Char()
|
|
account_token = fields.Char(default=lambda s: uuid.uuid4().hex)
|
|
company_ids = fields.Many2many('res.company')
|
|
|
|
@api.model_create_multi
|
|
def create(self, vals_list):
|
|
accounts = super().create(vals_list)
|
|
if self.env['ir.config_parameter'].sudo().get_param('database.is_neutralized'):
|
|
# Disable new accounts on a neutralized database
|
|
for account in accounts:
|
|
account.account_token = f"{account.account_token.split('+')[0]}+disabled"
|
|
return accounts
|
|
|
|
@api.model
|
|
def get(self, service_name, force_create=True):
|
|
domain = [
|
|
('service_name', '=', service_name),
|
|
'|',
|
|
('company_ids', 'in', self.env.companies.ids),
|
|
('company_ids', '=', False)
|
|
]
|
|
accounts = self.search(domain, order='id desc')
|
|
accounts_without_token = accounts.filtered(lambda acc: not acc.account_token)
|
|
if accounts_without_token:
|
|
with self.pool.cursor() as cr:
|
|
# In case of a further error that will rollback the database, we should
|
|
# use a different SQL cursor to avoid undo the accounts deletion.
|
|
|
|
# Flush the pending operations to avoid a deadlock.
|
|
self.env.flush_all()
|
|
IapAccount = self.with_env(self.env(cr=cr))
|
|
# Need to use sudo because regular users do not have delete right
|
|
IapAccount.search(domain + [('account_token', '=', False)]).sudo().unlink()
|
|
accounts = accounts - accounts_without_token
|
|
if not accounts:
|
|
with self.pool.cursor() as cr:
|
|
# Since the account did not exist yet, we will encounter a NoCreditError,
|
|
# which is going to rollback the database and undo the account creation,
|
|
# preventing the process to continue any further.
|
|
|
|
# Flush the pending operations to avoid a deadlock.
|
|
self.env.flush_all()
|
|
IapAccount = self.with_env(self.env(cr=cr))
|
|
account = IapAccount.search(domain, order='id desc', limit=1)
|
|
if not account:
|
|
if not force_create:
|
|
return account
|
|
account = IapAccount.create({'service_name': service_name})
|
|
# fetch 'account_token' into cache with this cursor,
|
|
# as self's cursor cannot see this account
|
|
account_token = account.account_token
|
|
account = self.browse(account.id)
|
|
self.env.cache.set(account, IapAccount._fields['account_token'], account_token)
|
|
return account
|
|
accounts_with_company = accounts.filtered(lambda acc: acc.company_ids)
|
|
if accounts_with_company:
|
|
return accounts_with_company[0]
|
|
return accounts[0]
|
|
|
|
@api.model
|
|
def get_credits_url(self, service_name, base_url='', credit=0, trial=False):
|
|
""" Called notably by ajax crash manager, buy more widget, partner_autocomplete, sanilmail. """
|
|
dbuuid = self.env['ir.config_parameter'].sudo().get_param('database.uuid')
|
|
if not base_url:
|
|
endpoint = iap_tools.iap_get_endpoint(self.env)
|
|
route = '/iap/1/credit'
|
|
base_url = endpoint + route
|
|
account_token = self.get(service_name).account_token
|
|
d = {
|
|
'dbuuid': dbuuid,
|
|
'service_name': service_name,
|
|
'account_token': account_token,
|
|
'credit': credit,
|
|
}
|
|
if trial:
|
|
d.update({'trial': trial})
|
|
return '%s?%s' % (base_url, werkzeug.urls.url_encode(d))
|
|
|
|
@api.model
|
|
def get_account_url(self):
|
|
""" Called only by res settings """
|
|
route = '/iap/services'
|
|
endpoint = iap_tools.iap_get_endpoint(self.env)
|
|
all_accounts = self.search([
|
|
'|',
|
|
('company_ids', '=', self.env.company.id),
|
|
('company_ids', '=', False),
|
|
])
|
|
|
|
global_account_per_service = {
|
|
account.service_name: account.account_token
|
|
for account in all_accounts.filtered(lambda acc: not acc.company_ids)
|
|
}
|
|
company_account_per_service = {
|
|
account.service_name: account.account_token
|
|
for account in all_accounts.filtered(lambda acc: acc.company_ids)
|
|
}
|
|
|
|
# Prioritize company specific accounts over global accounts
|
|
account_per_service = {**global_account_per_service, **company_account_per_service}
|
|
|
|
parameters = {'tokens': list(account_per_service.values())}
|
|
|
|
return '%s?%s' % (endpoint + route, werkzeug.urls.url_encode(parameters))
|
|
|
|
@api.model
|
|
def get_config_account_url(self):
|
|
""" Called notably by ajax partner_autocomplete. """
|
|
account = self.env['iap.account'].get('partner_autocomplete')
|
|
action = self.env.ref('iap.iap_account_action')
|
|
menu = self.env.ref('iap.iap_account_menu')
|
|
no_one = self.user_has_groups('base.group_no_one')
|
|
if account:
|
|
url = "/web#id=%s&action=%s&model=iap.account&view_type=form&menu_id=%s" % (account.id, action.id, menu.id)
|
|
else:
|
|
url = "/web#action=%s&model=iap.account&view_type=form&menu_id=%s" % (action.id, menu.id)
|
|
return no_one and url
|
|
|
|
@api.model
|
|
def get_credits(self, service_name):
|
|
account = self.get(service_name, force_create=False)
|
|
credit = 0
|
|
|
|
if account:
|
|
route = '/iap/1/balance'
|
|
endpoint = iap_tools.iap_get_endpoint(self.env)
|
|
url = endpoint + route
|
|
params = {
|
|
'dbuuid': self.env['ir.config_parameter'].sudo().get_param('database.uuid'),
|
|
'account_token': account.account_token,
|
|
'service_name': service_name,
|
|
}
|
|
try:
|
|
credit = iap_tools.iap_jsonrpc(url=url, params=params)
|
|
except Exception as e:
|
|
_logger.info('Get credit error : %s', str(e))
|
|
credit = -1
|
|
|
|
return credit
|