[FIX]Context values missing

[IMP]Add the availability to choose the company that will invoice

stock_picking_invoicing/models/__init__.py

@@ -3,4 +3,4 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
 from . import stock
-from . import invoice
+from . import account_invoice

stock_picking_invoicing/models/stock.py

@@ -100,7 +100,7 @@ def _get_master_data(self, move, company):
     def _get_taxes(self, fiscal_position):
         self.ensure_one()
         taxes_ids = self.product_id.taxes_id        
-        my_taxes = taxes_ids.filtered(lambda r: r.company_id.id == self.company_id.id)    
+        my_taxes = taxes_ids.filtered(lambda r: r.company_id.id == self.env.context['force_company'])    
         my_taxes = fiscal_position.map_tax(my_taxes)
         my_taxes = [t.id for t in my_taxes]
 
@@ -274,6 +274,7 @@ def _create_invoice_from_picking(self, picking, vals):
         ''' This function simply creates the invoice from the given values. It is overriden in delivery module to add the delivery costs.
         '''
         invoice_obj = self.env['account.invoice']
+        _logger.debug("Create invoice with %s" % vals)
         return invoice_obj.create(vals)
 
     @api.model
@@ -319,21 +320,24 @@ def _invoice_create_line(self, moves, journal_id, inv_type='out_invoice'):
 
         is_extra_move, extra_move_tax = move_obj._get_moves_taxes(moves, inv_type)
         product_price_unit = {}
-
+        _logger.debug("COntexte %s" % self.env.context)
         invoice_id = None
+        company_id  =  self.env.context['force_company'] or self.env['res.users']._get_company() 
         for move in moves:
-            company = move.company_id
-            origin = move.picking_id.name
+            company = self.env['res.company'].search([('id', '=', company_id)])
+            origin = move.sudo().picking_id.name
             partner, user_id, currency_id = move_obj._get_master_data(move, company)
             key = (partner, currency_id, company.id, user_id)
-
+            _logger.debug("contexte %s" % self.env.context)
             invoice_vals = self._get_invoice_vals(key, inv_type, journal_id, move)
             if key not in invoices:
                 # Get account and payment terms
+                _logger.debug("Add invoice to list")
                 invoice_id = self._create_invoice_from_picking(move.picking_id, invoice_vals)
                 invoices[key] = invoice_id.id
-
+                _logger.debug("Add invoice to list after key")
             else:
+                _logger.debug("Add found invoice to list")
                 invoice = invoice_obj.search([('id', '=', invoices[key])])
                 merge_vals = {}
                 if not invoice.origin or invoice_vals['origin'] not in invoice.origin.split(', '):
@@ -344,10 +348,11 @@ def _invoice_create_line(self, moves, journal_id, inv_type='out_invoice'):
                     invoice_name = filter(None, [invoice.name, invoice_vals['name']])
                     merge_vals['name'] = ', '.join(invoice_name)
                 if merge_vals:
+                    _logger.debug("Merge vues %s" % merge_vals)
                     invoice.write(merge_vals)
 
             move.with_context(
-                              fp_id=invoice_vals.get('fiscal_position_id', False)
+                              fp_id=invoice_vals.get('fiscal_position_id', False),
                               )
 
             invoice_line_vals = move._get_invoice_line_vals(partner, inv_type)
@@ -368,6 +373,7 @@ def _invoice_create_line(self, moves, journal_id, inv_type='out_invoice'):
                     invoice_line_vals['invoice_line_tax_id'] = extra_move_tax[0, move.product_id]
 
             invoice_line = move._create_invoice_line_from_vals(invoice_line_vals)
+            _logger.debug("Before invoicing write")
             if invoice_line :
                 move.write({'invoice_line_id': invoice_line.id,
                             'invoice_state': 'invoiced'

stock_picking_invoicing/views/stock_view.xml

@@ -34,7 +34,7 @@
                     <button name="%(action_stock_invoice_onshipping)d" string="Create Invoice" attrs="{'invisible': ['|',('state','&lt;&gt;','done'),('invoice_state','&lt;&gt;','2binvoiced')]}" type="action" class="oe_highlight" groups="base.group_user"/>
                 </xpath>
                 <field name="move_lines" position="attributes">
-                    <attribute name="context">{'default_invoice_state': invoice_state, 'address_in_id': partner_id, 'form_view_ref':'stock.view_move_picking_form', 'tree_view_ref':'stock.view_move_picking_tree', 'default_picking_type_id': picking_type_id,'default_picking_id': active_id}</attribute>
+                    <attribute name="context">{'default_invoice_state': invoice_state, 'address_in_id': partner_id, 'form_view_ref':'stock.view_move_picking_form', 'tree_view_ref':'stock.view_move_picking_tree', 'default_picking_type_id': picking_type_id,'default_picking_id': active_id, 'default_location_id': location_id, 'default_location_dest_id': location_dest_id}</attribute>
                 </field>
                 <xpath expr="//field[@name='move_type']" position="after">
                     <field name="invoice_state" groups="account.group_account_invoice"/>

stock_picking_invoicing/wizard/stock_invoice_onshipping.py

@@ -110,12 +110,14 @@ def _get_journal_type(self):
         context = self.env.context or {}                        
         res_ids = context and context.get('active_ids', [])
         pick_obj = self.env['stock.picking']
-        pickings = pick_obj.search([('id', 'in', res_ids)])
-        pick = pickings and pickings[0]
-        if not pick or not pick.move_lines:
+        pickings = pick_obj.sudo().search([('id', 'in', res_ids)])
+        pick = pickings and pickings[0]        
+        if not pick or not pick.sudo().move_lines:
             return 'sale'
-        type = pick.picking_type_id.code
-        usage = pick.move_lines[0].location_id.usage if type == 'incoming' else pick.move_lines[0].location_dest_id.usage
+        type = pick.sudo().picking_type_id.code
+        usage = pick.sudo().move_lines[0].location_id.usage \
+                        if type == 'incoming'  \
+                        else pick.sudo().move_lines[0].location_dest_id.usage
 
         return JOURNAL_TYPE_MAP.get((type, usage), ['sale'])[0]
 
@@ -156,7 +158,10 @@ def _get_journal_type(self):
     show_purchase_journal = fields.Boolean(string="Show Purchase Journal")
     show_purchase_refund_journal = fields.Boolean(
         string="Show Refund Purchase Journal")
-
+
+    company_id = fields.Many2one(comodel_name="res.company", 
+                                string='Company to invoice',
+                                default=lambda self:self.env.user.company_id.id)
 
     #=================
     # Business part
@@ -207,10 +212,11 @@ def create_invoice(self):
             pickings  =   self.env['stock.picking'].search([('id', 'in', active_ids)])
             pickings.set_invoiced()
 
-
-        self = self.with_context(
+        force_company_id = self.company_id.id or self.env['res.users']._get_company()
+        picking_pool = picking_pool.with_context(
                 date_inv = self.invoice_date,
-                inv_type = inv_type
+                inv_type = inv_type,
+                force_company=force_company_id
                )
 
         res = picking_pool.action_invoice_create(active_ids,
@@ -258,14 +264,14 @@ def create_invoice(self):
     def get_partner_sum(self, pickings, partner, inv_type, picking_type,
                         usage):
         move_obj = self.env['stock.move']
-        pickings = pickings.filtered(lambda x: x.picking_type_id.code ==
+        pickings = pickings.filtered(lambda x: x.sudo().picking_type_id.code ==
                                      picking_type and x.partner_id == partner)
         if picking_type == 'outgoing':
-            moves = pickings.mapped('move_lines').filtered(
-                lambda x: x.location_dest_id.usage == usage)
+            moves = pickings.sudo().mapped('move_lines').filtered(
+                lambda x: x.sudo().location_dest_id.usage == usage)
         else:
-            moves = pickings.mapped('move_lines').filtered(
-                lambda x: x.location_id.usage == usage)
+            moves = pickings.sudo().mapped('move_lines').filtered(
+                lambda x: x.sudo().location_id.usage == usage)
         return (sum([(m._get_price_unit_invoice( inv_type) *
                       m.product_uom_qty) for m in moves]),
                 moves.mapped('picking_id'))
@@ -300,18 +306,19 @@ def get_split_pickings_grouped(self, pickings):
     @api.multi
     def get_split_pickings_nogrouped(self, pickings):
         sale_pickings = pickings.filtered(
-            lambda x: x.picking_type_id.code == 'outgoing' and
-            x.move_lines[:1].location_dest_id.usage == 'customer')
+            lambda x: x.sudo().picking_type_id.code == 'outgoing' and
+            x.sudo().move_lines[:1].location_dest_id.usage == 'customer')
         # use [:1] instead of [0] to avoid a errors on empty pickings
         sale_refund_pickings = pickings.filtered(
-            lambda x: x.picking_type_id.code == 'incoming' and
-            x.move_lines[:1].location_id.usage == 'customer')
+            lambda x: x.sudo().picking_type_id.code == 'incoming' and
+            x.sudo().move_lines[:1].location_id.usage == 'customer')
         purchase_pickings = pickings.filtered(
-            lambda x: x.picking_type_id.code == 'incoming' and
-            x.move_lines[:1].location_id.usage == 'supplier')
+            lambda x: x.sudo().picking_type_id.code == 'incoming' and
+            x.sudo().move_lines[:1].location_id.usage == 'supplier')
         purchase_refund_pickings = pickings.filtered(
-            lambda x: x.picking_type_id.code == 'outgoing' and
-            x.move_lines[:1].location_dest_id.usage == 'supplier')
+            lambda x: x.sudo().picking_type_id.code == 'outgoing' and
+            x.sudo().move_lines[:1].location_dest_id.usage == 'supplier')
+
         return (sale_pickings, sale_refund_pickings, purchase_pickings,
                 purchase_refund_pickings)
 

stock_picking_invoicing/wizard/stock_invoice_onshipping_view.xml

@@ -22,9 +22,11 @@
                         <field name="sale_refund_journal" attrs="{'invisible':[('show_sale_refund_journal', '=', False)], 'required':[('show_sale_refund_journal', '=', True)]}"/>
                         <field name="purchase_journal" attrs="{'invisible':[('show_purchase_journal', '=', False)], 'required':[('show_purchase_journal', '=', True)]}"/>
                         <field name="purchase_refund_journal" domain="[('type', '=', 'purchase_refund')]" attrs="{'invisible':[('show_purchase_refund_journal', '=', False)], 'required':[('show_purchase_refund_journal', '=', True)]}"/>
+                        <field name="company_id" invisible="0" required="1"/>
                         <field name="group"/>
                         <field name="invoice_date" />
                         <field name="invoice_force" />
+
                     </group>
                     <footer>
                         <button name="open_invoice" string="Create" type="object" class="oe_highlight"/>