[XML][PYTHON][TRANS] +partner county_code field
[burette/lhc.git] / lhc.py
diff --git a/lhc.py b/lhc.py
index 420ce97..de5776a 100644 (file)
--- a/lhc.py
+++ b/lhc.py
@@ -30,9 +30,35 @@ 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', help='This contact is a usual contact for L\'Heureux Cyclage employees. This field can be used to discriminated contacts for differents usages.'),
+        'usual_contact': fields.boolean(
+            'Usual contact',
+            help="""This contact is a usual contact for L\'Heureux Cyclage
+            employees. This field can be used to discriminated contacts for
+            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 = {
@@ -44,7 +70,13 @@ class product_template(orm.Model):
     _inherit = 'product.template'
 
     _columns = {
-        'standard_price': fields.float('Cost', digits_compute=dp.get_precision('Product Price'), help="Cost price of the product used for standard stock valuation in accounting and used as a base price on purchase orders.", groups="base.group_user,lhc.group_volunteer"),
+        'standard_price': fields.float(
+            'Cost',
+            digits_compute=dp.get_precision('Product Price'),
+            help="""Cost price of the product used for standard stock valuation
+            in accounting and used as a base price on purchase orders.",
+            groups="base.group_user,lhc.group_volunteer"""
+        ),
     }
 
 
@@ -56,8 +88,6 @@ class res_users(orm.Model):
         v = {}
         partners = self.pool.get('res.partner')
         partner = partners.browse(cr, uid, partner_id)
-        import pdb
-        pdb.set_trace()
         if partner.email:
             v['email'] = partner.email
         else:
@@ -65,6 +95,18 @@ class res_users(orm.Model):
         return {'value': v}
 
 
+class event_event(orm.Model):
+    _inherit = 'event.event'
+
+    _columns = {
+        'duration': fields.float(
+            'Duration',
+            digits_compute=dp.get_precision('Product Unit of Measure'),
+            help='Duration in hours'
+        ),
+    }
+
+
 class event_registration(orm.Model):
     _inherit = 'event.registration'
 
@@ -86,6 +128,36 @@ class event_registration(orm.Model):
             ('pole_emploi', 'Pole Emploi')],
             'Main funding', help='Main funding origin'),
         'fundings_others': fields.char('Others fundings origins', size=128),
+        'sale_order_ids': fields.many2many(
+            'sale.order',
+            'event_registration_sale_order_rel',
+            'event_registration_id',
+            'sale_order_id',
+            'Related sale order(s)'
+        ),
+        'invoice_ids': fields.many2many(
+            'account.invoice',
+            'event_registration_invoice_id_rel',
+            'event_registration_id',
+            'invoice_id',
+            'Related invoice(s)'
+        ),
     }
 
+    def onchange_contact_id(self, cr, uid, ids, contact, partner, context=None):
+        """Concat phone with mobile phone if exist. If mobile exist and not
+        phone, add mobile number"""
+        vals = super(event_registration, self).onchange_contact_id(cr, uid, ids, contact, partner, context)
+        addr_obj = self.pool.get('res.partner')
+        contact_id = addr_obj.browse(cr, uid, contact, context=context)
+        phone = vals['value']['phone']
+        mobile = contact_id.mobile
+        if mobile:
+            if phone:
+                vals['value']['phone'] = '%s - %s' % (phone, mobile)
+            else:
+                vals['value']['phone'] = mobile
+        return vals
+
+
 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: