Source

trytond-stock_origin_sale / sale.py

Diff from to

File sale.py

 #This file is part stock_origin_purchase 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.model import ModelView, ModelSQL, fields
-from trytond.transaction import Transaction
-from trytond.pool import Pool
+__all__ = ['Sale']
+__metaclass__ = PoolMeta
 
-class Sale(ModelSQL, ModelView):
-    _name = 'sale.sale'
+class Sale:
+    'Sale'
+    __name__ = 'sale.sale'
 
-    def create_shipment(self, sale, shipment_type):
-        shipments = super(Sale, self).create_shipment(sale, shipment_type)
+    def create_shipment(self, shipment_type):
+        shipments = super(Sale, self).create_shipment(shipment_type)
         if shipment_type == 'out':
-            shipment_obj = Pool().get('stock.shipment.out')
+            Shipment = Pool().get('stock.shipment.out')
         elif shipment_type == 'return':
-            shipment_obj = Pool().get('stock.shipment.out.return')
+            Shipment = Pool().get('stock.shipment.out.return')
         else:
             return None
 
         if shipments:
             for shipment in shipments:
                 values = {
-                    'origin': 'sale.sale,%s' % (sale.id),
+                    'origin': 'sale.sale,%s' % (self.id),
                 }
-                shipment_id = shipment_obj.write([shipment], values)
+                Shipment.write([shipment], values)
+        return shipments
 
-Sale()