You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
3.4 KiB
77 lines
3.4 KiB
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2023-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Shafna K(odoo@cybrosys.com)
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU AFFERO GENERAL PUBLIC LICENSE (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
import boto3
|
|
from botocore.exceptions import ClientError
|
|
from odoo import fields, models, _
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class ResConfigSettings(models.TransientModel):
|
|
"""Class inheriting the res config settings for adding fields for the
|
|
authentication of AWS with Odoo"""
|
|
_inherit = 'res.config.settings'
|
|
|
|
amazon_forecast = fields.Boolean(
|
|
string="Amazon Forecast", required=True,
|
|
config_parameter='amazon_forecast_integration.amazon_forecast',
|
|
help="Enable to use Amazon Forecast services.")
|
|
amazon_access_key = fields.Char(
|
|
string="Access Key", help="Provide the access key.", required=True,
|
|
config_parameter='amazon_forecast_integration.amazon_access_key')
|
|
amazon_secret_access_key = fields.Char(
|
|
string="Secret Access Key", required=True,
|
|
config_parameter='amazon_forecast_integration.amazon_secret_access_key',
|
|
help="Provide the secret access key.")
|
|
amazon_region = fields.Char(
|
|
string="Access Key", help="Provide the region.", required=True,
|
|
config_parameter='amazon_forecast_integration.amazon_amazon_region')
|
|
|
|
def authenticate_amazon_forecast(self):
|
|
"""Function to authenticate the AWS connection with Odoo"""
|
|
try:
|
|
session = boto3.Session(
|
|
aws_access_key_id=self.amazon_access_key,
|
|
aws_secret_access_key=self.amazon_secret_access_key,
|
|
region_name=self.amazon_region
|
|
)
|
|
iam_client = session.client('iam')
|
|
response = iam_client.list_users()
|
|
return response
|
|
except ClientError:
|
|
error_message = _('Invalid Amazon Forecast credentials. Please'
|
|
' verify your access key and secret key.')
|
|
raise UserError(error_message)
|
|
|
|
def set_values(self):
|
|
"""To set the values in the corresponding fields"""
|
|
super().set_values()
|
|
if self.amazon_forecast:
|
|
self.authenticate_amazon_forecast()
|
|
self.env['ir.config_parameter'].set_param(
|
|
'amazon_forecast_integration.amazon_forecast_access_key',
|
|
self.amazon_access_key or '')
|
|
self.env['ir.config_parameter'].set_param(
|
|
'amazon_forecast_integration.amazon_forecast_secret_key',
|
|
self.amazon_secret_access_key or '')
|
|
self.env['ir.config_parameter'].set_param(
|
|
'amazon_forecast_integration.amazon_region',
|
|
self.amazon_region or '')
|
|
|