[XML][PYTHON][TRANS] +partner county_code field

[burette/lhc.git] / lhc.py

diff --git a/lhc.py b/lhc.py
index 528034f..de5776a 100644 (file)
--- a/lhc.py
+++ b/lhc.py
@@ -30,6 +30,16 @@ import openerp.addons.decimal_precision as dp
 class res_partner(orm.Model):
     _inherit = 'res.partner'
 
+    def _get_county(self, cr, uid, ids, name, args, context=None):
+        """docstring for _county_get_fnc"""
+        res = {}
+        for obj in self.browse(cr, uid, ids, context=context):
+            if obj.zip:
+                res[obj.id] = obj.zip[0:2]
+            else:
+                res[obj.id] = ''
+        return res
+
     _columns = {
         'usual_contact': fields.boolean(
             'Usual contact',
@@ -38,6 +48,17 @@ class res_partner(orm.Model):
             differents usages."""
         ),
         'kit_sent': fields.boolean('Welcome kit sent'),
+        'county_code': fields.function(
+            _get_county,
+            method=True,
+            string='County code',
+            type='char',
+            size=2,
+            store={
+                'res.partner': (
+                    lambda self, cr, uid, ids, c={}: ids,
+                    ['zip'], 10)},
+        ),
     }
 
     _defaults = {