initial commit 2

2 files changed, 169 insertions, 0 deletions

diff --git a/addons/google_account/models/__init__.py b/addons/google_account/models/__init__.py
new file mode 100644
index 00000000..97134e71
--- /dev/null
+++ b/addons/google_account/models/__init__.py
@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import google_service
diff --git a/addons/google_account/models/google_service.py b/addons/google_account/models/google_service.py
new file mode 100644
index 00000000..5a69f35b
--- /dev/null
+++ b/addons/google_account/models/google_service.py
@@ -0,0 +1,165 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from datetime import datetime
+import json
+import logging
+
+import requests
+from werkzeug import urls
+
+from odoo import api, fields, models, _
+
+_logger = logging.getLogger(__name__)
+
+TIMEOUT = 20
+
+GOOGLE_AUTH_ENDPOINT = 'https://accounts.google.com/o/oauth2/auth'
+GOOGLE_TOKEN_ENDPOINT = 'https://accounts.google.com/o/oauth2/token'
+GOOGLE_API_BASE_URL = 'https://www.googleapis.com'
+
+
+class GoogleService(models.AbstractModel):
+    _name = 'google.service'
+    _description = 'Google Service'
+
+    @api.model
+    def generate_refresh_token(self, service, authorization_code):
+        """ Call Google API to refresh the token, with the given authorization code
+            :param service : the name of the google service to actualize
+            :param authorization_code : the code to exchange against the new refresh token
+            :returns the new refresh token
+        """
+        Parameters = self.env['ir.config_parameter'].sudo()
+        client_id = Parameters.get_param('google_%s_client_id' % service)
+        client_secret = Parameters.get_param('google_%s_client_secret' % service)
+        redirect_uri = Parameters.get_param('google_redirect_uri')
+
+        # Get the Refresh Token From Google And store it in ir.config_parameter
+        headers = {"Content-type": "application/x-www-form-urlencoded"}
+        data = {
+            'code': authorization_code,
+            'client_id': client_id,
+            'client_secret': client_secret,
+            'redirect_uri': redirect_uri,
+            'grant_type': "authorization_code"
+        }
+        try:
+            req = requests.post(GOOGLE_TOKEN_ENDPOINT, data=data, headers=headers, timeout=TIMEOUT)
+            req.raise_for_status()
+            content = req.json()
+        except IOError:
+            error_msg = _("Something went wrong during your token generation. Maybe your Authorization Code is invalid or already expired")
+            raise self.env['res.config.settings'].get_config_warning(error_msg)
+
+        return content.get('refresh_token')
+
+    @api.model
+    def _get_google_token_uri(self, service, scope):
+        get_param = self.env['ir.config_parameter'].sudo().get_param
+        encoded_params = urls.url_encode({
+            'scope': scope,
+            'redirect_uri': get_param('google_redirect_uri'),
+            'client_id': get_param('google_%s_client_id' % service),
+            'response_type': 'code',
+        })
+        return '%s?%s' % (GOOGLE_AUTH_ENDPOINT, encoded_params)
+
+    @api.model
+    def _get_authorize_uri(self, from_url, service, scope=False):
+        """ This method return the url needed to allow this instance of Odoo to access to the scope
+            of gmail specified as parameters
+        """
+        state = {
+            'd': self.env.cr.dbname,
+            's': service,
+            'f': from_url
+        }
+
+        get_param = self.env['ir.config_parameter'].sudo().get_param
+        base_url = get_param('web.base.url', default='http://www.odoo.com?NoBaseUrl')
+        client_id = get_param('google_%s_client_id' % (service,), default=False)
+
+        encoded_params = urls.url_encode({
+            'response_type': 'code',
+            'client_id': client_id,
+            'state': json.dumps(state),
+            'scope': scope or '%s/auth/%s' % (GOOGLE_API_BASE_URL, service),  # If no scope is passed, we use service by default to get a default scope
+            'redirect_uri': base_url + '/google_account/authentication',
+            'approval_prompt': 'force',
+            'access_type': 'offline'
+        })
+        return "%s?%s" % (GOOGLE_AUTH_ENDPOINT, encoded_params)
+
+    @api.model
+    def _get_google_tokens(self, authorize_code, service):
+        """ Call Google API to exchange authorization code against token, with POST request, to
+            not be redirected.
+        """
+        get_param = self.env['ir.config_parameter'].sudo().get_param
+        base_url = get_param('web.base.url', default='http://www.odoo.com?NoBaseUrl')
+        client_id = get_param('google_%s_client_id' % (service,), default=False)
+        client_secret = get_param('google_%s_client_secret' % (service,), default=False)
+
+        headers = {"content-type": "application/x-www-form-urlencoded"}
+        data = {
+            'code': authorize_code,
+            'client_id': client_id,
+            'client_secret': client_secret,
+            'grant_type': 'authorization_code',
+            'redirect_uri': base_url + '/google_account/authentication'
+        }
+        try:
+            dummy, response, dummy = self._do_request(GOOGLE_TOKEN_ENDPOINT, params=data, headers=headers, method='POST', preuri='')
+            access_token = response.get('access_token')
+            refresh_token = response.get('refresh_token')
+            ttl = response.get('expires_in')
+            return access_token, refresh_token, ttl
+        except requests.HTTPError:
+            error_msg = _("Something went wrong during your token generation. Maybe your Authorization Code is invalid")
+            raise self.env['res.config.settings'].get_config_warning(error_msg)
+
+    @api.model
+    def _do_request(self, uri, params=None, headers=None, method='POST', preuri="https://www.googleapis.com", timeout=TIMEOUT):
+        """ Execute the request to Google API. Return a tuple ('HTTP_CODE', 'HTTP_RESPONSE')
+            :param uri : the url to contact
+            :param params : dict or already encoded parameters for the request to make
+            :param headers : headers of request
+            :param method : the method to use to make the request
+            :param preuri : pre url to prepend to param uri.
+        """
+        if params is None:
+            params = {}
+        if headers is None:
+            headers = {}
+
+        _logger.debug("Uri: %s - Type : %s - Headers: %s - Params : %s !", (uri, method, headers, params))
+
+        ask_time = fields.Datetime.now()
+        try:
+            if method.upper() in ('GET', 'DELETE'):
+                res = requests.request(method.lower(), preuri + uri, params=params, timeout=timeout)
+            elif method.upper() in ('POST', 'PATCH', 'PUT'):
+                res = requests.request(method.lower(), preuri + uri, data=params, headers=headers, timeout=timeout)
+            else:
+                raise Exception(_('Method not supported [%s] not in [GET, POST, PUT, PATCH or DELETE]!') % (method))
+            res.raise_for_status()
+            status = res.status_code
+
+            if int(status) in (204, 404):  # Page not found, no response
+                response = False
+            else:
+                response = res.json()
+
+            try:
+                ask_time = datetime.strptime(res.headers.get('date'), "%a, %d %b %Y %H:%M:%S %Z")
+            except:
+                pass
+        except requests.HTTPError as error:
+            if error.response.status_code in (204, 404):
+                status = error.response.status_code
+                response = ""
+            else:
+                _logger.exception("Bad google request : %s !", error.response.content)
+                raise error
+        return (status, response, ask_time)