odoo/addons/account_followup/wizard/wizard_followup_print.py

##############################################################################
#
# Copyright (c) 2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

import wizard
import datetime
import pooler
import time

_followup_wizard_all_form = """<?xml version="1.0"?>
<form string="Select partners" colspan="4">
	<notebook>
		<page string="FollowUp selection">
			<separator string="Select partners to remind" colspan="4"/>
			<field name="partner_ids" colspan="4" nolabel="1"/>
		</page>
		<page string="Email confirmation">
			<label string="Not yet implemented !" colspan="4"/>
			<field name="email_conf" colspan="4"/>
			<separator string="Email body" colspan="4"/>
			<field name="email_body" colspan="4" nolabel="1"/>
			<separator string="Legend" colspan="4"/>
	
			<label string="%(partner_name)s: Partner name" colspan="2"/>
			<label string="%(user_signature)s: Partner name" colspan="2"/>
			<label string="%(followup_level)s: Followup level" colspan="2"/>
			<label string="%(followup_amount)s: Followup level" colspan="2"/>
			<label string="%(followup_details)s: Followup level" colspan="2"/>
		</page>
	</notebook>
</form>"""

_followup_wizard_all_fields = {
	'partner_ids': {
		'string': "Partners", 
		'type': 'many2many', 
		'relation': 'account_followup.stat',
	},
	'email_conf': {
		'string': "Send email confirmation", 
		'type': 'boolean', 
	},
	'email_body': {
		'string': "Email body", 
		'type': 'text', 
		'default': '''
Dear %(partner_name)s,

Please find in attachment our reminder for a total amount due
of %(followup_amount).2f EUR.

The details are here:
%(followup_details)s

%(followup_level)s
Thanks,

-- 
%(user_signature)s
		'''
	}
}


class followup_all_print(wizard.interface):
	def _update_partners(self, cr, uid, data, context):
		to_update = data['form']['to_update']
		for id in to_update.keys():
			cr.execute(
				"UPDATE account_move_line "\
				"SET followup_line_id=%d, followup_date=%s "\
				"WHERE id=%d",
				(to_update[id],
				time.strftime('%Y-%m-%d'), int(id),))
		return {}

	def _get_partners(self, cr, uid, data, context):
		pool = pooler.get_pool(cr.dbname)
		cr.execute(
			"SELECT l.partner_id, l.followup_line_id, l.date, l.id "\
			"FROM account_move_line AS l "\
				"LEFT JOIN account_account AS a "\
				"ON (l.account_id=a.id) "\
			"WHERE (l.reconcile_id IS NULL) "\
				"AND (a.type='receivable') "\
				"AND (l.state<>'draft') "\
				"AND (l.reconcile_id is NULL) "\
				"AND partner_id is NOT NULL "\
				"AND a.active "\
			"ORDER BY l.date")
		move_lines = cr.fetchall()

		old = None
		fups = {}
		fup_ids = pool.get('account_followup.followup').search(cr, uid, [])
		if not fup_ids:
			raise wizard.except_wizard('No Follow up Defined', 
				'You must define at least one follow up for your company !')
		fup_id = fup_ids[0]

		current_date = datetime.date.today()
		cr.execute(
			"SELECT * "\
			"FROM account_followup_followup_line "\
			"WHERE followup_id=%d "\
			"ORDER BY sequence", (fup_id,))
		for result in cr.dictfetchall():
			delay = datetime.timedelta(days=result['delay'])
			fups[old] = (current_date - delay, result['id'])
			if result['start'] == 'end_of_month':
				fups[old][0].replace(day=1)
			old = result['id']

		fups[old] = (datetime.date(datetime.MAXYEAR, 12, 31), old)

		partner_list = []
		to_update = {}
		for partner_id, followup_line_id, date, id in move_lines: 
			if (partner_id in partner_list) or (not partner_id):
				continue
			if followup_line_id not in fups:
				continue
			if date <= fups[followup_line_id][0].strftime('%Y-%m-%d'):
				partner_list.append(partner_id)
				to_update[str(id)] = fups[followup_line_id][1]
		return {'partner_ids': partner_list, 'to_update': to_update}

	states = {
		'init' : {
			'actions': [_get_partners],
			'result': {'type': 'form',
				'arch':_followup_wizard_all_form,
				'fields':_followup_wizard_all_fields,
				'state':[('end','Cancel'),('print','Print Follow Ups')]
			},
		},
		'print': {
			'actions': [_update_partners],
			'result': {'type': 'print',
				'report':'account_followup.followup.print',
				'state':'end'},
		},
	}

followup_all_print('account_followup.followup.print.all')