[PYTHON][VIEW] +nrt is optionnal for each pos
[burette/nrt_point_of_sale.git] / point_of_sale.py
index dcfb14e..82144c0 100644 (file)
 from openerp.osv import osv
 from openerp.osv import orm
 from openerp.osv import fields
+from time import strftime
+import openerp.exceptions
+from openerp.tools.translate import _
+
+
+class pos_config(orm.Model):
+    _inherit = 'pos.config'
+
+    def _is_pos_used(self, cr, uid, ids, name, args, context=None):
+        """Return True if the pos is already in use."""
+        res = {}
+        session_obj =  self.pool.get('pos.session')
+        for id in ids:
+            sessions = session_obj.search(cr, uid, [('config_id', 'in', ids),('state', '!=', 'closed')], context=context)
+            if sessions:
+                res[id] = True
+            else:
+                res[id] = False
+        return res
+
+    _columns = {
+        'nrt': fields.boolean('Non-real time', help='This pos move lines will '\
+                              'be based on the session opening if this box is '\
+                              'checked. This field is readonly if the point of '\
+                              'sale is already in use.'),
+        'is_pos_used': fields.function(_is_pos_used,
+                            method=True,
+                            string='Is this pos in used?',
+                            type='boolean',
+                            store=False),
+    }
+
+    _defaults = {
+        'nrt': lambda *a: True,
+    }
 
 
 class pos_session(orm.Model):
     _inherit = 'pos.session'
 
+    def onchange_config_id(self, cr, uid, ids, config_id, context=None):
+        """Return is_nrt_pos and start_at date when config_id field changes."""
+        config_obj = self.pool.get('pos.config')
+        config = config_obj.browse(cr, uid, config_id, context=context)
+        start_at = strftime('%Y-%m-%d %H:%M:%S')
+        is_nrt_pos = False
+
+        if config.nrt:
+            is_nrt_pos = True
+        return {'value':
+                {
+                    'start_at': start_at,
+                    'is_nrt_pos': is_nrt_pos,
+                }
+               }
+
+    def _is_nrt_pos(self, cr, uid, ids, name, args, context=None):
+        """Return True if the current session refers to a nrt pos."""
+        res = {}
+        sessions = self.browse(cr, uid, ids, context=context)
+        config_obj =  self.pool.get('pos.config')
+        for session in sessions:
+            res[session.id] = False
+            if session.config_id.nrt:
+                res[session.id] = True
+        return res
+
     _columns = {
         'start_at': fields.datetime('Opening date'),
         'stop_at': fields.datetime('Closing date'),
+        'is_nrt_pos': fields.function(_is_nrt_pos,
+                            method=True,
+                            string='Is this session linked to a nrt pos?',
+                            type='boolean',
+                            store=False,),
+    }
+
+    _defaults = {
+        'start_at': strftime('%Y-%m-%d %H:%M:%S'),
     }
 
     def create(self, cr, uid, values, context=None):
@@ -49,9 +120,10 @@ class pos_session(orm.Model):
         id = super(pos_session, self).create(cr, uid, values, context=context)
         for session in self.browse(cr, uid, [id], context=context):
             default_date = session.start_at
-        bank_st_ids = bank_st_obj.search(cr, uid, [('pos_session_id', '=', id)], context=context)
-        period = period_obj.find(cr, uid, dt=default_date, context=dict(context or {}, account_period_prefer_normal=True))[0]
-        bank_st_obj.write(cr, uid, bank_st_ids, {'date': default_date, 'period_id': period}, context=context)
+            if session.config_id.nrt:
+                bank_st_ids = bank_st_obj.search(cr, uid, [('pos_session_id', '=', id)], context=context)
+                period = period_obj.find(cr, uid, dt=default_date, context=dict(context or {}, account_period_prefer_normal=True))[0]
+                bank_st_obj.write(cr, uid, bank_st_ids, {'date': default_date, 'period_id': period}, context=context)
         return id
 
     def _confirm_orders(self, cr, uid, ids, context=None):
@@ -62,18 +134,18 @@ class pos_session(orm.Model):
         super(pos_session, self)._confirm_orders(cr, uid, ids, context=context)
 
         for session in self.browse(cr, uid, ids, context=context):
