[MERGE] Merge with lp:openobject-addons
[odoo/odoo.git] / addons / point_of_sale / wizard / pos_add_product.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 from osv import osv, fields
23 from tools.translate import _
24
25
26 class add_product(osv.osv_memory):
27     _name = 'pos.add.product'
28     _description = 'Add Product'
29
30     _columns = {
31         'product_id': fields.many2one('product.product', 'Product', required=True),
32         'quantity': fields.float('Quantity', required=True),
33     }
34     _defaults = {
35         'quantity': 1,
36     }
37
38     def select_product(self, cr, uid, ids, context=None):
39         """
40              To get the product and quantity and add in order .
41              @param self: The object pointer.
42              @param cr: A database cursor
43              @param uid: ID of the user currently logged in
44              @param context: A standard dictionary
45              @return : Return the add product form again for adding more product
46         """
47         if context is None:
48             context = {}
49         this = self.browse(cr, uid, ids[0], context=context)
50         record_id = context and context.get('active_id', False)
51         assert record_id, _('Active ID is not found')
52         if record_id:
53             order_obj = self.pool.get('pos.order')
54             order_obj.add_product(cr, uid, record_id, this.product_id.id, this.quantity, context=context)
55         return {
56             'name': _('Add Product'),
57             'view_type': 'form',
58             'view_mode': 'form',
59             'res_model': 'pos.add.product',
60             'view_id': False,
61             'target': 'new',
62             'views': False,
63             'type': 'ir.actions.act_window',
64         }
65
66     def close_action(self, cr, uid, ids, context=None):
67         """
68              To get the product and Make the payment .
69              @param self: The object pointer.
70              @param cr: A database cursor
71              @param uid: ID of the user currently logged in
72              @param context: A standard dictionary
73              @return : Return the Make Payment
74         """
75         if context is None:
76             context = {}
77         record_id = context and context.get('active_id', False)
78         order_obj= self.pool.get('pos.order')
79         this = self.browse(cr, uid, ids[0], context)
80         order_obj.add_product(cr, uid, record_id, this.product_id.id, this.quantity, context=context)
81
82         order_obj.write(cr, uid, [record_id], {'state': 'done'}, context=context)
83         return {
84             'name': _('Make Payment'),
85             'context': context and context.get('active_id', False),
86             'view_type': 'form',
87             'view_mode': 'form',
88             'res_model': 'pos.make.payment',
89             'view_id': False,
90             'target': 'new',
91             'views': False,
92             'type': 'ir.actions.act_window',
93         }
94
95 add_product()
96
97 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
98