Added state on purchase request

purchase_request.py

@@ -27,6 +27,17 @@ class PurchaseRequest(OSV):
         'purchase.line', 'Purchase Line',readonly=True)
     company = fields.Many2One(
         'company.company', 'Company', required=True, readonly=True)
+    state = fields.Function(
+        'get_state', type='selection',
+        selection=[("running", "Running"),
+                   ("done", "Done"),
+                   ("draft", "Draft"),
+                   ("cancel", "Cancel")],
+        string="State", readonly=True)
+
+    def __init__(self):
+        super(PurchaseRequest, self).__init__()
+        self._order[0] = ('id', 'DESC')
 
     def default_company(self, cursor, user, context=None):
         company_obj = self.pool.get('company.company')
@@ -37,6 +48,18 @@ class PurchaseRequest(OSV):
                     context=context)[0]
         return False
 
+    def get_state(self, cursor, user, ids, name, args, context=None):
+        res = {}.fromkeys(ids, 'draft')
+        for request in self.browse(cursor, user, ids, context=context):
+            if request.purchase_line:
+                if request.purchase_line.purchase.state == 'cancel':
+                    res[request.id] = 'cancel'
+                if request.purchase_line.purchase.state == 'done':
+                    res[request.id] = 'done'
+            else:
+                res[request.id] = 'running'
+        return res
+
     def generate_requests(self, cursor, user, context=None):
         """
         For each product compute the purchase request that must be

purchase_request.xml

@@ -28,6 +28,8 @@
                         <field name="purchase_line"/>
                         <label name="stock_level"/>
                         <field name="stock_level"/>
+                        <label name="state"/>
+                        <field name="state"/>
                     </form>
                 ]]>
             </field>
@@ -39,15 +41,14 @@
                 <![CDATA[
                     <tree string="Locations">
                         <field name="product" select="1"/>
-                        <field name="party" select="1"/>
                         <field name="quantity" select="2"/>
                         <field name="uom" select="2"/>
                         <field name="warehouse" select="2"/>
-                        <field name="purchase_date"/>
-                        <field name="supply_date"/>
-                        <field name="purchase_line"/>
+                        <field name="party" select="1"/>
+                        <field name="purchase_date" select="2"/>
                         <field name="stock_level"/>
-                        <field name="company"/>
+                        <field name="company" select="2"/>
+                        <field name="state"/>
                     </tree>
                 ]]>
             </field>