-            account_move_ids = []
-            order_ids = [order.id for order in session.order_ids if order.state == 'done']
-            if order_ids:
-                company_id = order_obj.browse(cr, uid, order_ids, context=context)[0].company_id.id
-            else:
-                company_id = self.pool.get('res.users').browse(cr, uid, [uid], context=context)[0].company_id.id
-            for order in order_obj.browse(cr, uid, order_ids, context=context):
-                if order.account_move.id not in account_move_ids:
-                    account_move_ids.append(order.account_move.id)
-            period = period_obj.find(cr, uid, dt=session.start_at, context=dict(context or {}, company_id=company_id, account_period_prefer_normal=True))[0]
-            account_move_obj.write(cr, uid, account_move_ids, {'date': session.start_at, 'period_id': period}, context=context)
-
+            if session.config_id.nrt:
+                account_move_ids = []
+                order_ids = [order.id for order in session.order_ids if order.state == 'done']
+                if order_ids:
+                    company_id = order_obj.browse(cr, uid, order_ids, context=context)[0].company_id.id
+                else:
+                    company_id = self.pool.get('res.users').browse(cr, uid, [uid], context=context)[0].company_id.id
+                for order in order_obj.browse(cr, uid, order_ids, context=context):
+                    if order.account_move.id not in account_move_ids:
+                        account_move_ids.append(order.account_move.id)
+                period = period_obj.find(cr, uid, dt=session.start_at, context=dict(context or {}, company_id=company_id, account_period_prefer_normal=True))[0]
+                account_move_obj.write(cr, uid, account_move_ids, {'date': session.start_at, 'period_id': period}, context=context)
         return True
 
     def wkf_action_closing_control(self, cr, uid, ids, context=None):
@@ -82,7 +154,8 @@ class pos_session(orm.Model):
         """
         res = super(pos_session, self).wkf_action_closing_control(cr, uid, ids, context=context)
         for session in self.browse(cr, uid, ids, context=context):
-            self.write(cr, uid, [session.id], {'stop_at': session.start_at}, context=context)
+            if session.config_id.nrt:
+                self.write(cr, uid, [session.id], {'stop_at': session.start_at}, context=context)
         return res
 
     def onchange_start_at(self, cr, uid, ids, start_at, context=None):
@@ -91,38 +164,41 @@ class pos_session(orm.Model):
             return {'value': {'stop_at': start_at}}
         else:
             for session in self.browse(cr, uid, ids, context=context):
-                return {'value': {'stop_at': start_at}}
+                if session.config_id.nrt:
+                    return {'value': {'stop_at': start_at}}
 
 
 class pos_order(orm.Model):
     _inherit = 'pos.order'
 
     def create(self, cr, uid, vals, context=None):
-        session_obj = self.pool.get('pos.session')
-        vals.update({'date_order': session_obj.browse(cr, uid, vals['session_id'], context=context).start_at})
+        session = self.pool.get('pos.session').browse(cr, uid, vals['session_id'], context=context)
+        if session.config_id.nrt:
+            vals.update({'date_order': session.start_at})
         res = super(pos_order, self).create(cr, uid, vals, context=context)
         return res
 
-
     def write(self, cr, uid, ids, vals, context=None):
-        session_obj = self.pool.get('pos.session')
         for order in self.browse(cr, uid, ids, context=context):
-            default_date_order = order.session_id.start_at
-        vals.update({'date_order': default_date_order})
+            if order.session_id.config_id.nrt:
+                default_date_order = order.session_id.start_at
+                vals.update({'date_order': default_date_order})
         res = super(pos_order, self).write(cr, uid, ids, vals, context=context)
         return res
 
     def onchange_session_id(self, cr, uid, ids, session_id, context=None):
         """Define date order by start date of current session"""
-        res_obj = self.pool.get('pos.session')
-        data = {}
+        session_obj = self.pool.get('pos.session')
+        default_date_order = strftime('%Y-%m-%d %H:%M:%S')
+
         if not session_id:
-            return {'value': data}
+            return {'value': {}}
         else:
-            for session in res_obj.browse(cr, uid, [session_id],
+            for session in session_obj.browse(cr, uid, [session_id],
                                           context=context):
-                default_date_order = session.start_at
-            return {'value': {'date_order': default_date_order}}
+                if session.config_id.nrt:
+                    default_date_order = session.start_at
+        return {'value': {'date_order': default_date_order}}
 
     def add_payment(self, cr, uid, order_id, data, context=None):
         """Define payment date by the order date"""
@@ -130,11 +206,13 @@ class pos_order(orm.Model):
         statement_ids = []
         order = self.browse(cr, uid, [order_id], context=context)[0]
         date_order = order.date_order
+
         super(pos_order, self).add_payment(cr, uid, order_id, data, context=context)
 
-        for statement in order.statement_ids:
-            statement_ids.append(statement.id)
-        statement_obj.write(cr, uid, statement_ids, {'date': date_order}, context=context)
+        if order.session_id.config_id.nrt:
+            for statement in order.statement_ids:
+                statement_ids.append(statement.id)
+            statement_obj.write(cr, uid, statement_ids, {'date': date_order}, context=context)
         return order_id
 
 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: