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.
 
 
 
 
 

194 lines
9.3 KiB

# -*- coding: utf-8 -*-
"""Wizard for pdf and xlsx reports"""
################################################################################
#
# Cybrosys Technologies Pvt. Ltd.
# Copyright (C) 2017-TODAY Cybrosys Technologies(<https://www.cybrosys.com>).
# Author: Avinash Nk(<avinash@cybrosys.in>)
#
# This program is free software: you can modify
# it under the terms of the GNU Affero General Public License (AGPL) as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
#
################################################################################
import json
import pytz
from odoo import fields, models
from odoo.exceptions import ValidationError
from odoo.tools import date_utils, io, xlsxwriter
class HotelManagementWizard(models.TransientModel):
"""Class for wizard"""
_name = 'event.management.wizard'
_description = 'Event Management Wizard'
date_from = fields.Date(string="From")
date_to = fields.Date(string="To")
partner_id = fields.Many2one('res.partner', string='Customer')
type_event_ids = fields.Many2many('event.management.type', 'event_type_rel',
'report_id', 'type_id', string="Type")
event_state = fields.Selection(
[('draft', 'Draft'), ('confirm', 'Confirmed'), ('invoice', 'Invoiced'),
('close', 'Close'), ('cancel', 'Canceled')], string="State")
def print_pdf_report(self):
"""Method for printing pdf report"""
type_select = self.type_event_ids.ids
data = {
'model': 'event.management.wizard',
'form': self.read()[0],
'event_types': type_select
}
return self.env.ref(
'event_management.action_event_management_report').report_action(
self, data=data)
def print_xls_report(self):
"""Method of button for printing xlsx report"""
rec = self.env.user.sudo().company_id
if self.date_from and self.date_to and self.date_from > self.date_to:
raise ValidationError('From Date must be less than To Date')
user_tz = self.env.user.tz
current = fields.datetime.now()
current = pytz.UTC.localize(current)
current = current.astimezone(pytz.timezone(user_tz))
data = {
'event_type': self.type_event_ids.ids,
'date_from': self.date_from,
'date_to': self.date_to,
'customer': self.partner_id.id,
'state': self.event_state,
'today_date': current,
'company': [rec.partner_id.name, rec.street, rec.favicon]
}
return {
'type': 'ir.actions.report',
'report_type': 'xlsx',
'data': {'model': 'event.management.wizard',
'output_format': 'xlsx',
'options': json.dumps(data,
default=date_utils.json_default),
'report_name': 'Event Management Report', },
}
def get_xlsx_report(self, data, response):
"""Method for fetching data and printing xlsx report from controller"""
output = io.BytesIO()
workbook = xlsxwriter.Workbook(output, {'in_memory': True})
sheet = workbook.add_worksheet()
head = workbook.add_format({'align': 'center',
'bold': True, 'font_size': '20px'})
cell_head_format = workbook.add_format({'font_size': '12px'})
cell_data_format = workbook.add_format({'font_size': '10px'})
txt_head = workbook.add_format({'font_size': '10px', 'border': 2})
txt = workbook.add_format({'font_size': '10px', 'border': 2})
sheet.merge_range('F2:M3', 'EVENT MANAGEMENT REPORT', head)
sheet.merge_range('B4:E4', data['company'][0], cell_head_format)
sheet.merge_range('B5:E5', data['company'][1], cell_head_format)
sheet.write('B6', 'Date:', cell_head_format)
sheet.merge_range('C6:E6', data['today_date'], cell_data_format)
if data['date_from'] and data['date_to'] and data['customer']:
customer_name = self.env['res.partner'].browse(int(
data['customer'])).name
sheet.write('B8', 'From:', cell_head_format)
sheet.merge_range('C8:D8', data['date_from'], cell_data_format)
sheet.write('F8', 'To:', cell_head_format)
sheet.merge_range('G8:H8', data['date_to'], cell_data_format)
sheet.merge_range('J8:K8', 'Customer:', cell_head_format)
sheet.merge_range('L8:N8', customer_name, cell_data_format)
elif data['date_from'] and data['date_to']:
sheet.write('B8', 'From:', cell_head_format)
sheet.merge_range('C8:D8', data['date_from'],
cell_data_format)
sheet.write('F8', 'To:', cell_head_format)
sheet.merge_range('G8:H8', data['date_to'], cell_data_format)
elif data['date_from'] and data['customer']:
customer_name = self.env['res.partner'].browse(int(
data['customer'])).name
sheet.write('B8', 'From:', cell_head_format)
sheet.merge_range('C8:D8', data['date_from'], cell_data_format)
sheet.merge_range('F8:G8', 'Customer:', cell_head_format)
sheet.merge_range('H8:J8', customer_name, cell_data_format)
elif data['customer'] and data['date_to']:
customer_name = self.env['res.partner'].browse(int(
data['customer'])).name
sheet.write('B8', 'To:', cell_head_format)
sheet.merge_range('C8:D8', data['date_to'], cell_data_format)
sheet.merge_range('F8:G8', 'Customer:', cell_head_format)
sheet.merge_range('H8:J8', customer_name, cell_data_format)
elif data['date_from']:
sheet.write('B8', 'From:', cell_head_format)
sheet.merge_range('C8:D8', data['date_from'], cell_data_format)
elif data['date_to']:
sheet.write('B8', 'To:', cell_head_format)
sheet.merge_range('C8:D8', data['date_to'], cell_data_format)
elif data['customer']:
customer_name = self.env['res.partner'].browse(int(
data['customer'])).name
sheet.merge_range('B8:C8', 'Customer:', cell_head_format)
sheet.merge_range('D8:E8', customer_name, cell_data_format)
sheet.write(10, 0, 'Sl.no', txt_head)
sheet.merge_range('B11:E11', 'Name', txt_head)
sheet.merge_range('F11:H11', 'Type', txt_head)
sheet.merge_range('I11:K11', 'Customer', txt_head)
sheet.merge_range('L11:M11', 'Date', txt_head)
sheet.merge_range('N11:O11', 'Start Date', txt_head)
sheet.merge_range('P11:Q11', 'End Date', txt_head)
sheet.write(10, 17, 'State', txt_head)
where = '1=1'
if data["customer"]:
where += """AND e.partner_id = %s""" % int(data['customer'])
if data['date_from']:
where += """AND e.date>='%s'""" % (data['date_from'])
if data['date_to']:
where += """AND e.date <= '%s'""" % (data['date_to'])
if data['event_type']:
event_list = data['event_type']
event_ids = f"({event_list[0]})" if len(
event_list) == 1 else tuple(event_list)
where += """AND e.type_of_event_id IN {}""".format(event_ids)
if data['state']:
where += """AND e.state = '%s'""" % (data['state'])
self.env.cr.execute("""
SELECT e.name as event, t.name as type, r.name as partner,
e.state, e.date,
e.start_date, e.end_date
from event_management e inner join
res_partner r on e.partner_id = r.id
inner join event_management_type t on
e.type_of_event_id = t.id
where %s order by e.date""" % where)
rec = self.env.cr.fetchall()
j = 11
k = 1
for i in range(0, len(rec)):
sheet.write(j, 0, k, txt)
sheet.merge_range('B%d:E%d' % (j + 1, j + 1), rec[i][0], txt)
sheet.merge_range('F%d:H%d' % (j + 1, j + 1), rec[i][1], txt)
sheet.merge_range('I%d:K%d' % (j + 1, j + 1), rec[i][2], txt)
sheet.merge_range('L%d:M%d' % (j + 1, j + 1),
fields.Date.to_string(rec[i][4]), txt)
sheet.merge_range('N%d:O%d' % (j + 1, j + 1),
fields.Datetime.to_string(rec[i][5]), txt)
sheet.merge_range('P%d:Q%d' % (j + 1, j + 1),
fields.Datetime.to_string(rec[i][6]), txt)
sheet.write(j, 17,
dict(self.env['event.management']._fields[
'state'].selection).get(rec[i][3]), txt)
j += 1
k += 1
workbook.close()
output.seek(0)
response.stream.write(output.read())
output.close()