initial commit 2

1 files changed, 46 insertions, 0 deletions

diff --git a/addons/google_drive/models/res_config_settings.py b/addons/google_drive/models/res_config_settings.py
new file mode 100644
index 00000000..8107dd8d
--- /dev/null
+++ b/addons/google_drive/models/res_config_settings.py
@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+
+
+class ResConfigSettings(models.TransientModel):
+    _inherit = "res.config.settings"
+
+    google_drive_authorization_code = fields.Char(string='Authorization Code', config_parameter='google_drive_authorization_code')
+    google_drive_uri = fields.Char(compute='_compute_drive_uri', string='URI', help="The URL to generate the authorization code from Google")
+    is_google_drive_token_generated = fields.Boolean(string='Refresh Token Generated')
+
+    @api.depends('google_drive_authorization_code')
+    def _compute_drive_uri(self):
+        google_drive_uri = self.env['google.service']._get_google_token_uri('drive', scope=self.env['google.drive.config'].get_google_scope())
+        for config in self:
+            config.google_drive_uri = google_drive_uri
+
+    def get_values(self):
+        res = super(ResConfigSettings, self).get_values()
+        refresh_token = self.env['ir.config_parameter'].sudo().get_param('google_drive_refresh_token', False)
+        res.update(is_google_drive_token_generated=bool(refresh_token))
+        return res
+
+    def confirm_setup_token(self):
+        params = self.env['ir.config_parameter'].sudo()
+        authorization_code_before = params.get_param('google_drive_authorization_code')
+        authorization_code = self.google_drive_authorization_code
+        if authorization_code != authorization_code_before:
+            refresh_token = (
+                self.env['google.service'].generate_refresh_token('drive', authorization_code)
+                if authorization_code else False
+            )
+            params.set_param('google_drive_refresh_token', refresh_token)
+
+    def action_setup_token(self):
+        self.ensure_one()
+        template = self.env.ref('google_drive.google_drive_auth_code_wizard')
+        return {
+            'name': _('Set up refresh token'),
+            'type': 'ir.actions.act_window',
+            'res_model': 'res.config.settings',
+            'views': [(template.id, 'form')],
+            'target': 'new',
+        }