Xavier Morel
de590816d8
In 0.15 accessing werkzeug.urls functions directly through werkzeug is deprecated, the shortcut will be removed in the eventual werkzeug 1.0. Fix existing uses of these shortcuts. Also cleanup some imports when they're not far from a werkzeug* import being altered.
108 lines
4.5 KiB
Python
108 lines
4.5 KiB
Python
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
import json
|
|
import logging
|
|
|
|
import requests
|
|
from lxml import etree
|
|
import re
|
|
import werkzeug.urls
|
|
|
|
from odoo import api, models
|
|
from odoo.tools import misc
|
|
from odoo.addons.google_account import TIMEOUT
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class GoogleDrive(models.Model):
|
|
_inherit = 'google.drive.config'
|
|
|
|
def get_google_scope(self):
|
|
scope = super(GoogleDrive, self).get_google_scope()
|
|
return '%s https://spreadsheets.google.com/feeds' % scope
|
|
|
|
@api.model
|
|
def write_config_formula(self, attachment_id, spreadsheet_key, model, domain, groupbys, view_id):
|
|
access_token = self.get_access_token(scope='https://spreadsheets.google.com/feeds')
|
|
|
|
fields = self.env[model].fields_view_get(view_id=view_id, view_type='tree')
|
|
doc = etree.XML(fields.get('arch'))
|
|
display_fields = []
|
|
for node in doc.xpath("//field"):
|
|
if node.get('modifiers'):
|
|
modifiers = json.loads(node.get('modifiers'))
|
|
if not modifiers.get('invisible') and not modifiers.get('column_invisible'):
|
|
display_fields.append(node.get('name'))
|
|
fields = " ".join(display_fields)
|
|
domain = domain.replace("'", r"\'").replace('"', "'").replace('True', 'true').replace('False', 'false')
|
|
if groupbys:
|
|
fields = "%s %s" % (groupbys, fields)
|
|
formula = '=oe_read_group("%s";"%s";"%s";"%s")' % (model, fields, groupbys, domain)
|
|
else:
|
|
formula = '=oe_browse("%s";"%s";"%s")' % (model, fields, domain)
|
|
url = self.env['ir.config_parameter'].sudo().get_param('web.base.url')
|
|
dbname = self._cr.dbname
|
|
user = self.env['res.users'].browse(self.env.user.id).read(['login', 'password'])[0]
|
|
username = user['login']
|
|
password = user['password']
|
|
if not password:
|
|
config_formula = '=oe_settings("%s";"%s")' % (url, dbname)
|
|
else:
|
|
config_formula = '=oe_settings("%s";"%s";"%s";"%s")' % (url, dbname, username, password)
|
|
request = '''<feed xmlns="http://www.w3.org/2005/Atom"
|
|
xmlns:batch="http://schemas.google.com/gdata/batch"
|
|
xmlns:gs="http://schemas.google.com/spreadsheets/2006">
|
|
<id>https://spreadsheets.google.com/feeds/cells/{key}/od6/private/full</id>
|
|
<entry>
|
|
<batch:id>A1</batch:id>
|
|
<batch:operation type="update"/>
|
|
<id>https://spreadsheets.google.com/feeds/cells/{key}/od6/private/full/R1C1</id>
|
|
<link rel="edit" type="application/atom+xml"
|
|
href="https://spreadsheets.google.com/feeds/cells/{key}/od6/private/full/R1C1"/>
|
|
<gs:cell row="1" col="1" inputValue="{formula}"/>
|
|
</entry>
|
|
<entry>
|
|
<batch:id>A2</batch:id>
|
|
<batch:operation type="update"/>
|
|
<id>https://spreadsheets.google.com/feeds/cells/{key}/od6/private/full/R60C15</id>
|
|
<link rel="edit" type="application/atom+xml"
|
|
href="https://spreadsheets.google.com/feeds/cells/{key}/od6/private/full/R60C15"/>
|
|
<gs:cell row="60" col="15" inputValue="{config}"/>
|
|
</entry>
|
|
</feed>''' .format(key=spreadsheet_key, formula=misc.html_escape(formula), config=misc.html_escape(config_formula))
|
|
|
|
try:
|
|
req = requests.post(
|
|
'https://spreadsheets.google.com/feeds/cells/%s/od6/private/full/batch?%s' % (spreadsheet_key, werkzeug.urls.url_encode({'v': 3, 'access_token': access_token})),
|
|
data=request,
|
|
headers={'content-type': 'application/atom+xml', 'If-Match': '*'},
|
|
timeout=TIMEOUT,
|
|
)
|
|
except IOError:
|
|
_logger.warning("An error occured while writing the formula on the Google Spreadsheet.")
|
|
|
|
description = '''
|
|
formula: %s
|
|
''' % formula
|
|
if attachment_id:
|
|
self.env['ir.attachment'].browse(attachment_id).write({'description': description})
|
|
return True
|
|
|
|
@api.model
|
|
def set_spreadsheet(self, model, domain, groupbys, view_id):
|
|
try:
|
|
config_id = self.env['ir.model.data'].get_object_reference('google_spreadsheet', 'google_spreadsheet_template')[1]
|
|
except ValueError:
|
|
raise
|
|
config = self.browse(config_id)
|
|
title = 'Spreadsheet %s' % model
|
|
res = self.copy_doc(False, config.google_drive_resource_id, title, model)
|
|
|
|
mo = re.search("(key=|/d/)([A-Za-z0-9-_]+)", res['url'])
|
|
if mo:
|
|
key = mo.group(2)
|
|
|
|
self.write_config_formula(res.get('id'), key, model, domain, groupbys, view_id)
|
|
return res
|