OCA reporting engine fork for dev and update.
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.

354 lines
12 KiB

  1. # -*- coding: utf-8 -*-
  2. # Copyright 2015-2017 Onestein (<http://www.onestein.eu>)
  3. # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
  4. import json
  5. from openerp import api, fields, models, tools
  6. from openerp.exceptions import Warning as UserError
  7. from openerp.tools.translate import _
  8. class BveView(models.Model):
  9. _name = 'bve.view'
  10. _description = 'BI View Editor'
  11. @api.depends('group_ids')
  12. @api.multi
  13. def _compute_users(self):
  14. for bve_view in self:
  15. group_ids = bve_view.sudo().group_ids
  16. if group_ids:
  17. bve_view.user_ids = group_ids.mapped('users')
  18. else:
  19. bve_view.user_ids = self.env['res.users'].sudo().search([])
  20. name = fields.Char(required=True, copy=False)
  21. model_name = fields.Char()
  22. note = fields.Text(string='Notes')
  23. state = fields.Selection(
  24. [('draft', 'Draft'),
  25. ('created', 'Created')],
  26. default='draft',
  27. copy=False)
  28. data = fields.Text(
  29. help="Use the special query builder to define the query "
  30. "to generate your report dataset. "
  31. "NOTE: Te be edited, the query should be in 'Draft' status.")
  32. action_id = fields.Many2one('ir.actions.act_window', string='Action')
  33. view_id = fields.Many2one('ir.ui.view', string='View')
  34. group_ids = fields.Many2many(
  35. 'res.groups',
  36. string='Groups',
  37. help="User groups allowed to see the generated report; "
  38. "if NO groups are specified the report will be public "
  39. "for everyone.")
  40. user_ids = fields.Many2many(
  41. 'res.users',
  42. string='Users',
  43. compute=_compute_users,
  44. store=True)
  45. _sql_constraints = [
  46. ('name_uniq',
  47. 'unique(name)',
  48. _('Custom BI View names must be unique!')),
  49. ]
  50. @api.multi
  51. def unlink(self):
  52. for view in self:
  53. if view.state == 'created':
  54. raise UserError(
  55. _('You cannot delete a created view! '
  56. 'Reset the view to draft first.'))
  57. return super(BveView, self).unlink()
  58. @api.multi
  59. def action_reset(self):
  60. self.ensure_one()
  61. if self.action_id:
  62. if self.action_id.view_id:
  63. self.action_id.view_id.sudo().unlink()
  64. self.action_id.sudo().unlink()
  65. models = self.env['ir.model'].sudo().search(
  66. [('model', '=', self.model_name)])
  67. for model in models:
  68. model.sudo().unlink()
  69. table_name = self.model_name.replace('.', '_')
  70. tools.drop_view_if_exists(self.env.cr, table_name)
  71. self.state = 'draft'
  72. @api.multi
  73. def _create_view_arch(self):
  74. self.ensure_one()
  75. def _get_field_def(field_name, def_type):
  76. return """<field name="x_{}" type="{}" />""".format(
  77. field_name, def_type
  78. )
  79. def _get_field_type(field_info):
  80. row = field_info['row'] and 'row'
  81. column = field_info['column'] and 'col'
  82. measure = field_info['measure'] and 'measure'
  83. return row or column or measure
  84. fields_info = json.loads(self._get_format_data(self.data))
  85. view_fields = []
  86. for field_info in fields_info:
  87. field_name = field_info['name']
  88. def_type = _get_field_type(field_info)
  89. if def_type:
  90. field_def = _get_field_def(field_name, def_type)
  91. view_fields.append(field_def)
  92. return view_fields
  93. @api.model
  94. def _get_format_data(self, data):
  95. data = data.replace('\'', '"')
  96. data = data.replace(': u"', ':"')
  97. return data
  98. @api.multi
  99. def action_create(self):
  100. self.ensure_one()
  101. self._create_bve_object()
  102. self._create_bve_view()
  103. @api.multi
  104. def _create_bve_view(self):
  105. self.ensure_one()
  106. # create views
  107. View = self.env['ir.ui.view']
  108. old_views = View.sudo().search([('model', '=', self.model_name)])
  109. old_views.sudo().unlink()
  110. view_vals = [{
  111. 'name': 'Pivot Analysis',
  112. 'type': 'pivot',
  113. 'model': self.model_name,
  114. 'priority': 16,
  115. 'arch': """<?xml version="1.0"?>
  116. <pivot string="Pivot Analysis"> {} </pivot>
  117. """.format("".join(self._create_view_arch()))
  118. }, {
  119. 'name': 'Graph Analysis',
  120. 'type': 'graph',
  121. 'model': self.model_name,
  122. 'priority': 16,
  123. 'arch': """<?xml version="1.0"?>
  124. <graph string="Graph Analysis"
  125. type="bar"
  126. stacked="True"> {} </graph>
  127. """.format("".join(self._create_view_arch()))
  128. }]
  129. for vals in view_vals:
  130. View.sudo().create(vals)
  131. # create Tree view
  132. tree_view = View.sudo().create(
  133. {'name': 'Tree Analysis',
  134. 'type': 'tree',
  135. 'model': self.model_name,
  136. 'priority': 16,
  137. 'arch': """<?xml version="1.0"?>
  138. <tree string="List Analysis" create="false"> {} </tree>
  139. """.format("".join(self._create_view_arch()))
  140. })
  141. # set the Tree view as the default one
  142. action_vals = {
  143. 'name': self.name,
  144. 'res_model': self.model_name,
  145. 'type': 'ir.actions.act_window',
  146. 'view_type': 'form',
  147. 'view_mode': 'tree,graph,pivot',
  148. 'view_id': tree_view.id,
  149. 'context': "{'service_name': '%s'}" % self.name,
  150. }
  151. ActWindow = self.env['ir.actions.act_window']
  152. action_id = ActWindow.sudo().create(action_vals)
  153. self.write({
  154. 'action_id': action_id.id,
  155. 'view_id': tree_view.id,
  156. 'state': 'created'
  157. })
  158. @api.multi
  159. def _create_bve_object(self):
  160. self.ensure_one()
  161. def _get_fields_info(fields_data):
  162. fields_info = []
  163. for field_data in fields_data:
  164. field = self.env['ir.model.fields'].browse(field_data['id'])
  165. vals = {
  166. 'table': self.env[field.model_id.model]._table,
  167. 'table_alias': field_data['table_alias'],
  168. 'select_field': field.name,
  169. 'as_field': 'x_' + field_data['name'],
  170. 'join': False,
  171. 'model': field.model_id.model
  172. }
  173. if field_data.get('join_node'):
  174. vals.update({'join': field_data['join_node']})
  175. fields_info.append(vals)
  176. return fields_info
  177. def _build_query():
  178. data = self.data
  179. if not data:
  180. raise UserError(_('No data to process.'))
  181. formatted_data = json.loads(self._get_format_data(data))
  182. info = _get_fields_info(formatted_data)
  183. fields = [("{}.{}".format(f['table_alias'],
  184. f['select_field']),
  185. f['as_field']) for f in info if 'join_node' not in f]
  186. tables = set([(f['table'], f['table_alias']) for f in info])
  187. join_nodes = [
  188. (f['table_alias'],
  189. f['join'],
  190. f['select_field']) for f in info if f['join'] is not False]
  191. table_name = self.model_name.replace('.', '_')
  192. tools.drop_view_if_exists(self.env.cr, table_name)
  193. basic_fields = [
  194. ("t0.id", "id"),
  195. ("t0.write_uid", "write_uid"),
  196. ("t0.write_date", "write_date"),
  197. ("t0.create_uid", "create_uid"),
  198. ("t0.create_date", "create_date")
  199. ]
  200. q = """CREATE or REPLACE VIEW %s as (
  201. SELECT %s
  202. FROM %s
  203. WHERE %s
  204. )""" % (table_name, ','.join(
  205. ["{} AS {}".format(f[0], f[1])
  206. for f in basic_fields + fields]), ','.join(
  207. ["{} AS {}".format(t[0], t[1])
  208. for t in list(tables)]), " AND ".join(
  209. ["{}.{} = {}.id".format(j[0], j[2], j[1])
  210. for j in join_nodes] + ["TRUE"]))
  211. self.env.cr.execute(q)
  212. def _prepare_field(field_data):
  213. if not field_data['custom']:
  214. field = self.env['ir.model.fields'].browse(field_data['id'])
  215. vals = {
  216. 'name': 'x_' + field_data['name'],
  217. 'complete_name': field.complete_name,
  218. 'model': self.model_name,
  219. 'relation': field.relation,
  220. 'field_description': field_data.get(
  221. 'description', field.field_description),
  222. 'ttype': field.ttype,
  223. 'selection': field.selection,
  224. 'size': field.size,
  225. 'state': 'manual'
  226. }
  227. if vals['ttype'] == 'monetary':
  228. vals.update({'ttype': 'float'})
  229. if field.ttype == 'selection' and not field.selection:
  230. model_obj = self.env[field.model_id.model]
  231. selection = model_obj._columns[field.name].selection
  232. selection_domain = str(selection)
  233. vals.update({'selection': selection_domain})
  234. return vals
  235. def _prepare_object():
  236. data = json.loads(self._get_format_data(self.data))
  237. return {
  238. 'name': self.name,
  239. 'model': self.model_name,
  240. 'field_id': [
  241. (0, 0, _prepare_field(field))
  242. for field in data
  243. if 'join_node' not in field]
  244. }
  245. def _build_object():
  246. vals = _prepare_object()
  247. Model = self.env['ir.model']
  248. res_id = Model.sudo().with_context(bve=True).create(vals)
  249. return res_id
  250. def group_ids_with_access(model_name, access_mode):
  251. self.env.cr.execute('''SELECT
  252. g.id
  253. FROM
  254. ir_model_access a
  255. JOIN ir_model m ON (a.model_id=m.id)
  256. JOIN res_groups g ON (a.group_id=g.id)
  257. LEFT JOIN ir_module_category c ON (c.id=g.category_id)
  258. WHERE
  259. m.model=%s AND
  260. a.active IS True AND
  261. a.perm_''' + access_mode, (model_name,))
  262. return [x[0] for x in self.env.cr.fetchall()]
  263. def _build_access_rules(obj):
  264. info = json.loads(self._get_format_data(self.data))
  265. models = list(set([f['model'] for f in info]))
  266. read_groups = set.intersection(*[set(
  267. group_ids_with_access(model, 'read')) for model in models])
  268. # read access
  269. for group in read_groups:
  270. self.env['ir.model.access'].sudo().create({
  271. 'name': 'read access to ' + self.model_name,
  272. 'model_id': obj.id,
  273. 'group_id': group,
  274. 'perm_read': True,
  275. })
  276. # read and write access
  277. for group in self.group_ids:
  278. self.env['ir.model.access'].sudo().create({
  279. 'name': 'read-write access to ' + self.model_name,
  280. 'model_id': obj.id,
  281. 'group_id': group.id,
  282. 'perm_read': True,
  283. 'perm_write': True,
  284. })
  285. self.model_name = 'x_bve.' + ''.join(
  286. [x for x in self.name.lower()
  287. if x.isalnum()]).replace('_', '.').replace(' ', '.')
  288. _build_query()
  289. obj = _build_object()
  290. _build_access_rules(obj)
  291. @api.multi
  292. def open_view(self):
  293. self.ensure_one()
  294. return {
  295. 'name': _('BI View'),
  296. 'type': 'ir.actions.act_window',
  297. 'res_model': self.model_name,
  298. 'view_type': 'form',
  299. 'view_mode': 'tree,graph,pivot',
  300. }
  301. @api.multi
  302. def copy(self, default=None):
  303. self.ensure_one()
  304. default = dict(default or {}, name=_("%s (copy)") % self.name)
  305. return super(BveView, self).copy(default=default)