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
						
					
					
				
			
		
		
		
			
			
			
		
		
	
	
							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()
							 | 
						|
								
							 |