[MERGE] fix partial payment wizard
[odoo/odoo.git] / addons / point_of_sale / wizard / pos_payment.py
1 # -*- coding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
6 #
7 #    This program is free software: you can redistribute it and/or modify
8 #    it under the terms of the GNU Affero General Public License as
9 #    published by the Free Software Foundation, either version 3 of the
10 #    License, or (at your option) any later version.
11 #
12 #    This program is distributed in the hope that it will be useful,
13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 #    GNU Affero General Public License for more details.
16 #
17 #    You should have received a copy of the GNU Affero General Public License
18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 #
20 ##############################################################################
21
22 import time
23
24 from osv import osv, fields
25 from tools.translate import _
26 import pos_box_entries
27 import netsvc
28
29 class account_journal(osv.osv):
30     _inherit = 'account.journal'
31
32     def search(self, cr, uid, args, offset=0, limit=None, order=None, context=None, count=False):
33         if not context:
34             context = {}
35         session_id = context.get('pos_session_id', False) or False
36
37         if session_id:
38             session = self.pool.get('pos.session').browse(cr, uid, session_id, context=context)
39
40             if session:
41                 journal_ids = [journal.id for journal in session.config_id.journal_ids]
42                 args += [('id', 'in', journal_ids)]
43
44         return super(account_journal, self).search(cr, uid, args, offset=offset, limit=limit, order=order, context=context, count=count)
45
46 class pos_make_payment(osv.osv_memory):
47     _name = 'pos.make.payment'
48     _description = 'Point of Sale Payment'
49     def check(self, cr, uid, ids, context=None):
50         """Check the order:
51         if the order is not paid: continue payment,
52         if the order is paid print ticket.
53         """
54         context = context or {}
55         order_obj = self.pool.get('pos.order')
56         obj_partner = self.pool.get('res.partner')
57         active_id = context and context.get('active_id', False)
58
59         order = order_obj.browse(cr, uid, active_id, context=context)
60         amount = order.amount_total - order.amount_paid
61         data = self.read(cr, uid, ids, context=context)[0]
62         # this is probably a problem of osv_memory as it's not compatible with normal OSV's
63         data['journal'] = data['journal_id'][0]
64
65         if amount != 0.0:
66             order_obj.add_payment(cr, uid, active_id, data, context=context)
67
68         if order_obj.test_paid(cr, uid, [active_id]):
69             wf_service = netsvc.LocalService("workflow")
70             wf_service.trg_validate(uid, 'pos.order', active_id, 'paid', cr)
71             return {'type' : 'ir.actions.act_window_close' }
72          ##self.print_report(cr, uid, ids, context=context)
73
74         return self.launch_payment(cr, uid, ids, context=context)
75
76     def launch_payment(self, cr, uid, ids, context=None):
77         return {
78             'name': _('Payment'),
79             'view_type': 'form',
80             'view_mode': 'form',
81             'res_model': 'pos.make.payment',
82             'view_id': False,
83             'target': 'new',
84             'views': False,
85             'type': 'ir.actions.act_window',
86             'context': context,
87         }
88
89     def print_report(self, cr, uid, ids, context=None):
90         active_id = context.get('active_id', [])
91         datas = {'ids' : [active_id]}
92         return {
93             'type': 'ir.actions.report.xml',
94             'report_name': 'pos.receipt',
95             'datas': datas,
96         }
97
98     def _default_journal(self, cr, uid, context=None):
99         if not context:
100             context = {}
101         session = False
102         order_obj = self.pool.get('pos.order')
103         active_id = context and context.get('active_id', False)
104         if active_id:
105             order = order_obj.browse(cr, uid, active_id, context=context)
106             session = order.session_id
107         if session:
108             for journal in session.config_id.journal_ids:
109                 return journal.id
110         return False
111
112     def _default_amount(self, cr, uid, context=None):
113         order_obj = self.pool.get('pos.order')
114         active_id = context and context.get('active_id', False)
115         if active_id:
116             order = order_obj.browse(cr, uid, active_id, context=context)
117             return order.amount_total - order.amount_paid
118         return False
119
120     _columns = {
121         'journal_id' : fields.many2one('account.journal', 'Payment Mode', required=True),
122         'amount': fields.float('Amount', digits=(16,2), required= True),
123         'payment_name': fields.char('Payment Reference', size=32),
124         'payment_date': fields.date('Payment Date', required=True),
125     }
126     _defaults = {
127         'journal_id' : _default_journal,
128         'payment_date': time.strftime('%Y-%m-%d %H:%M:%S'),
129         'amount': _default_amount,
130     }
131
132 pos_make_payment()
133
134
135 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: