Create Sale Return Wizard

__init__.py

@@ -3,8 +3,12 @@
 #the full copyright notices and license terms.
 from trytond.pool import Pool
 from .sale import *
+from .shipment import *
 
 def register():
     Pool.register(
         Sale,
         module='sale_shipment_returns', type_='model')
+    Pool.register(
+        CreateSaleReturn,
+        module='sale_shipment_returns', type_='wizard')

shipment.py

@@ -0,0 +1,88 @@
+#This file is part sale_shipment_returns module for Tryton.
+#The COPYRIGHT file at the top level of this repository contains 
+#the full copyright notices and license terms.
+from trytond.pool import Pool, PoolMeta
+from trytond.wizard import Wizard, StateAction
+from trytond.transaction import Transaction
+
+__all__ = ['CreateSaleReturn']
+__metaclass__ = PoolMeta
+
+
+class CreateSaleReturn(Wizard):
+    'Create Sale from Customer Return Shipment'
+    __name__ = 'stock.sale.return.create'
+    start = StateAction('sale.act_sale_form')
+
+    @classmethod
+    def __setup__(cls):
+        super(CreateSaleReturn, cls).__setup__()
+        cls._error_messages.update({
+                'shipment_done_title': 'You can not create return sale',
+                'shipment_done_msg': ('The return shipment with code "%s" is not yet '
+                    'sent.'),
+                'shipment_description': ('Shipment Out Return "%s"'),
+                })
+
+    def do_start(self, action):
+        pool = Pool()
+        ShipmentOutReturn = pool.get('stock.shipment.out.return')
+        Sale = pool.get('sale.sale')
+        SaleLine = pool.get('sale.line')
+
+        shipment_ids = Transaction().context['active_ids']
+        shipment_out_returns = ShipmentOutReturn.browse(shipment_ids)
+
+        sales = []
+        for shipment_out_return in shipment_out_returns:
+            if shipment_out_return.state != 'done':
+                self.raise_user_error('shipment_done_title',
+                        error_description='shipment_done_msg',
+                        error_description_args=shipment_out_return.code)
+
+            party = shipment_out_return.customer
+            description = self.raise_user_error('shipment_description',
+                (shipment_out_return.code),raise_exception=False)
+
+            # create sale
+            sale_vals = Sale.get_sale_data(party, description)
+            sale = Sale(**sale_vals)
+            sale.save()
+            sales.append(sale)
+
+            # create sale lines from moves
+            lines_to_create = []
+            for move in shipment_out_return.incoming_moves:
+                product = move.product
+                quantity = -move.quantity
+                uom = move.uom.symbol
+                vals = SaleLine.get_sale_line_data(
+                    sale, product, quantity, uom)
+                lines_to_create.append(vals)
+
+            lines = SaleLine.create(lines_to_create)
+
+            # write stock move origin
+            lines_to_move = []
+            if lines:
+                for line in lines:
+                    lines_to_move.append({
+                        'product': line.product,
+                        'quantity': abs(line.quantity),
+                        'id': line.id,
+                        })
+            if lines_to_move:
+                for move in shipment_out_return.incoming_moves:
+                    for l2m in lines_to_move:
+                        if l2m['product'] == move.product and \
+                                l2m['quantity'] == move.quantity:
+                            move.origin = 'sale.line,%s' % l2m['id']
+                            move.save()
+
+        data = {'res_id': [x.id for x in sales]}
+        if len(sales) == 1:
+            action['views'].reverse()
+        return action, data
+
+    def transition_start(self):
+        return 'end'

shipment.xml

@@ -0,0 +1,19 @@
+<?xml version="1.0"?>
+<!-- This file is part sale_shipment_returns module for Tryton.
+The COPYRIGHT file at the top level of this repository contains the full copyright notices and license terms. -->
+<tryton>
+    <data>
+        <record model="ir.action.wizard" id="create_sale_return">
+            <field name="name">Create Return Sale</field>
+            <field name="wiz_name">stock.sale.return.create</field>
+            <field name="model">stock.shipment.out.return</field>
+        </record>
+        <record model="ir.action.keyword" id="create_sale_return_keyword">
+            <field name="keyword">form_action</field>
+            <field name="model">stock.shipment.out.return,-1</field>
+            <field name="action" ref="create_sale_return"/>
+        </record>
+    </data>
+</tryton>
+
+

tryton.cfg

@@ -3,7 +3,8 @@ version=3.0.0
 depends:
     ir
     res
-    sale
+    sale_data
     stock_origin
 xml:
     sale.xml
+    shipment.xml