[IMP] addons: use the magic methods signal_XXX() instead of signal_workflow('XXX')
[odoo/odoo.git] / addons / point_of_sale / point_of_sale.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 logging
23 import time
24
25 from openerp import tools
26 from openerp.osv import fields, osv
27 from openerp.tools.translate import _
28
29 import openerp.addons.decimal_precision as dp
30 import openerp.addons.product.product
31
32 _logger = logging.getLogger(__name__)
33
34 class pos_config(osv.osv):
35     _name = 'pos.config'
36
37     POS_CONFIG_STATE = [
38         ('active', 'Active'),
39         ('inactive', 'Inactive'),
40         ('deprecated', 'Deprecated')
41     ]
42
43     def _get_currency(self, cr, uid, ids, fieldnames, args, context=None):
44         result = dict.fromkeys(ids, False)
45         for pos_config in self.browse(cr, uid, ids, context=context):
46             if pos_config.journal_id:
47                 currency_id = pos_config.journal_id.currency.id or pos_config.journal_id.company_id.currency_id.id
48             else:
49                 currency_id = self.pool['res.users'].browse(cr, uid, uid, context=context).company_id.currency_id.id
50             result[pos_config.id] = currency_id
51         return result
52
53     _columns = {
54         'name' : fields.char('Point of Sale Name', select=1,
55              required=True, help="An internal identification of the point of sale"),
56         'journal_ids' : fields.many2many('account.journal', 'pos_config_journal_rel', 
57              'pos_config_id', 'journal_id', 'Available Payment Methods',
58              domain="[('journal_user', '=', True ), ('type', 'in', ['bank', 'cash'])]",),
59         'picking_type_id': fields.many2one('stock.picking.type', 'Picking Type'),
60         'stock_location_id': fields.many2one('stock.location', 'Stock Location', domain=[('usage', '=', 'internal')], required=True),
61         'journal_id' : fields.many2one('account.journal', 'Sale Journal',
62              domain=[('type', '=', 'sale')],
63              help="Accounting journal used to post sales entries."),
64         'currency_id' : fields.function(_get_currency, type="many2one", string="Currency", relation="res.currency"),
65         'iface_self_checkout' : fields.boolean('Self Checkout Mode',
66              help="Check this if this point of sale should open by default in a self checkout mode. If unchecked, OpenERP uses the normal cashier mode by default."),
67         'iface_cashdrawer' : fields.boolean('Cashdrawer', help="Automatically open the cashdrawer"),
68         'iface_payment_terminal' : fields.boolean('Payment Terminal', help="Enables Payment Terminal integration"),
69         'iface_electronic_scale' : fields.boolean('Electronic Scale', help="Enables Electronic Scale integration"),
70         'iface_vkeyboard' : fields.boolean('Virtual KeyBoard', help="Enables an integrated Virtual Keyboard"),
71         'iface_print_via_proxy' : fields.boolean('Print via Proxy', help="Bypass browser printing and prints via the hardware proxy"),
72         'iface_scan_via_proxy' : fields.boolean('Scan via Proxy', help="Enable barcode scanning with a remotely connected barcode scanner"),
73         'iface_invoicing': fields.boolean('Invoicing',help='Enables invoice generation from the Point of Sale'),
74         'iface_big_scrollbars': fields.boolean('Large Scrollbars',help='For imprecise industrial touchscreens'),
75         'receipt_header': fields.text('Receipt Header',help="A short text that will be inserted as a header in the printed receipt"),
76         'receipt_footer': fields.text('Receipt Footer',help="A short text that will be inserted as a footer in the printed receipt"),
77         'proxy_ip':       fields.char('IP Address', help='The hostname or ip address of the hardware proxy, Will be autodetected if left empty', size=45),
78
79         'state' : fields.selection(POS_CONFIG_STATE, 'Status', required=True, readonly=True, copy=False),
80         'sequence_id' : fields.many2one('ir.sequence', 'Order IDs Sequence', readonly=True,
81             help="This sequence is automatically created by OpenERP but you can change it "\
82                 "to customize the reference numbers of your orders.", copy=False),
83         'session_ids': fields.one2many('pos.session', 'config_id', 'Sessions'),
84         'group_by' : fields.boolean('Group Journal Items', help="Check this if you want to group the Journal Items by Product while closing a Session"),
85         'pricelist_id': fields.many2one('product.pricelist','Pricelist', required=True),
86         'company_id': fields.many2one('res.company', 'Company', required=True),
87         'barcode_product':  fields.char('Product Barcodes', size=64, help='The pattern that identifies product barcodes'),
88         'barcode_cashier':  fields.char('Cashier Barcodes', size=64, help='The pattern that identifies cashier login barcodes'),
89         'barcode_customer': fields.char('Customer Barcodes',size=64, help='The pattern that identifies customer\'s client card barcodes'),
90         'barcode_price':    fields.char('Price Barcodes',   size=64, help='The pattern that identifies a product with a barcode encoded price'),
91         'barcode_weight':   fields.char('Weight Barcodes',  size=64, help='The pattern that identifies a product with a barcode encoded weight'),
92         'barcode_discount': fields.char('Discount Barcodes',  size=64, help='The pattern that identifies a product with a barcode encoded discount'),
93     }
94
95     def _check_cash_control(self, cr, uid, ids, context=None):
96         return all(
97             (sum(int(journal.cash_control) for journal in record.journal_ids) <= 1)
98             for record in self.browse(cr, uid, ids, context=context)
99         )
100
101     _constraints = [
102         (_check_cash_control, "You cannot have two cash controls in one Point Of Sale !", ['journal_ids']),
103     ]
104
105     def name_get(self, cr, uid, ids, context=None):
106         result = []
107         states = {
108             'opening_control': _('Opening Control'),
109             'opened': _('In Progress'),
110             'closing_control': _('Closing Control'),
111             'closed': _('Closed & Posted'),
112         }
113         for record in self.browse(cr, uid, ids, context=context):
114             if (not record.session_ids) or (record.session_ids[0].state=='closed'):
115                 result.append((record.id, record.name+' ('+_('not used')+')'))
116                 continue
117             session = record.session_ids[0]
118             result.append((record.id, record.name + ' ('+session.user_id.name+')')) #, '+states[session.state]+')'))
119         return result
120
121     def _default_sale_journal(self, cr, uid, context=None):
122         company_id = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.id
123         res = self.pool.get('account.journal').search(cr, uid, [('type', '=', 'sale'), ('company_id', '=', company_id)], limit=1, context=context)
124         return res and res[0] or False
125
126     def _default_pricelist(self, cr, uid, context=None):
127         res = self.pool.get('product.pricelist').search(cr, uid, [('type', '=', 'sale')], limit=1, context=context)
128         return res and res[0] or False
129
130     def _get_default_location(self, cr, uid, context=None):
131         wh_obj = self.pool.get('stock.warehouse')
132         user = self.pool.get('res.users').browse(cr, uid, uid, context)
133         res = wh_obj.search(cr, uid, [('company_id', '=', user.company_id.id)], limit=1, context=context)
134         if res and res[0]:
135             return wh_obj.browse(cr, uid, res[0], context=context).lot_stock_id.id
136         return False
137
138     def _get_default_company(self, cr, uid, context=None):
139         company_id = self.pool.get('res.users')._get_company(cr, uid, context=context)
140         return company_id
141
142     _defaults = {
143         'state' : POS_CONFIG_STATE[0][0],
144         'journal_id': _default_sale_journal,
145         'group_by' : True,
146         'pricelist_id': _default_pricelist,
147         'iface_invoicing': True,
148         'stock_location_id': _get_default_location,
149         'company_id': _get_default_company,
150         'barcode_product': '*', 
151         'barcode_cashier': '041*', 
152         'barcode_customer':'042*', 
153         'barcode_weight':  '21xxxxxNNDDD', 
154         'barcode_discount':'22xxxxxxxxNN', 
155         'barcode_price':   '23xxxxxNNNDD', 
156     }
157
158     def onchange_picking_type_id(self, cr, uid, ids, picking_type_id, context=None):
159         p_type_obj = self.pool.get("stock.picking.type")
160         p_type = p_type_obj.browse(cr, uid, picking_type_id, context=context)
161         if p_type.default_location_src_id and p_type.default_location_src_id.usage == 'internal' and p_type.default_location_dest_id and p_type.default_location_dest_id.usage == 'customer':
162             return {'value': {'stock_location_id': p_type.default_location_src_id.id}}
163         return False
164
165     def set_active(self, cr, uid, ids, context=None):
166         return self.write(cr, uid, ids, {'state' : 'active'}, context=context)
167
168     def set_inactive(self, cr, uid, ids, context=None):
169         return self.write(cr, uid, ids, {'state' : 'inactive'}, context=context)
170
171     def set_deprecate(self, cr, uid, ids, context=None):
172         return self.write(cr, uid, ids, {'state' : 'deprecated'}, context=context)
173
174     def create(self, cr, uid, values, context=None):
175         proxy = self.pool.get('ir.sequence')
176         sequence_values = dict(
177             name='PoS %s' % values['name'],
178             padding=5,
179             prefix="%s/"  % values['name'],
180         )
181         sequence_id = proxy.create(cr, uid, sequence_values, context=context)
182         values['sequence_id'] = sequence_id
183         return super(pos_config, self).create(cr, uid, values, context=context)
184
185     def unlink(self, cr, uid, ids, context=None):
186         for obj in self.browse(cr, uid, ids, context=context):
187             if obj.sequence_id:
188                 obj.sequence_id.unlink()
189         return super(pos_config, self).unlink(cr, uid, ids, context=context)
190
191 class pos_session(osv.osv):
192     _name = 'pos.session'
193     _order = 'id desc'
194
195     POS_SESSION_STATE = [
196         ('opening_control', 'Opening Control'),  # Signal open
197         ('opened', 'In Progress'),                    # Signal closing
198         ('closing_control', 'Closing Control'),  # Signal close
199         ('closed', 'Closed & Posted'),
200     ]
201
202     def _compute_cash_all(self, cr, uid, ids, fieldnames, args, context=None):
203         result = dict()
204
205         for record in self.browse(cr, uid, ids, context=context):
206             result[record.id] = {
207                 'cash_journal_id' : False,
208                 'cash_register_id' : False,
209                 'cash_control' : False,
210             }
211             for st in record.statement_ids:
212                 if st.journal_id.cash_control == True:
213                     result[record.id]['cash_control'] = True
214                     result[record.id]['cash_journal_id'] = st.journal_id.id
215                     result[record.id]['cash_register_id'] = st.id
216
217         return result
218
219     _columns = {
220         'config_id' : fields.many2one('pos.config', 'Point of Sale',
221                                       help="The physical point of sale you will use.",
222                                       required=True,
223                                       select=1,
224                                       domain="[('state', '=', 'active')]",
225                                      ),
226
227         'name' : fields.char('Session ID', required=True, readonly=True),
228         'user_id' : fields.many2one('res.users', 'Responsible',
229                                     required=True,
230                                     select=1,
231                                     readonly=True,
232                                     states={'opening_control' : [('readonly', False)]}
233                                    ),
234         'currency_id' : fields.related('config_id', 'currency_id', type="many2one", relation='res.currency', string="Currnecy"),
235         'start_at' : fields.datetime('Opening Date', readonly=True), 
236         'stop_at' : fields.datetime('Closing Date', readonly=True),
237
238         'state' : fields.selection(POS_SESSION_STATE, 'Status',
239                 required=True, readonly=True,
240                 select=1, copy=False),
241         
242         'sequence_number': fields.integer('Order Sequence Number'),
243
244         'cash_control' : fields.function(_compute_cash_all,
245                                          multi='cash',
246                                          type='boolean', string='Has Cash Control'),
247         'cash_journal_id' : fields.function(_compute_cash_all,
248                                             multi='cash',
249                                             type='many2one', relation='account.journal',
250                                             string='Cash Journal', store=True),
251         'cash_register_id' : fields.function(_compute_cash_all,
252                                              multi='cash',
253                                              type='many2one', relation='account.bank.statement',
254                                              string='Cash Register', store=True),
255
256         'opening_details_ids' : fields.related('cash_register_id', 'opening_details_ids', 
257                 type='one2many', relation='account.cashbox.line',
258                 string='Opening Cash Control'),
259         'details_ids' : fields.related('cash_register_id', 'details_ids', 
260                 type='one2many', relation='account.cashbox.line',
261                 string='Cash Control'),
262
263         'cash_register_balance_end_real' : fields.related('cash_register_id', 'balance_end_real',
264                 type='float',
265                 digits_compute=dp.get_precision('Account'),
266                 string="Ending Balance",
267                 help="Total of closing cash control lines.",
268                 readonly=True),
269         'cash_register_balance_start' : fields.related('cash_register_id', 'balance_start',
270                 type='float',
271                 digits_compute=dp.get_precision('Account'),
272                 string="Starting Balance",
273                 help="Total of opening cash control lines.",
274                 readonly=True),
275         'cash_register_total_entry_encoding' : fields.related('cash_register_id', 'total_entry_encoding',
276                 string='Total Cash Transaction',
277                 readonly=True,
278                 help="Total of all paid sale orders"),
279         'cash_register_balance_end' : fields.related('cash_register_id', 'balance_end',
280                 type='float',
281                 digits_compute=dp.get_precision('Account'),
282                 string="Theoretical Closing Balance",
283                 help="Sum of opening balance and transactions.",
284                 readonly=True),
285         'cash_register_difference' : fields.related('cash_register_id', 'difference',
286                 type='float',
287                 string='Difference',
288                 help="Difference between the theoretical closing balance and the real closing balance.",
289                 readonly=True),
290
291         'journal_ids' : fields.related('config_id', 'journal_ids',
292                                        type='many2many',
293                                        readonly=True,
294                                        relation='account.journal',
295                                        string='Available Payment Methods'),
296         'order_ids' : fields.one2many('pos.order', 'session_id', 'Orders'),
297
298         'statement_ids' : fields.one2many('account.bank.statement', 'pos_session_id', 'Bank Statement', readonly=True),
299     }
300
301     _defaults = {
302         'name' : '/',
303         'user_id' : lambda obj, cr, uid, context: uid,
304         'state' : 'opening_control',
305         'sequence_number': 1,
306     }
307
308     _sql_constraints = [
309         ('uniq_name', 'unique(name)', "The name of this POS Session must be unique !"),
310     ]
311
312     def _check_unicity(self, cr, uid, ids, context=None):
313         for session in self.browse(cr, uid, ids, context=None):
314             # open if there is no session in 'opening_control', 'opened', 'closing_control' for one user
315             domain = [
316                 ('state', 'not in', ('closed','closing_control')),
317                 ('user_id', '=', session.user_id.id)
318             ]
319             count = self.search_count(cr, uid, domain, context=context)
320             if count>1:
321                 return False
322         return True
323
324     def _check_pos_config(self, cr, uid, ids, context=None):
325         for session in self.browse(cr, uid, ids, context=None):
326             domain = [
327                 ('state', '!=', 'closed'),
328                 ('config_id', '=', session.config_id.id)
329             ]
330             count = self.search_count(cr, uid, domain, context=context)
331             if count>1:
332                 return False
333         return True
334
335     _constraints = [
336         (_check_unicity, "You cannot create two active sessions with the same responsible!", ['user_id', 'state']),
337         (_check_pos_config, "You cannot create two active sessions related to the same point of sale!", ['config_id']),
338     ]
339
340     def create(self, cr, uid, values, context=None):
341         context = dict(context or {})
342         config_id = values.get('config_id', False) or context.get('default_config_id', False)
343         if not config_id:
344             raise osv.except_osv( _('Error!'),
345                 _("You should assign a Point of Sale to your session."))
346
347         # journal_id is not required on the pos_config because it does not
348         # exists at the installation. If nothing is configured at the
349         # installation we do the minimal configuration. Impossible to do in
350         # the .xml files as the CoA is not yet installed.
351         jobj = self.pool.get('pos.config')
352         pos_config = jobj.browse(cr, uid, config_id, context=context)
353         context.update({'company_id': pos_config.company_id.id})
354         if not pos_config.journal_id:
355             jid = jobj.default_get(cr, uid, ['journal_id'], context=context)['journal_id']
356             if jid:
357                 jobj.write(cr, uid, [pos_config.id], {'journal_id': jid}, context=context)
358             else:
359                 raise osv.except_osv( _('error!'),
360                     _("Unable to open the session. You have to assign a sale journal to your point of sale."))
361
362         # define some cash journal if no payment method exists
363         if not pos_config.journal_ids:
364             journal_proxy = self.pool.get('account.journal')
365             cashids = journal_proxy.search(cr, uid, [('journal_user', '=', True), ('type','=','cash')], context=context)
366             if not cashids:
367                 cashids = journal_proxy.search(cr, uid, [('type', '=', 'cash')], context=context)
368                 if not cashids:
369                     cashids = journal_proxy.search(cr, uid, [('journal_user','=',True)], context=context)
370
371             jobj.write(cr, uid, [pos_config.id], {'journal_ids': [(6,0, cashids)]})
372
373
374         pos_config = jobj.browse(cr, uid, config_id, context=context)
375         bank_statement_ids = []
376         for journal in pos_config.journal_ids:
377             bank_values = {
378                 'journal_id' : journal.id,
379                 'user_id' : uid,
380                 'company_id' : pos_config.company_id.id
381             }
382             statement_id = self.pool.get('account.bank.statement').create(cr, uid, bank_values, context=context)
383             bank_statement_ids.append(statement_id)
384
385         values.update({
386             'name' : pos_config.sequence_id._next(),
387             'statement_ids' : [(6, 0, bank_statement_ids)],
388             'config_id': config_id
389         })
390
391         return super(pos_session, self).create(cr, uid, values, context=context)
392
393     def unlink(self, cr, uid, ids, context=None):
394         for obj in self.browse(cr, uid, ids, context=context):
395             for statement in obj.statement_ids:
396                 statement.unlink(context=context)
397         return True
398
399
400     def open_cb(self, cr, uid, ids, context=None):
401         """
402         call the Point Of Sale interface and set the pos.session to 'opened' (in progress)
403         """
404         if context is None:
405             context = dict()
406
407         if isinstance(ids, (int, long)):
408             ids = [ids]
409
410         this_record = self.browse(cr, uid, ids[0], context=context)
411         this_record.signal_open()
412
413         context.update(active_id=this_record.id)
414
415         return {
416             'type' : 'ir.actions.act_url',
417             'url'  : '/pos/web/',
418             'target': 'self',
419         }
420
421     def wkf_action_open(self, cr, uid, ids, context=None):
422         # second browse because we need to refetch the data from the DB for cash_register_id
423         for record in self.browse(cr, uid, ids, context=context):
424             values = {}
425             if not record.start_at:
426                 values['start_at'] = time.strftime('%Y-%m-%d %H:%M:%S')
427             values['state'] = 'opened'
428             record.write(values)
429             for st in record.statement_ids:
430                 st.button_open()
431
432         return self.open_frontend_cb(cr, uid, ids, context=context)
433
434     def wkf_action_opening_control(self, cr, uid, ids, context=None):
435         return self.write(cr, uid, ids, {'state' : 'opening_control'}, context=context)
436
437     def wkf_action_closing_control(self, cr, uid, ids, context=None):
438         for session in self.browse(cr, uid, ids, context=context):
439             for statement in session.statement_ids:
440                 if (statement != session.cash_register_id) and (statement.balance_end != statement.balance_end_real):
441                     self.pool.get('account.bank.statement').write(cr, uid, [statement.id], {'balance_end_real': statement.balance_end})
442         return self.write(cr, uid, ids, {'state' : 'closing_control', 'stop_at' : time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
443
444     def wkf_action_close(self, cr, uid, ids, context=None):
445         # Close CashBox
446         bsl = self.pool.get('account.bank.statement.line')
447         for record in self.browse(cr, uid, ids, context=context):
448             for st in record.statement_ids:
449                 if abs(st.difference) > st.journal_id.amount_authorized_diff:
450                     # The pos manager can close statements with maximums.
451                     if not self.pool.get('ir.model.access').check_groups(cr, uid, "point_of_sale.group_pos_manager"):
452                         raise osv.except_osv( _('Error!'),
453                             _("Your ending balance is too different from the theoretical cash closing (%.2f), the maximum allowed is: %.2f. You can contact your manager to force it.") % (st.difference, st.journal_id.amount_authorized_diff))
454                 if (st.journal_id.type not in ['bank', 'cash']):
455                     raise osv.except_osv(_('Error!'), 
456                         _("The type of the journal for your payment method should be bank or cash "))
457                 if st.difference and st.journal_id.cash_control == True:
458                     if st.difference > 0.0:
459                         name= _('Point of Sale Profit')
460                     else:
461                         name= _('Point of Sale Loss')
462                     bsl.create(cr, uid, {
463                         'statement_id': st.id,
464                         'amount': st.difference,
465                         'ref': record.name,
466                         'name': name,
467                     }, context=context)
468
469                 if st.journal_id.type == 'bank':
470                     st.write({'balance_end_real' : st.balance_end})
471                 getattr(st, 'button_confirm_%s' % st.journal_id.type)(context=context)
472         self._confirm_orders(cr, uid, ids, context=context)
473         self.write(cr, uid, ids, {'state' : 'closed'}, context=context)
474
475         obj = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'point_of_sale', 'menu_point_root')[1]
476         return {
477             'type' : 'ir.actions.client',
478             'name' : 'Point of Sale Menu',
479             'tag' : 'reload',
480             'params' : {'menu_id': obj},
481         }
482
483     def _confirm_orders(self, cr, uid, ids, context=None):
484         account_move_obj = self.pool.get('account.move')
485         pos_order_obj = self.pool.get('pos.order')
486         for session in self.browse(cr, uid, ids, context=context):
487             local_context = dict(context or {}, force_company=session.config_id.journal_id.company_id.id)
488             order_ids = [order.id for order in session.order_ids if order.state == 'paid']
489
490             move_id = account_move_obj.create(cr, uid, {'ref' : session.name, 'journal_id' : session.config_id.journal_id.id, }, context=local_context)
491
492             pos_order_obj._create_account_move_line(cr, uid, order_ids, session, move_id, context=local_context)
493
494             for order in session.order_ids:
495                 if order.state not in ('paid', 'invoiced'):
496                     raise osv.except_osv(
497                         _('Error!'),
498                         _("You cannot confirm all orders of this session, because they have not the 'paid' status"))
499                 else:
500                     pos_order_obj.signal_done(cr, uid, [order.id])
501
502         return True
503
504     def open_frontend_cb(self, cr, uid, ids, context=None):
505         if not context:
506             context = {}
507         if not ids:
508             return {}
509         for session in self.browse(cr, uid, ids, context=context):
510             if session.user_id.id != uid:
511                 raise osv.except_osv(
512                         _('Error!'),
513                         _("You cannot use the session of another users. This session is owned by %s. Please first close this one to use this point of sale." % session.user_id.name))
514         context.update({'active_id': ids[0]})
515         return {
516             'type' : 'ir.actions.act_url',
517             'target': 'self',
518             'url':   '/pos/web/',
519         }
520
521 class pos_order(osv.osv):
522     _name = "pos.order"
523     _description = "Point of Sale"
524     _order = "id desc"
525
526     def _order_fields(self, cr, uid, ui_order, context=None):
527         return {
528             'name':         ui_order['name'],
529             'user_id':      ui_order['user_id'] or False,
530             'session_id':   ui_order['pos_session_id'],
531             'lines':        ui_order['lines'],
532             'pos_reference':ui_order['name'],
533             'partner_id':   ui_order['partner_id'] or False,
534         }
535
536     def _payment_fields(self, cr, uid, ui_paymentline, context=None):
537         return {
538             'amount':       ui_paymentline['amount'] or 0.0,
539             'payment_date': ui_paymentline['name'],
540             'statement_id': ui_paymentline['statement_id'],
541             'payment_name': ui_paymentline.get('note',False),
542             'journal':      ui_paymentline['journal_id'],
543         }
544
545     def create_from_ui(self, cr, uid, orders, context=None):
546         # Keep only new orders
547         submitted_references = [o['data']['name'] for o in orders]
548         existing_order_ids = self.search(cr, uid, [('pos_reference', 'in', submitted_references)], context=context)
549         existing_orders = self.read(cr, uid, existing_order_ids, ['pos_reference'], context=context)
550         existing_references = set([o['pos_reference'] for o in existing_orders])
551         orders_to_save = [o for o in orders if o['data']['name'] not in existing_references]
552
553         order_ids = []
554         for tmp_order in orders_to_save:
555             to_invoice = tmp_order['to_invoice']
556             order = tmp_order['data']
557             order_id = self.create(cr, uid, self._order_fields(cr, uid, order, context=context),context)
558
559             for payments in order['statement_ids']:
560                 self.add_payment(cr, uid, order_id, self._payment_fields(cr, uid, payments[2], context=context), context=context)
561
562             session = self.pool.get('pos.session').browse(cr, uid, order['pos_session_id'], context=context)
563             if session.sequence_number <= order['sequence_number']:
564                 session.write({'sequence_number': order['sequence_number'] + 1})
565                 session.refresh()
566
567             if order['amount_return']:
568                 cash_journal = session.cash_journal_id
569                 if not cash_journal:
570                     cash_journal_ids = filter(lambda st: st.journal_id.type=='cash', session.statement_ids)
571                     if not len(cash_journal_ids):
572                         raise osv.except_osv( _('error!'),
573                             _("No cash statement found for this session. Unable to record returned cash."))
574                     cash_journal = cash_journal_ids[0].journal_id
575                 self.add_payment(cr, uid, order_id, {
576                     'amount': -order['amount_return'],
577                     'payment_date': time.strftime('%Y-%m-%d %H:%M:%S'),
578                     'payment_name': _('return'),
579                     'journal': cash_journal.id,
580                 }, context=context)
581             order_ids.append(order_id)
582
583             try:
584                 self.signal_paid(cr, uid, [order_id])
585             except Exception as e:
586                 _logger.error('Could not fully process the POS Order: %s', tools.ustr(e))
587
588             if to_invoice:
589                 self.action_invoice(cr, uid, [order_id], context)
590                 order_obj = self.browse(cr, uid, order_id, context)
591                 self.pool['account.invoice'].signal_invoice_open(cr, uid, [order_obj.invoice_id.id])
592
593         return order_ids
594
595     def write(self, cr, uid, ids, vals, context=None):
596         res = super(pos_order, self).write(cr, uid, ids, vals, context=context)
597         #If you change the partner of the PoS order, change also the partner of the associated bank statement lines
598         partner_obj = self.pool.get('res.partner')
599         bsl_obj = self.pool.get("account.bank.statement.line")
600         if 'partner_id' in vals:
601             for posorder in self.browse(cr, uid, ids, context=context):
602                 if posorder.invoice_id:
603                     raise osv.except_osv( _('Error!'), _("You cannot change the partner of a POS order for which an invoice has already been issued."))
604                 if vals['partner_id']:
605                     p_id = partner_obj.browse(cr, uid, vals['partner_id'], context=context)
606                     part_id = partner_obj._find_accounting_partner(p_id).id
607                 else:
608                     part_id = False
609                 bsl_ids = [x.id for x in posorder.statement_ids]
610                 bsl_obj.write(cr, uid, bsl_ids, {'partner_id': part_id}, context=context)
611         return res
612
613     def unlink(self, cr, uid, ids, context=None):
614         for rec in self.browse(cr, uid, ids, context=context):
615             if rec.state not in ('draft','cancel'):
616                 raise osv.except_osv(_('Unable to Delete!'), _('In order to delete a sale, it must be new or cancelled.'))
617         return super(pos_order, self).unlink(cr, uid, ids, context=context)
618
619     def onchange_partner_id(self, cr, uid, ids, part=False, context=None):
620         if not part:
621             return {'value': {}}
622         pricelist = self.pool.get('res.partner').browse(cr, uid, part, context=context).property_product_pricelist.id
623         return {'value': {'pricelist_id': pricelist}}
624
625     def _amount_all(self, cr, uid, ids, name, args, context=None):
626         cur_obj = self.pool.get('res.currency')
627         res = {}
628         for order in self.browse(cr, uid, ids, context=context):
629             res[order.id] = {
630                 'amount_paid': 0.0,
631                 'amount_return':0.0,
632                 'amount_tax':0.0,
633             }
634             val1 = val2 = 0.0
635             cur = order.pricelist_id.currency_id
636             for payment in order.statement_ids:
637                 res[order.id]['amount_paid'] +=  payment.amount
638                 res[order.id]['amount_return'] += (payment.amount < 0 and payment.amount or 0)
639             for line in order.lines:
640                 val1 += line.price_subtotal_incl
641                 val2 += line.price_subtotal
642             res[order.id]['amount_tax'] = cur_obj.round(cr, uid, cur, val1-val2)
643             res[order.id]['amount_total'] = cur_obj.round(cr, uid, cur, val1)
644         return res
645
646     _columns = {
647         'name': fields.char('Order Ref', required=True, readonly=True, copy=False),
648         'company_id':fields.many2one('res.company', 'Company', required=True, readonly=True),
649         'date_order': fields.datetime('Order Date', readonly=True, select=True),
650         'user_id': fields.many2one('res.users', 'Salesman', help="Person who uses the the cash register. It can be a reliever, a student or an interim employee."),
651         'amount_tax': fields.function(_amount_all, string='Taxes', digits_compute=dp.get_precision('Account'), multi='all'),
652         'amount_total': fields.function(_amount_all, string='Total', multi='all'),
653         'amount_paid': fields.function(_amount_all, string='Paid', states={'draft': [('readonly', False)]}, readonly=True, digits_compute=dp.get_precision('Account'), multi='all'),
654         'amount_return': fields.function(_amount_all, 'Returned', digits_compute=dp.get_precision('Account'), multi='all'),
655         'lines': fields.one2many('pos.order.line', 'order_id', 'Order Lines', states={'draft': [('readonly', False)]}, readonly=True, copy=True),
656         'statement_ids': fields.one2many('account.bank.statement.line', 'pos_statement_id', 'Payments', states={'draft': [('readonly', False)]}, readonly=True),
657         'pricelist_id': fields.many2one('product.pricelist', 'Pricelist', required=True, states={'draft': [('readonly', False)]}, readonly=True),
658         'partner_id': fields.many2one('res.partner', 'Customer', change_default=True, select=1, states={'draft': [('readonly', False)], 'paid': [('readonly', False)]}),
659         'sequence_number': fields.integer('Sequence Number', help='A session-unique sequence number for the order'),
660
661         'session_id' : fields.many2one('pos.session', 'Session', 
662                                         #required=True,
663                                         select=1,
664                                         domain="[('state', '=', 'opened')]",
665                                         states={'draft' : [('readonly', False)]},
666                                         readonly=True),
667
668         'state': fields.selection([('draft', 'New'),
669                                    ('cancel', 'Cancelled'),
670                                    ('paid', 'Paid'),
671                                    ('done', 'Posted'),
672                                    ('invoiced', 'Invoiced')],
673                                   'Status', readonly=True, copy=False),
674
675         'invoice_id': fields.many2one('account.invoice', 'Invoice', copy=False),
676         'account_move': fields.many2one('account.move', 'Journal Entry', readonly=True, copy=False),
677         'picking_id': fields.many2one('stock.picking', 'Picking', readonly=True, copy=False),
678         'picking_type_id': fields.related('session_id', 'config_id', 'picking_type_id', string="Picking Type", type='many2one', relation='stock.picking.type'),
679         'location_id': fields.related('session_id', 'config_id', 'stock_location_id', string="Location", type='many2one', store=True, relation='stock.location'),
680         'note': fields.text('Internal Notes'),
681         'nb_print': fields.integer('Number of Print', readonly=True, copy=False),
682         'pos_reference': fields.char('Receipt Ref', readonly=True, copy=False),
683         'sale_journal': fields.related('session_id', 'config_id', 'journal_id', relation='account.journal', type='many2one', string='Sale Journal', store=True, readonly=True),
684     }
685
686     def _default_session(self, cr, uid, context=None):
687         so = self.pool.get('pos.session')
688         session_ids = so.search(cr, uid, [('state','=', 'opened'), ('user_id','=',uid)], context=context)
689         return session_ids and session_ids[0] or False
690
691     def _default_pricelist(self, cr, uid, context=None):
692         session_ids = self._default_session(cr, uid, context) 
693         if session_ids:
694             session_record = self.pool.get('pos.session').browse(cr, uid, session_ids, context=context)
695             return session_record.config_id.pricelist_id and session_record.config_id.pricelist_id.id or False
696         return False
697
698     def _get_out_picking_type(self, cr, uid, context=None):
699         return self.pool.get('ir.model.data').xmlid_to_res_id(
700                     cr, uid, 'point_of_sale.picking_type_posout', context=context)
701
702     _defaults = {
703         'user_id': lambda self, cr, uid, context: uid,
704         'state': 'draft',
705         'name': '/', 
706         'date_order': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
707         'nb_print': 0,
708         'sequence_number': 1,
709         'session_id': _default_session,
710         'company_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
711         'pricelist_id': _default_pricelist,
712     }
713
714     def create(self, cr, uid, values, context=None):
715         values['name'] = self.pool.get('ir.sequence').get(cr, uid, 'pos.order')
716         return super(pos_order, self).create(cr, uid, values, context=context)
717
718     def test_paid(self, cr, uid, ids, context=None):
719         """A Point of Sale is paid when the sum
720         @return: True
721         """
722         for order in self.browse(cr, uid, ids, context=context):
723             if order.lines and not order.amount_total:
724                 return True
725             if (not order.lines) or (not order.statement_ids) or \
726                 (abs(order.amount_total-order.amount_paid) > 0.00001):
727                 return False
728         return True
729
730     def create_picking(self, cr, uid, ids, context=None):
731         """Create a picking for each order and validate it."""
732         picking_obj = self.pool.get('stock.picking')
733         partner_obj = self.pool.get('res.partner')
734         move_obj = self.pool.get('stock.move')
735
736         for order in self.browse(cr, uid, ids, context=context):
737             addr = order.partner_id and partner_obj.address_get(cr, uid, [order.partner_id.id], ['delivery']) or {}
738             picking_type = order.picking_type_id
739             picking_id = False
740             if picking_type:
741                 picking_id = picking_obj.create(cr, uid, {
742                     'origin': order.name,
743                     'partner_id': addr.get('delivery',False),
744                     'picking_type_id': picking_type.id,
745                     'company_id': order.company_id.id,
746                     'move_type': 'direct',
747                     'note': order.note or "",
748                     'invoice_state': 'none',
749                 }, context=context)
750                 self.write(cr, uid, [order.id], {'picking_id': picking_id}, context=context)
751             location_id = order.location_id.id
752             if order.partner_id:
753                 destination_id = order.partner_id.property_stock_customer.id
754             elif picking_type:
755                 if not picking_type.default_location_dest_id:
756                     raise osv.except_osv(_('Error!'), _('Missing source or destination location for picking type %s. Please configure those fields and try again.' % (picking_type.name,)))
757                 destination_id = picking_type.default_location_dest_id.id
758             else:
759                 destination_id = partner_obj.default_get(cr, uid, ['property_stock_customer'], context=context)['property_stock_customer']
760
761             move_list = []
762             for line in order.lines:
763                 if line.product_id and line.product_id.type == 'service':
764                     continue
765
766                 move_list.append(move_obj.create(cr, uid, {
767                     'name': line.name,
768                     'product_uom': line.product_id.uom_id.id,
769                     'product_uos': line.product_id.uom_id.id,
770                     'picking_id': picking_id,
771                     'picking_type_id': picking_type.id, 
772                     'product_id': line.product_id.id,
773                     'product_uos_qty': abs(line.qty),
774                     'product_uom_qty': abs(line.qty),
775                     'state': 'draft',
776                     'location_id': location_id if line.qty >= 0 else destination_id,
777                     'location_dest_id': destination_id if line.qty >= 0 else location_id,
778                 }, context=context))
779                 
780             if picking_id:
781                 picking_obj.action_confirm(cr, uid, [picking_id], context=context)
782                 picking_obj.force_assign(cr, uid, [picking_id], context=context)
783                 picking_obj.action_done(cr, uid, [picking_id], context=context)
784             elif move_list:
785                 move_obj.action_confirm(cr, uid, move_list, context=context)
786                 move_obj.force_assign(cr, uid, move_list, context=context)
787                 move_obj.action_done(cr, uid, move_list, context=context)
788         return True
789
790     def cancel_order(self, cr, uid, ids, context=None):
791         """ Changes order state to cancel
792         @return: True
793         """
794         stock_picking_obj = self.pool.get('stock.picking')
795         for order in self.browse(cr, uid, ids, context=context):
796             stock_picking_obj.action_cancel(cr, uid, [order.picking_id.id])
797             if stock_picking_obj.browse(cr, uid, order.picking_id.id, context=context).state <> 'cancel':
798                 raise osv.except_osv(_('Error!'), _('Unable to cancel the picking.'))
799         self.write(cr, uid, ids, {'state': 'cancel'}, context=context)
800         return True
801
802     def add_payment(self, cr, uid, order_id, data, context=None):
803         """Create a new payment for the order"""
804         context = dict(context or {})
805         statement_line_obj = self.pool.get('account.bank.statement.line')
806         property_obj = self.pool.get('ir.property')
807         order = self.browse(cr, uid, order_id, context=context)
808         args = {
809             'amount': data['amount'],
810             'date': data.get('payment_date', time.strftime('%Y-%m-%d')),
811             'name': order.name + ': ' + (data.get('payment_name', '') or ''),
812             'partner_id': order.partner_id and order.partner_id.id or None,
813         }
814
815         context.pop('pos_session_id', False)
816
817         journal_id = data.get('journal', False)
818         statement_id = data.get('statement_id', False)
819         assert journal_id or statement_id, "No statement_id or journal_id passed to the method!"
820
821         for statement in order.session_id.statement_ids:
822             if statement.id == statement_id:
823                 journal_id = statement.journal_id.id
824                 break
825             elif statement.journal_id.id == journal_id:
826                 statement_id = statement.id
827                 break
828
829         if not statement_id:
830             raise osv.except_osv(_('Error!'), _('You have to open at least one cashbox.'))
831
832         args.update({
833             'statement_id' : statement_id,
834             'pos_statement_id' : order_id,
835             'journal_id' : journal_id,
836             'ref' : order.session_id.name,
837         })
838
839         statement_line_obj.create(cr, uid, args, context=context)
840
841         return statement_id
842
843     def refund(self, cr, uid, ids, context=None):
844         """Create a copy of order  for refund order"""
845         clone_list = []
846         line_obj = self.pool.get('pos.order.line')
847         
848         for order in self.browse(cr, uid, ids, context=context):
849             current_session_ids = self.pool.get('pos.session').search(cr, uid, [
850                 ('state', '!=', 'closed'),
851                 ('user_id', '=', uid)], context=context)
852             if not current_session_ids:
853                 raise osv.except_osv(_('Error!'), _('To return product(s), you need to open a session that will be used to register the refund.'))
854
855             clone_id = self.copy(cr, uid, order.id, {
856                 'name': order.name + ' REFUND', # not used, name forced by create
857                 'session_id': current_session_ids[0],
858                 'date_order': time.strftime('%Y-%m-%d %H:%M:%S'),
859             }, context=context)
860             clone_list.append(clone_id)
861
862         for clone in self.browse(cr, uid, clone_list, context=context):
863             for order_line in clone.lines:
864                 line_obj.write(cr, uid, [order_line.id], {
865                     'qty': -order_line.qty
866                 }, context=context)
867
868         abs = {
869             'name': _('Return Products'),
870             'view_type': 'form',
871             'view_mode': 'form',
872             'res_model': 'pos.order',
873             'res_id':clone_list[0],
874             'view_id': False,
875             'context':context,
876             'type': 'ir.actions.act_window',
877             'nodestroy': True,
878             'target': 'current',
879         }
880         return abs
881
882     def action_invoice_state(self, cr, uid, ids, context=None):
883         return self.write(cr, uid, ids, {'state':'invoiced'}, context=context)
884
885     def action_invoice(self, cr, uid, ids, context=None):
886         inv_ref = self.pool.get('account.invoice')
887         inv_line_ref = self.pool.get('account.invoice.line')
888         product_obj = self.pool.get('product.product')
889         inv_ids = []
890
891         for order in self.pool.get('pos.order').browse(cr, uid, ids, context=context):
892             if order.invoice_id:
893                 inv_ids.append(order.invoice_id.id)
894                 continue
895
896             if not order.partner_id:
897                 raise osv.except_osv(_('Error!'), _('Please provide a partner for the sale.'))
898
899             acc = order.partner_id.property_account_receivable.id
900             inv = {
901                 'name': order.name,
902                 'origin': order.name,
903                 'account_id': acc,
904                 'journal_id': order.sale_journal.id or None,
905                 'type': 'out_invoice',
906                 'reference': order.name,
907                 'partner_id': order.partner_id.id,
908                 'comment': order.note or '',
909                 'currency_id': order.pricelist_id.currency_id.id, # considering partner's sale pricelist's currency
910             }
911             inv.update(inv_ref.onchange_partner_id(cr, uid, [], 'out_invoice', order.partner_id.id)['value'])
912             if not inv.get('account_id', None):
913                 inv['account_id'] = acc
914             inv_id = inv_ref.create(cr, uid, inv, context=context)
915
916             self.write(cr, uid, [order.id], {'invoice_id': inv_id, 'state': 'invoiced'}, context=context)
917             inv_ids.append(inv_id)
918             for line in order.lines:
919                 inv_line = {
920                     'invoice_id': inv_id,
921                     'product_id': line.product_id.id,
922                     'quantity': line.qty,
923                 }
924                 inv_name = product_obj.name_get(cr, uid, [line.product_id.id], context=context)[0][1]
925                 inv_line.update(inv_line_ref.product_id_change(cr, uid, [],
926                                                                line.product_id.id,
927                                                                line.product_id.uom_id.id,
928                                                                line.qty, partner_id = order.partner_id.id,
929                                                                fposition_id=order.partner_id.property_account_position.id)['value'])
930                 inv_line['price_unit'] = line.price_unit
931                 inv_line['discount'] = line.discount
932                 inv_line['name'] = inv_name
933                 inv_line['invoice_line_tax_id'] = [(6, 0, [x.id for x in line.product_id.taxes_id] )]
934                 inv_line_ref.create(cr, uid, inv_line, context=context)
935             inv_ref.button_reset_taxes(cr, uid, [inv_id], context=context)
936             self.signal_invoice(cr, uid, [order.id])
937             inv_ref.signal_validate(cr, uid, [inv_id])
938
939         if not inv_ids: return {}
940
941         mod_obj = self.pool.get('ir.model.data')
942         res = mod_obj.get_object_reference(cr, uid, 'account', 'invoice_form')
943         res_id = res and res[1] or False
944         return {
945             'name': _('Customer Invoice'),
946             'view_type': 'form',
947             'view_mode': 'form',
948             'view_id': [res_id],
949             'res_model': 'account.invoice',
950             'context': "{'type':'out_invoice'}",
951             'type': 'ir.actions.act_window',
952             'nodestroy': True,
953             'target': 'current',
954             'res_id': inv_ids and inv_ids[0] or False,
955         }
956
957     def create_account_move(self, cr, uid, ids, context=None):
958         return self._create_account_move_line(cr, uid, ids, None, None, context=context)
959
960     def _create_account_move_line(self, cr, uid, ids, session=None, move_id=None, context=None):
961         # Tricky, via the workflow, we only have one id in the ids variable
962         """Create a account move line of order grouped by products or not."""
963         account_move_obj = self.pool.get('account.move')
964         account_period_obj = self.pool.get('account.period')
965         account_tax_obj = self.pool.get('account.tax')
966         property_obj = self.pool.get('ir.property')
967         cur_obj = self.pool.get('res.currency')
968
969         #session_ids = set(order.session_id for order in self.browse(cr, uid, ids, context=context))
970
971         if session and not all(session.id == order.session_id.id for order in self.browse(cr, uid, ids, context=context)):
972             raise osv.except_osv(_('Error!'), _('Selected orders do not have the same session!'))
973
974         grouped_data = {}
975         have_to_group_by = session and session.config_id.group_by or False
976
977         def compute_tax(amount, tax, line):
978             if amount > 0:
979                 tax_code_id = tax['base_code_id']
980                 tax_amount = line.price_subtotal * tax['base_sign']
981             else:
982                 tax_code_id = tax['ref_base_code_id']
983                 tax_amount = line.price_subtotal * tax['ref_base_sign']
984
985             return (tax_code_id, tax_amount,)
986
987         for order in self.browse(cr, uid, ids, context=context):
988             if order.account_move:
989                 continue
990             if order.state != 'paid':
991                 continue
992
993             current_company = order.sale_journal.company_id
994
995             group_tax = {}
996             account_def = property_obj.get(cr, uid, 'property_account_receivable', 'res.partner', context=context)
997
998             order_account = order.partner_id and \
999                             order.partner_id.property_account_receivable and \
1000                             order.partner_id.property_account_receivable.id or \
1001                             account_def and account_def.id or current_company.account_receivable.id
1002
1003             if move_id is None:
1004                 # Create an entry for the sale
1005                 move_id = account_move_obj.create(cr, uid, {
1006                     'ref' : order.name,
1007                     'journal_id': order.sale_journal.id,
1008                 }, context=context)
1009
1010             def insert_data(data_type, values):
1011                 # if have_to_group_by:
1012
1013                 sale_journal_id = order.sale_journal.id
1014                 period = account_period_obj.find(cr, uid, context=dict(context or {}, company_id=current_company.id))[0]
1015
1016                 # 'quantity': line.qty,
1017                 # 'product_id': line.product_id.id,
1018                 values.update({
1019                     'date': order.date_order[:10],
1020                     'ref': order.name,
1021                     'journal_id' : sale_journal_id,
1022                     'period_id' : period,
1023                     'move_id' : move_id,
1024                     'company_id': current_company.id,
1025                 })
1026
1027                 if data_type == 'product':
1028                     key = ('product', values['partner_id'], values['product_id'], values['debit'] > 0)
1029                 elif data_type == 'tax':
1030                     key = ('tax', values['partner_id'], values['tax_code_id'], values['debit'] > 0)
1031                 elif data_type == 'counter_part':
1032                     key = ('counter_part', values['partner_id'], values['account_id'], values['debit'] > 0)
1033                 else:
1034                     return
1035
1036                 grouped_data.setdefault(key, [])
1037
1038                 # if not have_to_group_by or (not grouped_data[key]):
1039                 #     grouped_data[key].append(values)
1040                 # else:
1041                 #     pass
1042
1043                 if have_to_group_by:
1044                     if not grouped_data[key]:
1045                         grouped_data[key].append(values)
1046                     else:
1047                         current_value = grouped_data[key][0]
1048                         current_value['quantity'] = current_value.get('quantity', 0.0) +  values.get('quantity', 0.0)
1049                         current_value['credit'] = current_value.get('credit', 0.0) + values.get('credit', 0.0)
1050                         current_value['debit'] = current_value.get('debit', 0.0) + values.get('debit', 0.0)
1051                         current_value['tax_amount'] = current_value.get('tax_amount', 0.0) + values.get('tax_amount', 0.0)
1052                 else:
1053                     grouped_data[key].append(values)
1054
1055             #because of the weird way the pos order is written, we need to make sure there is at least one line, 
1056             #because just after the 'for' loop there are references to 'line' and 'income_account' variables (that 
1057             #are set inside the for loop)
1058             #TOFIX: a deep refactoring of this method (and class!) is needed in order to get rid of this stupid hack
1059             assert order.lines, _('The POS order must have lines when calling this method')
1060             # Create an move for each order line
1061
1062             cur = order.pricelist_id.currency_id
1063             for line in order.lines:
1064                 tax_amount = 0
1065                 taxes = []
1066                 for t in line.product_id.taxes_id:
1067                     if t.company_id.id == current_company.id:
1068                         taxes.append(t)
1069                 computed_taxes = account_tax_obj.compute_all(cr, uid, taxes, line.price_unit * (100.0-line.discount) / 100.0, line.qty)['taxes']
1070
1071                 for tax in computed_taxes:
1072                     tax_amount += cur_obj.round(cr, uid, cur, tax['amount'])
1073                     group_key = (tax['tax_code_id'], tax['base_code_id'], tax['account_collected_id'], tax['id'])
1074
1075                     group_tax.setdefault(group_key, 0)
1076                     group_tax[group_key] += cur_obj.round(cr, uid, cur, tax['amount'])
1077
1078                 amount = line.price_subtotal
1079
1080                 # Search for the income account
1081                 if  line.product_id.property_account_income.id:
1082                     income_account = line.product_id.property_account_income.id
1083                 elif line.product_id.categ_id.property_account_income_categ.id:
1084                     income_account = line.product_id.categ_id.property_account_income_categ.id
1085                 else:
1086                     raise osv.except_osv(_('Error!'), _('Please define income '\
1087                         'account for this product: "%s" (id:%d).') \
1088                         % (line.product_id.name, line.product_id.id, ))
1089
1090                 # Empty the tax list as long as there is no tax code:
1091                 tax_code_id = False
1092                 tax_amount = 0
1093                 while computed_taxes:
1094                     tax = computed_taxes.pop(0)
1095                     tax_code_id, tax_amount = compute_tax(amount, tax, line)
1096
1097                     # If there is one we stop
1098                     if tax_code_id:
1099                         break
1100
1101                 # Create a move for the line
1102                 insert_data('product', {
1103                     'name': line.product_id.name,
1104                     'quantity': line.qty,
1105                     'product_id': line.product_id.id,
1106                     'account_id': income_account,
1107                     'credit': ((amount>0) and amount) or 0.0,
1108                     'debit': ((amount<0) and -amount) or 0.0,
1109                     'tax_code_id': tax_code_id,
1110                     'tax_amount': tax_amount,
1111                     'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1112                 })
1113
1114                 # For each remaining tax with a code, whe create a move line
1115                 for tax in computed_taxes:
1116                     tax_code_id, tax_amount = compute_tax(amount, tax, line)
1117                     if not tax_code_id:
1118                         continue
1119
1120                     insert_data('tax', {
1121                         'name': _('Tax'),
1122                         'product_id':line.product_id.id,
1123                         'quantity': line.qty,
1124                         'account_id': income_account,
1125                         'credit': 0.0,
1126                         'debit': 0.0,
1127                         'tax_code_id': tax_code_id,
1128                         'tax_amount': tax_amount,
1129                         'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1130                     })
1131
1132             # Create a move for each tax group
1133             (tax_code_pos, base_code_pos, account_pos, tax_id)= (0, 1, 2, 3)
1134
1135             for key, tax_amount in group_tax.items():
1136                 tax = self.pool.get('account.tax').browse(cr, uid, key[tax_id], context=context)
1137                 insert_data('tax', {
1138                     'name': _('Tax') + ' ' + tax.name,
1139                     'quantity': line.qty,
1140                     'product_id': line.product_id.id,
1141                     'account_id': key[account_pos] or income_account,
1142                     'credit': ((tax_amount>0) and tax_amount) or 0.0,
1143                     'debit': ((tax_amount<0) and -tax_amount) or 0.0,
1144                     'tax_code_id': key[tax_code_pos],
1145                     'tax_amount': tax_amount,
1146                     'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1147                 })
1148
1149             # counterpart
1150             insert_data('counter_part', {
1151                 'name': _("Trade Receivables"), #order.name,
1152                 'account_id': order_account,
1153                 'credit': ((order.amount_total < 0) and -order.amount_total) or 0.0,
1154                 'debit': ((order.amount_total > 0) and order.amount_total) or 0.0,
1155                 'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1156             })
1157
1158             order.write({'state':'done', 'account_move': move_id})
1159
1160         all_lines = []
1161         for group_key, group_data in grouped_data.iteritems():
1162             for value in group_data:
1163                 all_lines.append((0, 0, value),)
1164         if move_id: #In case no order was changed
1165             self.pool.get("account.move").write(cr, uid, [move_id], {'line_id':all_lines}, context=context)
1166
1167         return True
1168
1169     def action_payment(self, cr, uid, ids, context=None):
1170         return self.write(cr, uid, ids, {'state': 'payment'}, context=context)
1171
1172     def action_paid(self, cr, uid, ids, context=None):
1173         self.write(cr, uid, ids, {'state': 'paid'}, context=context)
1174         self.create_picking(cr, uid, ids, context=context)
1175         return True
1176
1177     def action_cancel(self, cr, uid, ids, context=None):
1178         self.write(cr, uid, ids, {'state': 'cancel'}, context=context)
1179         return True
1180
1181     def action_done(self, cr, uid, ids, context=None):
1182         self.create_account_move(cr, uid, ids, context=context)
1183         return True
1184
1185 class account_bank_statement(osv.osv):
1186     _inherit = 'account.bank.statement'
1187     _columns= {
1188         'user_id': fields.many2one('res.users', 'User', readonly=True),
1189     }
1190     _defaults = {
1191         'user_id': lambda self,cr,uid,c={}: uid
1192     }
1193
1194 class account_bank_statement_line(osv.osv):
1195     _inherit = 'account.bank.statement.line'
1196     _columns= {
1197         'pos_statement_id': fields.many2one('pos.order', ondelete='cascade'),
1198     }
1199
1200
1201 class pos_order_line(osv.osv):
1202     _name = "pos.order.line"
1203     _description = "Lines of Point of Sale"
1204     _rec_name = "product_id"
1205
1206     def _amount_line_all(self, cr, uid, ids, field_names, arg, context=None):
1207         res = dict([(i, {}) for i in ids])
1208         account_tax_obj = self.pool.get('account.tax')
1209         cur_obj = self.pool.get('res.currency')
1210         for line in self.browse(cr, uid, ids, context=context):
1211             taxes_ids = [ tax for tax in line.product_id.taxes_id if tax.company_id.id == line.order_id.company_id.id ]
1212             price = line.price_unit * (1 - (line.discount or 0.0) / 100.0)
1213             taxes = account_tax_obj.compute_all(cr, uid, taxes_ids, price, line.qty, product=line.product_id, partner=line.order_id.partner_id or False)
1214
1215             cur = line.order_id.pricelist_id.currency_id
1216             res[line.id]['price_subtotal'] = cur_obj.round(cr, uid, cur, taxes['total'])
1217             res[line.id]['price_subtotal_incl'] = cur_obj.round(cr, uid, cur, taxes['total_included'])
1218         return res
1219
1220     def onchange_product_id(self, cr, uid, ids, pricelist, product_id, qty=0, partner_id=False, context=None):
1221        context = context or {}
1222        if not product_id:
1223             return {}
1224        if not pricelist:
1225            raise osv.except_osv(_('No Pricelist!'),
1226                _('You have to select a pricelist in the sale form !\n' \
1227                'Please set one before choosing a product.'))
1228
1229        price = self.pool.get('product.pricelist').price_get(cr, uid, [pricelist],
1230                product_id, qty or 1.0, partner_id)[pricelist]
1231
1232        result = self.onchange_qty(cr, uid, ids, product_id, 0.0, qty, price, context=context)
1233        result['value']['price_unit'] = price
1234        return result
1235
1236     def onchange_qty(self, cr, uid, ids, product, discount, qty, price_unit, context=None):
1237         result = {}
1238         if not product:
1239             return result
1240         account_tax_obj = self.pool.get('account.tax')
1241         cur_obj = self.pool.get('res.currency')
1242
1243         prod = self.pool.get('product.product').browse(cr, uid, product, context=context)
1244
1245         price = price_unit * (1 - (discount or 0.0) / 100.0)
1246         taxes = account_tax_obj.compute_all(cr, uid, prod.taxes_id, price, qty, product=prod, partner=False)
1247
1248         result['price_subtotal'] = taxes['total']
1249         result['price_subtotal_incl'] = taxes['total_included']
1250         return {'value': result}
1251
1252     _columns = {
1253         'company_id': fields.many2one('res.company', 'Company', required=True),
1254         'name': fields.char('Line No', required=True, copy=False),
1255         'notice': fields.char('Discount Notice'),
1256         'product_id': fields.many2one('product.product', 'Product', domain=[('sale_ok', '=', True)], required=True, change_default=True),
1257         'price_unit': fields.float(string='Unit Price', digits_compute=dp.get_precision('Account')),
1258         'qty': fields.float('Quantity', digits_compute=dp.get_precision('Product UoS')),
1259         'price_subtotal': fields.function(_amount_line_all, multi='pos_order_line_amount', string='Subtotal w/o Tax', store=True),
1260         'price_subtotal_incl': fields.function(_amount_line_all, multi='pos_order_line_amount', string='Subtotal', store=True),
1261         'discount': fields.float('Discount (%)', digits_compute=dp.get_precision('Account')),
1262         'order_id': fields.many2one('pos.order', 'Order Ref', ondelete='cascade'),
1263         'create_date': fields.datetime('Creation Date', readonly=True),
1264     }
1265
1266     _defaults = {
1267         'name': lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'pos.order.line'),
1268         'qty': lambda *a: 1,
1269         'discount': lambda *a: 0.0,
1270         'company_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
1271     }
1272
1273 class ean_wizard(osv.osv_memory):
1274     _name = 'pos.ean_wizard'
1275     _columns = {
1276         'ean13_pattern': fields.char('Reference', size=13, required=True, translate=True),
1277     }
1278     def sanitize_ean13(self, cr, uid, ids, context):
1279         for r in self.browse(cr,uid,ids):
1280             ean13 = openerp.addons.product.product.sanitize_ean13(r.ean13_pattern)
1281             m = context.get('active_model')
1282             m_id =  context.get('active_id')
1283             self.pool[m].write(cr,uid,[m_id],{'ean13':ean13})
1284         return { 'type' : 'ir.actions.act_window_close' }
1285
1286 class pos_category(osv.osv):
1287     _name = "pos.category"
1288     _description = "Public Category"
1289     _order = "sequence, name"
1290
1291     _constraints = [
1292         (osv.osv._check_recursion, 'Error ! You cannot create recursive categories.', ['parent_id'])
1293     ]
1294
1295     def name_get(self, cr, uid, ids, context=None):
1296         if not len(ids):
1297             return []
1298         reads = self.read(cr, uid, ids, ['name','parent_id'], context=context)
1299         res = []
1300         for record in reads:
1301             name = record['name']
1302             if record['parent_id']:
1303                 name = record['parent_id'][1]+' / '+name
1304             res.append((record['id'], name))
1305         return res
1306
1307     def _name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
1308         res = self.name_get(cr, uid, ids, context=context)
1309         return dict(res)
1310
1311     def _get_image(self, cr, uid, ids, name, args, context=None):
1312         result = dict.fromkeys(ids, False)
1313         for obj in self.browse(cr, uid, ids, context=context):
1314             result[obj.id] = tools.image_get_resized_images(obj.image)
1315         return result
1316     
1317     def _set_image(self, cr, uid, id, name, value, args, context=None):
1318         return self.write(cr, uid, [id], {'image': tools.image_resize_image_big(value)}, context=context)
1319
1320     _columns = {
1321         'name': fields.char('Name', required=True, translate=True),
1322         'complete_name': fields.function(_name_get_fnc, type="char", string='Name'),
1323         'parent_id': fields.many2one('pos.category','Parent Category', select=True),
1324         'child_id': fields.one2many('pos.category', 'parent_id', string='Children Categories'),
1325         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of product categories."),
1326         
1327         # NOTE: there is no 'default image', because by default we don't show thumbnails for categories. However if we have a thumbnail
1328         # for at least one category, then we display a default image on the other, so that the buttons have consistent styling.
1329         # In this case, the default image is set by the js code.
1330         # NOTE2: image: all image fields are base64 encoded and PIL-supported
1331         'image': fields.binary("Image",
1332             help="This field holds the image used as image for the cateogry, limited to 1024x1024px."),
1333         'image_medium': fields.function(_get_image, fnct_inv=_set_image,
1334             string="Medium-sized image", type="binary", multi="_get_image",
1335             store={
1336                 'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
1337             },
1338             help="Medium-sized image of the category. It is automatically "\
1339                  "resized as a 128x128px image, with aspect ratio preserved. "\
1340                  "Use this field in form views or some kanban views."),
1341         'image_small': fields.function(_get_image, fnct_inv=_set_image,
1342             string="Smal-sized image", type="binary", multi="_get_image",
1343             store={
1344                 'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
1345             },
1346             help="Small-sized image of the category. It is automatically "\
1347                  "resized as a 64x64px image, with aspect ratio preserved. "\
1348                  "Use this field anywhere a small image is required."),
1349     }
1350
1351 class product_template(osv.osv):
1352     _inherit = 'product.template'
1353
1354     _columns = {
1355         'income_pdt': fields.boolean('Point of Sale Cash In', help="Check if, this is a product you can use to put cash into a statement for the point of sale backend."),
1356         'expense_pdt': fields.boolean('Point of Sale Cash Out', help="Check if, this is a product you can use to take cash from a statement for the point of sale backend, example: money lost, transfer to bank, etc."),
1357         'available_in_pos': fields.boolean('Available in the Point of Sale', help='Check if you want this product to appear in the Point of Sale'), 
1358         'to_weight' : fields.boolean('To Weigh', help="Check if the product should be weighted (mainly used with self check-out interface)."),
1359         'pos_categ_id': fields.many2one('pos.category','Point of Sale Category', help="Those categories are used to group similar products for point of sale."),
1360     }
1361
1362     _defaults = {
1363         'to_weight' : False,
1364         'available_in_pos': True,
1365     }
1366
1367     def edit_ean(self, cr, uid, ids, context):
1368         return {
1369             'name': _("Assign a Custom EAN"),
1370             'type': 'ir.actions.act_window',
1371             'view_type': 'form',
1372             'view_mode': 'form',
1373             'res_model': 'pos.ean_wizard',
1374             'target' : 'new',
1375             'view_id': False,
1376             'context':context,
1377         }
1378
1379 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: