audrius/reporting-engine
1
0
Fork 1
mirror of https://github.com/OCA/reporting-engine.git synced 2025-02-16 16:30:38 +02:00
Code Issues Packages Projects Releases Wiki Activity

[IMP] sql_export_mail: black, isort, prettier

Browse Source
This commit is contained in:
hkapatel-initos
2021-06-14 16:57:16 +05:30
committed by Sylvain LE GAL
parent 1dedee2aae
commit f7585a5b92
5 changed files with 122 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
sql_export_mail/__manifest__.py
View File

@@ -1,20 +1,19 @@
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
{
'name': 'SQL Export Mail',
'version': '12.0.1.0.0',
'category': 'Generic Modules',
'summary': 'Send csv file generated by sql query by mail.',
'author':
"Akretion, Odoo Community Association (OCA)",
'website': 'https://github.com/OCA/server-tools',
'depends': [
'sql_export',
'mail',
"name": "SQL Export Mail",
"version": "12.0.1.0.0",
"category": "Generic Modules",
"summary": "Send csv file generated by sql query by mail.",
"author": "Akretion, Odoo Community Association (OCA)",
"website": "https://github.com/OCA/server-tools",
"depends": [
"sql_export",
"mail",
],
'license': 'AGPL-3',
'data': [
'views/sql_export_view.xml',
'mail_template.xml',
"license": "AGPL-3",
"data": [
"views/sql_export_view.xml",
"mail_template.xml",
],
'installable': True,
"installable": True,
}

11
sql_export_mail/mail_template.xml
View File

@@ -1,5 +1,4 @@
<?xml version="1.0" encoding="utf-8"?>
<?xml version="1.0" encoding="utf-8" ?>
<odoo noupdate="1">
<!-- Error Email template -->
@@ -8,9 +7,11 @@
<field name="email_from">admin@example.com</field>
<field name="email_to">${object.get_email_address_for_template()}</field>
<field name="subject">${object.name or ''}</field>
<field name="model_id" ref="sql_export.model_sql_export"/>
<field name="auto_delete" eval="True"/>
<field name="body_html"><![CDATA[
<field name="model_id" ref="sql_export.model_sql_export" />
<field name="auto_delete" eval="True" />
<field
name="body_html"
><![CDATA[
<div style="font-family: 'Lucida Grande', Ubuntu, Arial, Verdana, sans-serif; font-size: 12px; color: rgb(34, 34, 34); background-color: #FFF; ">
<p>You will find the report ${object.name or ''} as an attachment of the mail.</p>

150
sql_export_mail/models/sql_export.py
View File

@@ -1,148 +1,148 @@
# Copyright 2019 Akretion
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from odoo import models, fields, api, _
from odoo.tools import DEFAULT_SERVER_DATETIME_FORMAT
from odoo.exceptions import UserError
from datetime import datetime, timedelta
from odoo import SUPERUSER_ID
from odoo import SUPERUSER_ID, _, api, fields, models
from odoo.exceptions import UserError
from odoo.tools import DEFAULT_SERVER_DATETIME_FORMAT
class SqlExport(models.Model):
_inherit = 'sql.export'
_inherit = "sql.export"
mail_user_ids = fields.Many2many(
'res.users',
'mail_user_sqlquery_rel',
'sql_id',
'user_id',
'User to notify',
help='Add the users who want to receive the report by e-mail. You '
'need to link the sql query with a cron to send mail automatically')
"res.users",
"mail_user_sqlquery_rel",
"sql_id",
"user_id",
"User to notify",
help="Add the users who want to receive the report by e-mail. You "
"need to link the sql query with a cron to send mail automatically",
)
cron_ids = fields.Many2many(
'ir.cron',
'cron_sqlquery_rel',
'sql_id',
'cron_id',
'Crons')
"ir.cron", "cron_sqlquery_rel", "sql_id", "cron_id", "Crons"
)
# We could implement other conditions, that is why it is a selection field
mail_condition = fields.Selection(
[('not_empty', 'File Not Empty')], default='not_empty')
[("not_empty", "File Not Empty")], default="not_empty"
)
@api.multi
def create_cron(self):
self.ensure_one()
nextcall = datetime.now() + timedelta(hours=2)
cron_vals = {
'active': True,
'model_id': self.env.ref('sql_export.model_sql_export').id,
'state': 'code',
'code': 'model._run_all_sql_export_for_cron()',
'name': 'SQL Export : %s' % self.name,
'nextcall': nextcall,
'doall': False,
'numbercall': -1,
'user_id': SUPERUSER_ID,
"active": True,
"model_id": self.env.ref("sql_export.model_sql_export").id,
"state": "code",
"code": "model._run_all_sql_export_for_cron()",
"name": "SQL Export : %s" % self.name,
"nextcall": nextcall,
"doall": False,
"numbercall": -1,
"user_id": SUPERUSER_ID,
}
cron = self.env['ir.cron'].create(cron_vals)
cron = self.env["ir.cron"].create(cron_vals)
# We need to pass cron_id in the cron args because a cron is not
# aware of itself in the end method and we need it to find all
# linked sql exports
write_vals = {
'code': 'model._run_all_sql_export_for_cron([%s])' % cron.id
}
write_vals = {"code": "model._run_all_sql_export_for_cron([%s])" % cron.id}
cron.write(write_vals)
self.write({'cron_ids': [(4, cron.id)]})
self.write({"cron_ids": [(4, cron.id)]})
@api.multi
def send_mail(self, params=None):
self.ensure_one()
mail_template = self.env.ref('sql_export_mail.sql_export_mailer')
now_time = datetime.strftime(datetime.now(),
DEFAULT_SERVER_DATETIME_FORMAT)
attach_obj = self.env['ir.attachment']
if self.mail_condition == 'not_empty':
res = self._execute_sql_request(
params=params, mode='fetchone')
mail_template = self.env.ref("sql_export_mail.sql_export_mailer")
now_time = datetime.strftime(datetime.now(), DEFAULT_SERVER_DATETIME_FORMAT)
attach_obj = self.env["ir.attachment"]
if self.mail_condition == "not_empty":
res = self._execute_sql_request(params=params, mode="fetchone")
if not res:
return
ctx = self.env.context.copy()
if params:
if 'user_id' in params:
ctx['force_user'] = params['user_id']
if 'company_id' in params:
ctx['force_company'] = params['company_id']
wizard = self.env['sql.file.wizard'].create({
'sql_export_id': self.id,
})
if "user_id" in params:
ctx["force_user"] = params["user_id"]
if "company_id" in params:
ctx["force_company"] = params["company_id"]
wizard = self.env["sql.file.wizard"].create(
{
"sql_export_id": self.id,
}
)
wizard.with_context(ctx).export_sql()
binary = wizard.binary_file
filename = wizard.file_name
msg_id = mail_template.send_mail(self.id, force_send=False)
mail = self.env['mail.mail'].browse(msg_id)
mail = self.env["mail.mail"].browse(msg_id)
attach_vals = {
'name': now_time + ' - ' + self.name,
'datas_fname': filename,
'datas': binary,
'res_model': 'mail.mail',
'res_id': mail.id,
"name": now_time + " - " + self.name,
"datas_fname": filename,
"datas": binary,
"res_model": "mail.mail",
"res_id": mail.id,
}
attachment = attach_obj.create(attach_vals)
mail.write({'attachment_ids': [(4, attachment.id)]})
mail.write({"attachment_ids": [(4, attachment.id)]})
@api.model
def _run_all_sql_export_for_cron(self, cron_ids):
exports = self.search([('cron_ids', 'in', cron_ids)])
exports = self.search([("cron_ids", "in", cron_ids)])
for export in exports:
if "%(company_id)s" in export.query and \
"%(user_id)s" not in export.query:
if "%(company_id)s" in export.query and "%(user_id)s" not in export.query:
variable_dict = {}
companies = self.env['res.company'].search([])
companies = self.env["res.company"].search([])
for company in companies:
users = export.mail_user_ids.filtered(
lambda u: u.company_id == company)
lambda u: u.company_id == company
)
if users:
variable_dict['company_id'] = users[0].company_id.id
variable_dict["company_id"] = users[0].company_id.id
export.with_context(mail_to=users.ids).send_mail(
params=variable_dict)
params=variable_dict
)
elif "%(user_id)s" in export.query:
variable_dict = {}
for user in export.mail_user_ids:
variable_dict['user_id'] = user.id
variable_dict["user_id"] = user.id
if "%(company_id)s" in export.query:
variable_dict['company_id'] = user.company_id.id
variable_dict["company_id"] = user.company_id.id
export.with_context(mail_to=[user.id]).send_mail(
params=variable_dict)
params=variable_dict
)
else:
export.send_mail()
@api.multi
@api.constrains('field_ids', 'mail_user_ids')
@api.constrains("field_ids", "mail_user_ids")
def check_no_parameter_if_sent_by_mail(self):
for export in self:
if export.field_ids and export.mail_user_ids:
raise UserError(_(
"It is not possible to execute and send a query "
"automatically by mail if there are parameters to fill"))
raise UserError(
_(
"It is not possible to execute and send a query "
"automatically by mail if there are parameters to fill"
)
)
@api.multi
@api.constrains('mail_user_ids')
@api.constrains("mail_user_ids")
def check_mail_user(self):
for export in self:
for user in export.mail_user_ids:
if not user.email:
raise UserError(_(
"The user does not have any e-mail address."))
raise UserError(_("The user does not have any e-mail address."))
@api.multi
def get_email_address_for_template(self):
"""
Called from mail template
Called from mail template
"""
self.ensure_one()
if self.env.context.get('mail_to'):
mail_users = self.env['res.users'].browse(
self.env.context.get('mail_to'))
if self.env.context.get("mail_to"):
mail_users = self.env["res.users"].browse(self.env.context.get("mail_to"))
else:
mail_users = self.mail_user_ids
return ','.join([x.email for x in mail_users if x.email])
return ",".join([x.email for x in mail_users if x.email])

