[MERGE] forward port of branch saas-5 up to 655a5b9
[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, Odoo 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 Odoo 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_workflow('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 == 'done':
496                     continue
497                 if order.state not in ('paid', 'invoiced'):
498                     raise osv.except_osv(
499                         _('Error!'),
500                         _("You cannot confirm all orders of this session, because they have not the 'paid' status"))
501                 else:
502                     pos_order_obj.signal_workflow(cr, uid, [order.id], 'done')
503
504         return True
505
506     def open_frontend_cb(self, cr, uid, ids, context=None):
507         if not context:
508             context = {}
509         if not ids:
510             return {}
511         for session in self.browse(cr, uid, ids, context=context):
512             if session.user_id.id != uid:
513                 raise osv.except_osv(
514                         _('Error!'),
515                         _("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))
516         context.update({'active_id': ids[0]})
517         return {
518             'type' : 'ir.actions.act_url',
519             'target': 'self',
520             'url':   '/pos/web/',
521         }
522
523 class pos_order(osv.osv):
524     _name = "pos.order"
525     _description = "Point of Sale"
526     _order = "id desc"
527
528     def _order_fields(self, cr, uid, ui_order, context=None):
529         return {
530             'name':         ui_order['name'],
531             'user_id':      ui_order['user_id'] or False,
532             'session_id':   ui_order['pos_session_id'],
533             'lines':        ui_order['lines'],
534             'pos_reference':ui_order['name'],
535             'partner_id':   ui_order['partner_id'] or False,
536         }
537
538     def _payment_fields(self, cr, uid, ui_paymentline, context=None):
539         return {
540             'amount':       ui_paymentline['amount'] or 0.0,
541             'payment_date': ui_paymentline['name'],
542             'statement_id': ui_paymentline['statement_id'],
543             'payment_name': ui_paymentline.get('note',False),
544             'journal':      ui_paymentline['journal_id'],
545         }
546
547     def create_from_ui(self, cr, uid, orders, context=None):
548         # Keep only new orders
549         submitted_references = [o['data']['name'] for o in orders]
550         existing_order_ids = self.search(cr, uid, [('pos_reference', 'in', submitted_references)], context=context)
551         existing_orders = self.read(cr, uid, existing_order_ids, ['pos_reference'], context=context)
552         existing_references = set([o['pos_reference'] for o in existing_orders])
553         orders_to_save = [o for o in orders if o['data']['name'] not in existing_references]
554
555         order_ids = []
556         for tmp_order in orders_to_save:
557             to_invoice = tmp_order['to_invoice']
558             order = tmp_order['data']
559             order_id = self.create(cr, uid, self._order_fields(cr, uid, order, context=context),context)
560
561             for payments in order['statement_ids']:
562                 self.add_payment(cr, uid, order_id, self._payment_fields(cr, uid, payments[2], context=context), context=context)
563
564             session = self.pool.get('pos.session').browse(cr, uid, order['pos_session_id'], context=context)
565             if session.sequence_number <= order['sequence_number']:
566                 session.write({'sequence_number': order['sequence_number'] + 1})
567                 session.refresh()
568
569             if order['amount_return']:
570                 cash_journal = session.cash_journal_id
571                 if not cash_journal:
572                     cash_journal_ids = filter(lambda st: st.journal_id.type=='cash', session.statement_ids)
573                     if not len(cash_journal_ids):
574                         raise osv.except_osv( _('error!'),
575                             _("No cash statement found for this session. Unable to record returned cash."))
576                     cash_journal = cash_journal_ids[0].journal_id
577                 self.add_payment(cr, uid, order_id, {
578                     'amount': -order['amount_return'],
579                     'payment_date': time.strftime('%Y-%m-%d %H:%M:%S'),
580                     'payment_name': _('return'),
581                     'journal': cash_journal.id,
582                 }, context=context)
583             order_ids.append(order_id)
584
585             try:
586                 self.signal_workflow(cr, uid, [order_id], 'paid')
587             except Exception as e:
588                 _logger.error('Could not fully process the POS Order: %s', tools.ustr(e))
589
590             if to_invoice:
591                 self.action_invoice(cr, uid, [order_id], context)
592                 order_obj = self.browse(cr, uid, order_id, context)
593                 self.pool['account.invoice'].signal_workflow(cr, uid, [order_obj.invoice_id.id], 'invoice_open')
594
595         return order_ids
596
597     def write(self, cr, uid, ids, vals, context=None):
598         res = super(pos_order, self).write(cr, uid, ids, vals, context=context)
599         #If you change the partner of the PoS order, change also the partner of the associated bank statement lines
600         partner_obj = self.pool.get('res.partner')
601         bsl_obj = self.pool.get("account.bank.statement.line")
602         if 'partner_id' in vals:
603             for posorder in self.browse(cr, uid, ids, context=context):
604                 if posorder.invoice_id:
605                     raise osv.except_osv( _('Error!'), _("You cannot change the partner of a POS order for which an invoice has already been issued."))
606                 if vals['partner_id']:
607                     p_id = partner_obj.browse(cr, uid, vals['partner_id'], context=context)
608                     part_id = partner_obj._find_accounting_partner(p_id).id
609                 else:
610                     part_id = False
611                 bsl_ids = [x.id for x in posorder.statement_ids]
612                 bsl_obj.write(cr, uid, bsl_ids, {'partner_id': part_id}, context=context)
613         return res
614
615     def unlink(self, cr, uid, ids, context=None):
616         for rec in self.browse(cr, uid, ids, context=context):
617             if rec.state not in ('draft','cancel'):
618                 raise osv.except_osv(_('Unable to Delete!'), _('In order to delete a sale, it must be new or cancelled.'))
619         return super(pos_order, self).unlink(cr, uid, ids, context=context)
620
621     def onchange_partner_id(self, cr, uid, ids, part=False, context=None):
622         if not part:
623             return {'value': {}}
624         pricelist = self.pool.get('res.partner').browse(cr, uid, part, context=context).property_product_pricelist.id
625         return {'value': {'pricelist_id': pricelist}}
626
627     def _amount_all(self, cr, uid, ids, name, args, context=None):
628         cur_obj = self.pool.get('res.currency')
629         res = {}
630         for order in self.browse(cr, uid, ids, context=context):
631             res[order.id] = {
632                 'amount_paid': 0.0,
633                 'amount_return':0.0,
634                 'amount_tax':0.0,
635             }
636             val1 = val2 = 0.0
637             cur = order.pricelist_id.currency_id
638             for payment in order.statement_ids:
639                 res[order.id]['amount_paid'] +=  payment.amount
640                 res[order.id]['amount_return'] += (payment.amount < 0 and payment.amount or 0)
641             for line in order.lines:
642                 val1 += line.price_subtotal_incl
643                 val2 += line.price_subtotal
644             res[order.id]['amount_tax'] = cur_obj.round(cr, uid, cur, val1-val2)
645             res[order.id]['amount_total'] = cur_obj.round(cr, uid, cur, val1)
646         return res
647
648     _columns = {
649         'name': fields.char('Order Ref', required=True, readonly=True, copy=False),
650         'company_id':fields.many2one('res.company', 'Company', required=True, readonly=True),
651         'date_order': fields.datetime('Order Date', readonly=True, select=True),
652         '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."),
653         'amount_tax': fields.function(_amount_all, string='Taxes', digits_compute=dp.get_precision('Account'), multi='all'),
654         'amount_total': fields.function(_amount_all, string='Total', multi='all'),
655         'amount_paid': fields.function(_amount_all, string='Paid', states={'draft': [('readonly', False)]}, readonly=True, digits_compute=dp.get_precision('Account'), multi='all'),
656         'amount_return': fields.function(_amount_all, 'Returned', digits_compute=dp.get_precision('Account'), multi='all'),
657         'lines': fields.one2many('pos.order.line', 'order_id', 'Order Lines', states={'draft': [('readonly', False)]}, readonly=True, copy=True),
658         'statement_ids': fields.one2many('account.bank.statement.line', 'pos_statement_id', 'Payments', states={'draft': [('readonly', False)]}, readonly=True),
659         'pricelist_id': fields.many2one('product.pricelist', 'Pricelist', required=True, states={'draft': [('readonly', False)]}, readonly=True),
660         'partner_id': fields.many2one('res.partner', 'Customer', change_default=True, select=1, states={'draft': [('readonly', False)], 'paid': [('readonly', False)]}),
661         'sequence_number': fields.integer('Sequence Number', help='A session-unique sequence number for the order'),
662
663         'session_id' : fields.many2one('pos.session', 'Session', 
664                                         #required=True,
665                                         select=1,
666                                         domain="[('state', '=', 'opened')]",
667                                         states={'draft' : [('readonly', False)]},
668                                         readonly=True),
669
670         'state': fields.selection([('draft', 'New'),
671                                    ('cancel', 'Cancelled'),
672                                    ('paid', 'Paid'),
673                                    ('done', 'Posted'),
674                                    ('invoiced', 'Invoiced')],
675                                   'Status', readonly=True, copy=False),
676
677         'invoice_id': fields.many2one('account.invoice', 'Invoice', copy=False),
678         'account_move': fields.many2one('account.move', 'Journal Entry', readonly=True, copy=False),
679         'picking_id': fields.many2one('stock.picking', 'Picking', readonly=True, copy=False),
680         'picking_type_id': fields.related('session_id', 'config_id', 'picking_type_id', string="Picking Type", type='many2one', relation='stock.picking.type'),
681         'location_id': fields.related('session_id', 'config_id', 'stock_location_id', string="Location", type='many2one', store=True, relation='stock.location'),
682         'note': fields.text('Internal Notes'),
683         'nb_print': fields.integer('Number of Print', readonly=True, copy=False),
684         'pos_reference': fields.char('Receipt Ref', readonly=True, copy=False),
685         'sale_journal': fields.related('session_id', 'config_id', 'journal_id', relation='account.journal', type='many2one', string='Sale Journal', store=True, readonly=True),
686     }
687
688     def _default_session(self, cr, uid, context=None):
689         so = self.pool.get('pos.session')
690         session_ids = so.search(cr, uid, [('state','=', 'opened'), ('user_id','=',uid)], context=context)
691         return session_ids and session_ids[0] or False
692
693     def _default_pricelist(self, cr, uid, context=None):
694         session_ids = self._default_session(cr, uid, context) 
695         if session_ids:
696             session_record = self.pool.get('pos.session').browse(cr, uid, session_ids, context=context)
697             return session_record.config_id.pricelist_id and session_record.config_id.pricelist_id.id or False
698         return False
699
700     def _get_out_picking_type(self, cr, uid, context=None):
701         return self.pool.get('ir.model.data').xmlid_to_res_id(
702                     cr, uid, 'point_of_sale.picking_type_posout', context=context)
703
704     _defaults = {
705         'user_id': lambda self, cr, uid, context: uid,
706         'state': 'draft',
707         'name': '/', 
708         'date_order': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
709         'nb_print': 0,
710         'sequence_number': 1,
711         'session_id': _default_session,
712         'company_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
713         'pricelist_id': _default_pricelist,
714     }
715
716     def create(self, cr, uid, values, context=None):
717         values['name'] = self.pool.get('ir.sequence').get(cr, uid, 'pos.order')
718         return super(pos_order, self).create(cr, uid, values, context=context)
719
720     def test_paid(self, cr, uid, ids, context=None):
721         """A Point of Sale is paid when the sum
722         @return: True
723         """
724         for order in self.browse(cr, uid, ids, context=context):
725             if order.lines and not order.amount_total:
726                 return True
727             if (not order.lines) or (not order.statement_ids) or \
728                 (abs(order.amount_total-order.amount_paid) > 0.00001):
729                 return False
730         return True
731
732     def create_picking(self, cr, uid, ids, context=None):
733         """Create a picking for each order and validate it."""
734         picking_obj = self.pool.get('stock.picking')
735         partner_obj = self.pool.get('res.partner')
736         move_obj = self.pool.get('stock.move')
737
738         for order in self.browse(cr, uid, ids, context=context):
739             addr = order.partner_id and partner_obj.address_get(cr, uid, [order.partner_id.id], ['delivery']) or {}
740             picking_type = order.picking_type_id
741             picking_id = False
742             if picking_type:
743                 picking_id = picking_obj.create(cr, uid, {
744                     'origin': order.name,
745                     'partner_id': addr.get('delivery',False),
746                     'picking_type_id': picking_type.id,
747                     'company_id': order.company_id.id,
748                     'move_type': 'direct',
749                     'note': order.note or "",
750                     'invoice_state': 'none',
751                 }, context=context)
752                 self.write(cr, uid, [order.id], {'picking_id': picking_id}, context=context)
753             location_id = order.location_id.id
754             if order.partner_id:
755                 destination_id = order.partner_id.property_stock_customer.id
756             elif picking_type:
757                 if not picking_type.default_location_dest_id:
758                     raise osv.except_osv(_('Error!'), _('Missing source or destination location for picking type %s. Please configure those fields and try again.' % (picking_type.name,)))
759                 destination_id = picking_type.default_location_dest_id.id
760             else:
761                 destination_id = partner_obj.default_get(cr, uid, ['property_stock_customer'], context=context)['property_stock_customer']
762
763             move_list = []
764             for line in order.lines:
765                 if line.product_id and line.product_id.type == 'service':
766                     continue
767
768                 move_list.append(move_obj.create(cr, uid, {
769                     'name': line.name,
770                     'product_uom': line.product_id.uom_id.id,
771                     'product_uos': line.product_id.uom_id.id,
772                     'picking_id': picking_id,
773                     'picking_type_id': picking_type.id, 
774                     'product_id': line.product_id.id,
775                     'product_uos_qty': abs(line.qty),
776                     'product_uom_qty': abs(line.qty),
777                     'state': 'draft',
778                     'location_id': location_id if line.qty >= 0 else destination_id,
779                     'location_dest_id': destination_id if line.qty >= 0 else location_id,
780                 }, context=context))
781                 
782             if picking_id:
783                 picking_obj.action_confirm(cr, uid, [picking_id], context=context)
784                 picking_obj.force_assign(cr, uid, [picking_id], context=context)
785                 picking_obj.action_done(cr, uid, [picking_id], context=context)
786             elif move_list:
787                 move_obj.action_confirm(cr, uid, move_list, context=context)
788                 move_obj.force_assign(cr, uid, move_list, context=context)
789                 move_obj.action_done(cr, uid, move_list, context=context)
790         return True
791
792     def cancel_order(self, cr, uid, ids, context=None):
793         """ Changes order state to cancel
794         @return: True
795         """
796         stock_picking_obj = self.pool.get('stock.picking')
797         for order in self.browse(cr, uid, ids, context=context):
798             stock_picking_obj.action_cancel(cr, uid, [order.picking_id.id])
799             if stock_picking_obj.browse(cr, uid, order.picking_id.id, context=context).state <> 'cancel':
800                 raise osv.except_osv(_('Error!'), _('Unable to cancel the picking.'))
801         self.write(cr, uid, ids, {'state': 'cancel'}, context=context)
802         return True
803
804     def add_payment(self, cr, uid, order_id, data, context=None):
805         """Create a new payment for the order"""
806         context = dict(context or {})
807         statement_line_obj = self.pool.get('account.bank.statement.line')
808         property_obj = self.pool.get('ir.property')
809         order = self.browse(cr, uid, order_id, context=context)
810         args = {
811             'amount': data['amount'],
812             'date': data.get('payment_date', time.strftime('%Y-%m-%d')),
813             'name': order.name + ': ' + (data.get('payment_name', '') or ''),
814             'partner_id': order.partner_id and order.partner_id.id or None,
815         }
816         account_def = property_obj.get(cr, uid, 'property_account_receivable', 'res.partner', context=context)
817         args['account_id'] = (order.partner_id and order.partner_id.property_account_receivable \
818                              and order.partner_id.property_account_receivable.id) or (account_def and account_def.id) or False
819
820         if not args['account_id']:
821             if not args['partner_id']:
822                 msg = _('There is no receivable account defined to make payment.')
823             else:
824                 msg = _('There is no receivable account defined to make payment for the partner: "%s" (id:%d).') % (order.partner_id.name, order.partner_id.id,)
825             raise osv.except_osv(_('Configuration Error!'), msg)
826
827         context.pop('pos_session_id', False)
828
829         journal_id = data.get('journal', False)
830         statement_id = data.get('statement_id', False)
831         assert journal_id or statement_id, "No statement_id or journal_id passed to the method!"
832
833         for statement in order.session_id.statement_ids:
834             if statement.id == statement_id:
835                 journal_id = statement.journal_id.id
836                 break
837             elif statement.journal_id.id == journal_id:
838                 statement_id = statement.id
839                 break
840
841         if not statement_id:
842             raise osv.except_osv(_('Error!'), _('You have to open at least one cashbox.'))
843
844         args.update({
845             'statement_id': statement_id,
846             'pos_statement_id': order_id,
847             'journal_id': journal_id,
848             'ref': order.session_id.name,
849         })
850
851         statement_line_obj.create(cr, uid, args, context=context)
852
853         return statement_id
854
855     def refund(self, cr, uid, ids, context=None):
856         """Create a copy of order  for refund order"""
857         clone_list = []
858         line_obj = self.pool.get('pos.order.line')
859         
860         for order in self.browse(cr, uid, ids, context=context):
861             current_session_ids = self.pool.get('pos.session').search(cr, uid, [
862                 ('state', '!=', 'closed'),
863                 ('user_id', '=', uid)], context=context)
864             if not current_session_ids:
865                 raise osv.except_osv(_('Error!'), _('To return product(s), you need to open a session that will be used to register the refund.'))
866
867             clone_id = self.copy(cr, uid, order.id, {
868                 'name': order.name + ' REFUND', # not used, name forced by create
869                 'session_id': current_session_ids[0],
870                 'date_order': time.strftime('%Y-%m-%d %H:%M:%S'),
871             }, context=context)
872             clone_list.append(clone_id)
873
874         for clone in self.browse(cr, uid, clone_list, context=context):
875             for order_line in clone.lines:
876                 line_obj.write(cr, uid, [order_line.id], {
877                     'qty': -order_line.qty
878                 }, context=context)
879
880         abs = {
881             'name': _('Return Products'),
882             'view_type': 'form',
883             'view_mode': 'form',
884             'res_model': 'pos.order',
885             'res_id':clone_list[0],
886             'view_id': False,
887             'context':context,
888             'type': 'ir.actions.act_window',
889             'nodestroy': True,
890             'target': 'current',
891         }
892         return abs
893
894     def action_invoice_state(self, cr, uid, ids, context=None):
895         return self.write(cr, uid, ids, {'state':'invoiced'}, context=context)
896
897     def action_invoice(self, cr, uid, ids, context=None):
898         inv_ref = self.pool.get('account.invoice')
899         inv_line_ref = self.pool.get('account.invoice.line')
900         product_obj = self.pool.get('product.product')
901         inv_ids = []
902
903         for order in self.pool.get('pos.order').browse(cr, uid, ids, context=context):
904             if order.invoice_id:
905                 inv_ids.append(order.invoice_id.id)
906                 continue
907
908             if not order.partner_id:
909                 raise osv.except_osv(_('Error!'), _('Please provide a partner for the sale.'))
910
911             acc = order.partner_id.property_account_receivable.id
912             inv = {
913                 'name': order.name,
914                 'origin': order.name,
915                 'account_id': acc,
916                 'journal_id': order.sale_journal.id or None,
917                 'type': 'out_invoice',
918                 'reference': order.name,
919                 'partner_id': order.partner_id.id,
920                 'comment': order.note or '',
921                 'currency_id': order.pricelist_id.currency_id.id, # considering partner's sale pricelist's currency
922             }
923             inv.update(inv_ref.onchange_partner_id(cr, uid, [], 'out_invoice', order.partner_id.id)['value'])
924             if not inv.get('account_id', None):
925                 inv['account_id'] = acc
926             inv_id = inv_ref.create(cr, uid, inv, context=context)
927
928             self.write(cr, uid, [order.id], {'invoice_id': inv_id, 'state': 'invoiced'}, context=context)
929             inv_ids.append(inv_id)
930             for line in order.lines:
931                 inv_line = {
932                     'invoice_id': inv_id,
933                     'product_id': line.product_id.id,
934                     'quantity': line.qty,
935                 }
936                 inv_name = product_obj.name_get(cr, uid, [line.product_id.id], context=context)[0][1]
937                 inv_line.update(inv_line_ref.product_id_change(cr, uid, [],
938                                                                line.product_id.id,
939                                                                line.product_id.uom_id.id,
940                                                                line.qty, partner_id = order.partner_id.id,
941                                                                fposition_id=order.partner_id.property_account_position.id)['value'])
942                 inv_line['price_unit'] = line.price_unit
943                 inv_line['discount'] = line.discount
944                 inv_line['name'] = inv_name
945                 inv_line['invoice_line_tax_id'] = [(6, 0, [x.id for x in line.product_id.taxes_id] )]
946                 inv_line_ref.create(cr, uid, inv_line, context=context)
947             inv_ref.button_reset_taxes(cr, uid, [inv_id], context=context)
948             self.signal_workflow(cr, uid, [order.id], 'invoice')
949             inv_ref.signal_workflow(cr, uid, [inv_id], 'validate')
950
951         if not inv_ids: return {}
952
953         mod_obj = self.pool.get('ir.model.data')
954         res = mod_obj.get_object_reference(cr, uid, 'account', 'invoice_form')
955         res_id = res and res[1] or False
956         return {
957             'name': _('Customer Invoice'),
958             'view_type': 'form',
959             'view_mode': 'form',
960             'view_id': [res_id],
961             'res_model': 'account.invoice',
962             'context': "{'type':'out_invoice'}",
963             'type': 'ir.actions.act_window',
964             'nodestroy': True,
965             'target': 'current',
966             'res_id': inv_ids and inv_ids[0] or False,
967         }
968
969     def create_account_move(self, cr, uid, ids, context=None):
970         return self._create_account_move_line(cr, uid, ids, None, None, context=context)
971
972     def _create_account_move_line(self, cr, uid, ids, session=None, move_id=None, context=None):
973         # Tricky, via the workflow, we only have one id in the ids variable
974         """Create a account move line of order grouped by products or not."""
975         account_move_obj = self.pool.get('account.move')
976         account_period_obj = self.pool.get('account.period')
977         account_tax_obj = self.pool.get('account.tax')
978         property_obj = self.pool.get('ir.property')
979         cur_obj = self.pool.get('res.currency')
980
981         #session_ids = set(order.session_id for order in self.browse(cr, uid, ids, context=context))
982
983         if session and not all(session.id == order.session_id.id for order in self.browse(cr, uid, ids, context=context)):
984             raise osv.except_osv(_('Error!'), _('Selected orders do not have the same session!'))
985
986         grouped_data = {}
987         have_to_group_by = session and session.config_id.group_by or False
988
989         def compute_tax(amount, tax, line):
990             if amount > 0:
991                 tax_code_id = tax['base_code_id']
992                 tax_amount = line.price_subtotal * tax['base_sign']
993             else:
994                 tax_code_id = tax['ref_base_code_id']
995                 tax_amount = line.price_subtotal * tax['ref_base_sign']
996
997             return (tax_code_id, tax_amount,)
998
999         for order in self.browse(cr, uid, ids, context=context):
1000             if order.account_move:
1001                 continue
1002             if order.state != 'paid':
1003                 continue
1004
1005             current_company = order.sale_journal.company_id
1006
1007             group_tax = {}
1008             account_def = property_obj.get(cr, uid, 'property_account_receivable', 'res.partner', context=context)
1009
1010             order_account = order.partner_id and \
1011                             order.partner_id.property_account_receivable and \
1012                             order.partner_id.property_account_receivable.id or \
1013                             account_def and account_def.id or current_company.account_receivable.id
1014
1015             if move_id is None:
1016                 # Create an entry for the sale
1017                 move_id = account_move_obj.create(cr, uid, {
1018                     'ref' : order.name,
1019                     'journal_id': order.sale_journal.id,
1020                 }, context=context)
1021
1022             def insert_data(data_type, values):
1023                 # if have_to_group_by:
1024
1025                 sale_journal_id = order.sale_journal.id
1026                 period = account_period_obj.find(cr, uid, context=dict(context or {}, company_id=current_company.id))[0]
1027
1028                 # 'quantity': line.qty,
1029                 # 'product_id': line.product_id.id,
1030                 values.update({
1031                     'date': order.date_order[:10],
1032                     'ref': order.name,
1033                     'journal_id' : sale_journal_id,
1034                     'period_id' : period,
1035                     'move_id' : move_id,
1036                     'company_id': current_company.id,
1037                 })
1038
1039                 if data_type == 'product':
1040                     key = ('product', values['partner_id'], values['product_id'], values['debit'] > 0)
1041                 elif data_type == 'tax':
1042                     key = ('tax', values['partner_id'], values['tax_code_id'], values['debit'] > 0)
1043                 elif data_type == 'counter_part':
1044                     key = ('counter_part', values['partner_id'], values['account_id'], values['debit'] > 0)
1045                 else:
1046                     return
1047
1048                 grouped_data.setdefault(key, [])
1049
1050                 # if not have_to_group_by or (not grouped_data[key]):
1051                 #     grouped_data[key].append(values)
1052                 # else:
1053                 #     pass
1054
1055                 if have_to_group_by:
1056                     if not grouped_data[key]:
1057                         grouped_data[key].append(values)
1058                     else:
1059                         current_value = grouped_data[key][0]
1060                         current_value['quantity'] = current_value.get('quantity', 0.0) +  values.get('quantity', 0.0)
1061                         current_value['credit'] = current_value.get('credit', 0.0) + values.get('credit', 0.0)
1062                         current_value['debit'] = current_value.get('debit', 0.0) + values.get('debit', 0.0)
1063                         current_value['tax_amount'] = current_value.get('tax_amount', 0.0) + values.get('tax_amount', 0.0)
1064                 else:
1065                     grouped_data[key].append(values)
1066
1067             #because of the weird way the pos order is written, we need to make sure there is at least one line, 
1068             #because just after the 'for' loop there are references to 'line' and 'income_account' variables (that 
1069             #are set inside the for loop)
1070             #TOFIX: a deep refactoring of this method (and class!) is needed in order to get rid of this stupid hack
1071             assert order.lines, _('The POS order must have lines when calling this method')
1072             # Create an move for each order line
1073
1074             cur = order.pricelist_id.currency_id
1075             for line in order.lines:
1076                 tax_amount = 0
1077                 taxes = []
1078                 for t in line.product_id.taxes_id:
1079                     if t.company_id.id == current_company.id:
1080                         taxes.append(t)
1081                 computed_taxes = account_tax_obj.compute_all(cr, uid, taxes, line.price_unit * (100.0-line.discount) / 100.0, line.qty)['taxes']
1082
1083                 for tax in computed_taxes:
1084                     tax_amount += cur_obj.round(cr, uid, cur, tax['amount'])
1085                     group_key = (tax['tax_code_id'], tax['base_code_id'], tax['account_collected_id'], tax['id'])
1086
1087                     group_tax.setdefault(group_key, 0)
1088                     group_tax[group_key] += cur_obj.round(cr, uid, cur, tax['amount'])
1089
1090                 amount = line.price_subtotal
1091
1092                 # Search for the income account
1093                 if  line.product_id.property_account_income.id:
1094                     income_account = line.product_id.property_account_income.id
1095                 elif line.product_id.categ_id.property_account_income_categ.id:
1096                     income_account = line.product_id.categ_id.property_account_income_categ.id
1097                 else:
1098                     raise osv.except_osv(_('Error!'), _('Please define income '\
1099                         'account for this product: "%s" (id:%d).') \
1100                         % (line.product_id.name, line.product_id.id, ))
1101
1102                 # Empty the tax list as long as there is no tax code:
1103                 tax_code_id = False
1104                 tax_amount = 0
1105                 while computed_taxes:
1106                     tax = computed_taxes.pop(0)
1107                     tax_code_id, tax_amount = compute_tax(amount, tax, line)
1108
1109                     # If there is one we stop
1110                     if tax_code_id:
1111                         break
1112
1113                 # Create a move for the line
1114                 insert_data('product', {
1115                     'name': line.product_id.name,
1116                     'quantity': line.qty,
1117                     'product_id': line.product_id.id,
1118                     'account_id': income_account,
1119                     'credit': ((amount>0) and amount) or 0.0,
1120                     'debit': ((amount<0) and -amount) or 0.0,
1121                     'tax_code_id': tax_code_id,
1122                     'tax_amount': tax_amount,
1123                     'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1124                 })
1125
1126                 # For each remaining tax with a code, whe create a move line
1127                 for tax in computed_taxes:
1128                     tax_code_id, tax_amount = compute_tax(amount, tax, line)
1129                     if not tax_code_id:
1130                         continue
1131
1132                     insert_data('tax', {
1133                         'name': _('Tax'),
1134                         'product_id':line.product_id.id,
1135                         'quantity': line.qty,
1136                         'account_id': income_account,
1137                         'credit': 0.0,
1138                         'debit': 0.0,
1139                         'tax_code_id': tax_code_id,
1140                         'tax_amount': tax_amount,
1141                         'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1142                     })
1143
1144             # Create a move for each tax group
1145             (tax_code_pos, base_code_pos, account_pos, tax_id)= (0, 1, 2, 3)
1146
1147             for key, tax_amount in group_tax.items():
1148                 tax = self.pool.get('account.tax').browse(cr, uid, key[tax_id], context=context)
1149                 insert_data('tax', {
1150                     'name': _('Tax') + ' ' + tax.name,
1151                     'quantity': line.qty,
1152                     'product_id': line.product_id.id,
1153                     'account_id': key[account_pos] or income_account,
1154                     'credit': ((tax_amount>0) and tax_amount) or 0.0,
1155                     'debit': ((tax_amount<0) and -tax_amount) or 0.0,
1156                     'tax_code_id': key[tax_code_pos],
1157                     'tax_amount': tax_amount,
1158                     'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1159                 })
1160
1161             # counterpart
1162             insert_data('counter_part', {
1163                 'name': _("Trade Receivables"), #order.name,
1164                 'account_id': order_account,
1165                 'credit': ((order.amount_total < 0) and -order.amount_total) or 0.0,
1166                 'debit': ((order.amount_total > 0) and order.amount_total) or 0.0,
1167                 'partner_id': order.partner_id and self.pool.get("res.partner")._find_accounting_partner(order.partner_id).id or False
1168             })
1169
1170             order.write({'state':'done', 'account_move': move_id})
1171
1172         all_lines = []
1173         for group_key, group_data in grouped_data.iteritems():
1174             for value in group_data:
1175                 all_lines.append((0, 0, value),)
1176         if move_id: #In case no order was changed
1177             self.pool.get("account.move").write(cr, uid, [move_id], {'line_id':all_lines}, context=context)
1178
1179         return True
1180
1181     def action_payment(self, cr, uid, ids, context=None):
1182         return self.write(cr, uid, ids, {'state': 'payment'}, context=context)
1183
1184     def action_paid(self, cr, uid, ids, context=None):
1185         self.write(cr, uid, ids, {'state': 'paid'}, context=context)
1186         self.create_picking(cr, uid, ids, context=context)
1187         return True
1188
1189     def action_cancel(self, cr, uid, ids, context=None):
1190         self.write(cr, uid, ids, {'state': 'cancel'}, context=context)
1191         return True
1192
1193     def action_done(self, cr, uid, ids, context=None):
1194         self.create_account_move(cr, uid, ids, context=context)
1195         return True
1196
1197 class account_bank_statement(osv.osv):
1198     _inherit = 'account.bank.statement'
1199     _columns= {
1200         'user_id': fields.many2one('res.users', 'User', readonly=True),
1201     }
1202     _defaults = {
1203         'user_id': lambda self,cr,uid,c={}: uid
1204     }
1205
1206 class account_bank_statement_line(osv.osv):
1207     _inherit = 'account.bank.statement.line'
1208     _columns= {
1209         'pos_statement_id': fields.many2one('pos.order', ondelete='cascade'),
1210     }
1211
1212
1213 class pos_order_line(osv.osv):
1214     _name = "pos.order.line"
1215     _description = "Lines of Point of Sale"
1216     _rec_name = "product_id"
1217
1218     def _amount_line_all(self, cr, uid, ids, field_names, arg, context=None):
1219         res = dict([(i, {}) for i in ids])
1220         account_tax_obj = self.pool.get('account.tax')
1221         cur_obj = self.pool.get('res.currency')
1222         for line in self.browse(cr, uid, ids, context=context):
1223             taxes_ids = [ tax for tax in line.product_id.taxes_id if tax.company_id.id == line.order_id.company_id.id ]
1224             price = line.price_unit * (1 - (line.discount or 0.0) / 100.0)
1225             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)
1226
1227             cur = line.order_id.pricelist_id.currency_id
1228             res[line.id]['price_subtotal'] = cur_obj.round(cr, uid, cur, taxes['total'])
1229             res[line.id]['price_subtotal_incl'] = cur_obj.round(cr, uid, cur, taxes['total_included'])
1230         return res
1231
1232     def onchange_product_id(self, cr, uid, ids, pricelist, product_id, qty=0, partner_id=False, context=None):
1233        context = context or {}
1234        if not product_id:
1235             return {}
1236        if not pricelist:
1237            raise osv.except_osv(_('No Pricelist!'),
1238                _('You have to select a pricelist in the sale form !\n' \
1239                'Please set one before choosing a product.'))
1240
1241        price = self.pool.get('product.pricelist').price_get(cr, uid, [pricelist],
1242                product_id, qty or 1.0, partner_id)[pricelist]
1243
1244        result = self.onchange_qty(cr, uid, ids, product_id, 0.0, qty, price, context=context)
1245        result['value']['price_unit'] = price
1246        return result
1247
1248     def onchange_qty(self, cr, uid, ids, product, discount, qty, price_unit, context=None):
1249         result = {}
1250         if not product:
1251             return result
1252         account_tax_obj = self.pool.get('account.tax')
1253         cur_obj = self.pool.get('res.currency')
1254
1255         prod = self.pool.get('product.product').browse(cr, uid, product, context=context)
1256
1257         price = price_unit * (1 - (discount or 0.0) / 100.0)
1258         taxes = account_tax_obj.compute_all(cr, uid, prod.taxes_id, price, qty, product=prod, partner=False)
1259
1260         result['price_subtotal'] = taxes['total']
1261         result['price_subtotal_incl'] = taxes['total_included']
1262         return {'value': result}
1263
1264     _columns = {
1265         'company_id': fields.many2one('res.company', 'Company', required=True),
1266         'name': fields.char('Line No', required=True, copy=False),
1267         'notice': fields.char('Discount Notice'),
1268         'product_id': fields.many2one('product.product', 'Product', domain=[('sale_ok', '=', True)], required=True, change_default=True),
1269         'price_unit': fields.float(string='Unit Price', digits_compute=dp.get_precision('Account')),
1270         'qty': fields.float('Quantity', digits_compute=dp.get_precision('Product UoS')),
1271         'price_subtotal': fields.function(_amount_line_all, multi='pos_order_line_amount', string='Subtotal w/o Tax', store=True),
1272         'price_subtotal_incl': fields.function(_amount_line_all, multi='pos_order_line_amount', string='Subtotal', store=True),
1273         'discount': fields.float('Discount (%)', digits_compute=dp.get_precision('Account')),
1274         'order_id': fields.many2one('pos.order', 'Order Ref', ondelete='cascade'),
1275         'create_date': fields.datetime('Creation Date', readonly=True),
1276     }
1277
1278     _defaults = {
1279         'name': lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'pos.order.line'),
1280         'qty': lambda *a: 1,
1281         'discount': lambda *a: 0.0,
1282         'company_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
1283     }
1284
1285 class ean_wizard(osv.osv_memory):
1286     _name = 'pos.ean_wizard'
1287     _columns = {
1288         'ean13_pattern': fields.char('Reference', size=13, required=True, translate=True),
1289     }
1290     def sanitize_ean13(self, cr, uid, ids, context):
1291         for r in self.browse(cr,uid,ids):
1292             ean13 = openerp.addons.product.product.sanitize_ean13(r.ean13_pattern)
1293             m = context.get('active_model')
1294             m_id =  context.get('active_id')
1295             self.pool[m].write(cr,uid,[m_id],{'ean13':ean13})
1296         return { 'type' : 'ir.actions.act_window_close' }
1297
1298 class pos_category(osv.osv):
1299     _name = "pos.category"
1300     _description = "Public Category"
1301     _order = "sequence, name"
1302
1303     _constraints = [
1304         (osv.osv._check_recursion, 'Error ! You cannot create recursive categories.', ['parent_id'])
1305     ]
1306
1307     def name_get(self, cr, uid, ids, context=None):
1308         if not len(ids):
1309             return []
1310         reads = self.read(cr, uid, ids, ['name','parent_id'], context=context)
1311         res = []
1312         for record in reads:
1313             name = record['name']
1314             if record['parent_id']:
1315                 name = record['parent_id'][1]+' / '+name
1316             res.append((record['id'], name))
1317         return res
1318
1319     def _name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
1320         res = self.name_get(cr, uid, ids, context=context)
1321         return dict(res)
1322
1323     def _get_image(self, cr, uid, ids, name, args, context=None):
1324         result = dict.fromkeys(ids, False)
1325         for obj in self.browse(cr, uid, ids, context=context):
1326             result[obj.id] = tools.image_get_resized_images(obj.image)
1327         return result
1328     
1329     def _set_image(self, cr, uid, id, name, value, args, context=None):
1330         return self.write(cr, uid, [id], {'image': tools.image_resize_image_big(value)}, context=context)
1331
1332     _columns = {
1333         'name': fields.char('Name', required=True, translate=True),
1334         'complete_name': fields.function(_name_get_fnc, type="char", string='Name'),
1335         'parent_id': fields.many2one('pos.category','Parent Category', select=True),
1336         'child_id': fields.one2many('pos.category', 'parent_id', string='Children Categories'),
1337         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of product categories."),
1338         
1339         # NOTE: there is no 'default image', because by default we don't show thumbnails for categories. However if we have a thumbnail
1340         # for at least one category, then we display a default image on the other, so that the buttons have consistent styling.
1341         # In this case, the default image is set by the js code.
1342         # NOTE2: image: all image fields are base64 encoded and PIL-supported
1343         'image': fields.binary("Image",
1344             help="This field holds the image used as image for the cateogry, limited to 1024x1024px."),
1345         'image_medium': fields.function(_get_image, fnct_inv=_set_image,
1346             string="Medium-sized image", type="binary", multi="_get_image",
1347             store={
1348                 'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
1349             },
1350             help="Medium-sized image of the category. It is automatically "\
1351                  "resized as a 128x128px image, with aspect ratio preserved. "\
1352                  "Use this field in form views or some kanban views."),
1353         'image_small': fields.function(_get_image, fnct_inv=_set_image,
1354             string="Smal-sized image", type="binary", multi="_get_image",
1355             store={
1356                 'pos.category': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
1357             },
1358             help="Small-sized image of the category. It is automatically "\
1359                  "resized as a 64x64px image, with aspect ratio preserved. "\
1360                  "Use this field anywhere a small image is required."),
1361     }
1362
1363 class product_template(osv.osv):
1364     _inherit = 'product.template'
1365
1366     _columns = {
1367         '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."),
1368         '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."),
1369         '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'), 
1370         'to_weight' : fields.boolean('To Weigh', help="Check if the product should be weighted (mainly used with self check-out interface)."),
1371         'pos_categ_id': fields.many2one('pos.category','Point of Sale Category', help="Those categories are used to group similar products for point of sale."),
1372     }
1373
1374     _defaults = {
1375         'to_weight' : False,
1376         'available_in_pos': True,
1377     }
1378
1379 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: