diff --git a/easy_my_coop_eater/models/coop.py b/easy_my_coop_eater/models/coop.py index bc48df3..34ac9b5 100644 --- a/easy_my_coop_eater/models/coop.py +++ b/easy_my_coop_eater/models/coop.py @@ -1,27 +1,28 @@ # -*- coding: utf-8 -*- -from openerp import api, fields, models +from openerp import api, models + + +class SubscriptionRequest(models.Model): + _inherit = 'subscription.request' -class subscription_request(models.Model): - _inherit='subscription.request' - def get_eater_vals(self, partner, share_product_id): vals = {} eater = share_product_id.eater - + if partner.is_company or partner.age < 18: eater = 'eater' - + vals['eater'] = eater - + return vals - + @api.one def validate_subscription_request(self): - invoice = super(subscription_request, self).validate_subscription_request()[0] + invoice = super(SubscriptionRequest, self).validate_subscription_request()[0] partner = invoice.partner_id - + vals = self.get_eater_vals(partner, self.share_product_id) partner.write(vals) - - return invoice \ No newline at end of file + + return invoice diff --git a/easy_my_coop_eater/models/product.py b/easy_my_coop_eater/models/product.py index f190628..a19225f 100644 --- a/easy_my_coop_eater/models/product.py +++ b/easy_my_coop_eater/models/product.py @@ -1,7 +1,10 @@ # -*- coding: utf-8 -*- -from openerp import api, fields, models, _ +from openerp import fields, models + class ProductTemplate(models.Model): _inherit = 'product.template' - - eater = fields.Selection([('eater', 'Eater'), ('worker_eater', 'Worker and Eater')], string="Eater/Worker") \ No newline at end of file + + eater = fields.Selection([('eater', 'Eater'), + ('worker_eater', 'Worker and Eater')], + string="Eater/Worker")