17
sql_export_mail/tests/test_sql_query_mail.py
View File

@@ -2,28 +2,27 @@
# @author: Florian da Costa
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo.tests.common import TransactionCase
from odoo import SUPERUSER_ID
from odoo.tests.common import TransactionCase
class TestExportSqlQueryMail(TransactionCase):
def setUp(self):
super(TestExportSqlQueryMail, self).setUp()
self.sql_report_demo = self.env.ref('sql_export.sql_export_partner')
self.sql_report_demo.write({'mail_user_ids': [(4, SUPERUSER_ID)]})
self.sql_report_demo = self.env.ref("sql_export.sql_export_partner")
self.sql_report_demo.write({"mail_user_ids": [(4, SUPERUSER_ID)]})
def test_sql_query_mail(self):
mail_obj = self.env['mail.mail']
mail_obj = self.env["mail.mail"]
mails = mail_obj.search(
[('model', '=', 'sql.export'),
('res_id', '=', self.sql_report_demo.id)])
[("model", "=", "sql.export"), ("res_id", "=", self.sql_report_demo.id)]
)
self.assertFalse(mails)
self.sql_report_demo.create_cron()
self.assertTrue(self.sql_report_demo.cron_ids)
self.sql_report_demo.cron_ids.method_direct_trigger()
mails = mail_obj.search(
[('model', '=', 'sql.export'),
('res_id', '=', self.sql_report_demo.id)])
[("model", "=", "sql.export"), ("res_id", "=", self.sql_report_demo.id)]
)
self.assertTrue(mails)
self.assertTrue(mails.attachment_ids)

