# Copyright 2020 Coop IT Easy SCRL fs # Robin Keunen # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html). import json import logging import requests from werkzeug.exceptions import ( InternalServerError, NotFound, ) from odoo import _, api, fields, models from odoo.exceptions import ( AccessDenied, ) _logger = logging.getLogger(__name__) class EMCBackend(models.Model): _name = "emc.backend" _description = "EMC Backend" name = fields.Char(string="Name", required=True) location = fields.Char(string="Location") api_key = fields.Char(string="API Key") description = fields.Text(string="Description", required=False) active = fields.Boolean(string="active", default=True) @api.multi def http_get(self, url, params=None, headers=None): self.ensure_one() headers = self._add_api_key(headers) if url.startswith("/"): url = self.location + url return requests.get(url, params=params, headers=headers) @api.multi def http_get_content(self, url, params=None, headers=None): self.ensure_one() response = self.http_get(url, params=params, headers=headers) if response.status_code == 200: content = response.content.decode("utf-8") return json.loads(content) elif response.status_code == 403: raise AccessDenied(_("You are not allowed to access this resource")) elif response.status_code == 404: raise NotFound( _("Resource not found %s on server" % response.status_code) ) else: # 500 et al. content = response.content.decode("utf-8") raise InternalServerError( _("request returned status code %s with message %s" % ( response.status_code, content)) ) @api.multi def http_post(self, url, data, headers=None): self.ensure_one() headers = self._add_api_key(headers) if url.startswith("/"): url = self.location + url return requests.post(url, json=data, headers=headers) @api.multi def _add_api_key(self, headers): self.ensure_one() key_dict = {"API-KEY": self.api_key} if headers: headers.update(key_dict) else: headers = key_dict return headers @api.multi def action_ping(self): self.ensure_one() url = self.location + "/api/ping/test" try: response = requests.get(url) except Exception as e: _logger.error(e) raise Warning(_("Failed to connect to backend: %s" % str(e))) if response.status_code == 200: content = json.loads(response.content.decode("utf-8")) raise Warning(_("Success: %s") % content["message"]) else: raise Warning( _("Failed to connect to backend: %s" % str(response.content)) )