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