25
sql_export_mail/views/sql_export_view.xml
View File

@@ -1,4 +1,4 @@
<?xml version="1.0" encoding="utf-8"?>
<?xml version="1.0" encoding="utf-8" ?>
<odoo>
@@ -7,19 +7,32 @@
<field name="inherit_id" ref="sql_export.sql_export_view_form" />
<field name="arch" type="xml">
<button name="export_sql_query" position="after">
<button name="create_cron" string="Create Cron" type="object" groups="base.group_system"
attrs="{'invisible': ['|', ('state', '=', 'draft'), ('mail_user_ids', '=', [(6, False, [])])]}"/>
<button
name="create_cron"
string="Create Cron"
type="object"
groups="base.group_system"
attrs="{'invisible': ['|', ('state', '=', 'draft'), ('mail_user_ids', '=', [(6, False, [])])]}"
/>
</button>
<field name="copy_options" position="after">
<field name="mail_condition" attrs="{'invisible': [('cron_ids', '=', [(6, False, [])])]}"/>
<field
name="mail_condition"
attrs="{'invisible': [('cron_ids', '=', [(6, False, [])])]}"
/>
</field>
<group name="parameters" position="after">
<group groups="sql_request_abstract.group_sql_request_user">
<group string="Users Notified by e-mail">
<field name="mail_user_ids" nolabel="1"/>
<field name="mail_user_ids" nolabel="1" />
</group>
<group string="Crons" groups="base.group_system">
<field name="cron_ids" nolabel="1" domain="[('function', '=', '_run_all_sql_export_for_cron')]" groups="base.group_system"/>
<field
name="cron_ids"
nolabel="1"
domain="[('function', '=', '_run_all_sql_export_for_cron')]"
groups="base.group_system"
/>
</group>
</group>
</group>