vertical-cooperative/easy_my_coop/wizard/create_subscription_from_partner.py

# -*- coding: utf-8 -*-

import time

from openerp import api, fields, models, _
import openerp.addons.decimal_precision as dp
from openerp.exceptions import UserError

class PartnerCreateSubscription(models.TransientModel):
    _name = "partner.create.subscription"
    _description = "Create Subscription From Partner"

    @api.multi
    @api.onchange('share_product')
    def on_change_share_type(self):
        self.share_qty = self.share_product.minimum_quantity
        
    @api.model
    def _default_product_id(self):
        domain = [('is_share','=',True),('default_share_product','=',True)]
        active_id = self.env.context.get('active_id')
        if active_id:
            partner = self.env['res.partner'].browse(active_id)
            if partner.is_company:
                domain.append(('by_company','=',True))
            else:
                domain.append(('by_individual','=',True))
        
        return self.env['product.product'].search(domain)[0]
    
    def _get_representative(self):
        partner = self._get_partner()
        if partner.is_company:
            return self.env['res.partner'].search([('parent_id','=',partner.id),
                                                    ('representative','=',True)])
        return False
    
    @api.model
    def _get_representative_number(self):
        representative = self._get_representative()
        if representative:
            return representative.national_register_number
        return False
    
    @api.model
    def _get_representative_name(self):
        representative = self._get_representative()
        if representative:
            return representative.name
        return False
    
    @api.model
    def _get_partner(self):
        active_id = self.env.context.get('active_id')
        return self.env['res.partner'].browse(active_id)
    
    @api.model
    def _get_is_company(self):
        return self._get_partner().is_company
    
    @api.model
    def _get_email(self):
        partner = self._get_partner()
        return partner.email

    @api.model
    def _get_register_number(self):
        partner = self._get_partner()
        if partner.is_company:
            return partner.company_register_number
        else:
            return partner.national_register_number
    
    @api.model
    def _get_bank_account(self):
        partner = self._get_partner()
        if len(partner.bank_ids) > 0:
            return partner.bank_ids[0].acc_number
        return None
        
    @api.model
    def _get_possible_share(self):
        domain = [('is_share','=',True)]
        partner = self._get_partner()
        if partner.is_company:
            domain.append(('by_company','=',True))
        else:
            domain.append(('by_individual','=',True))
    
        return domain
    
    @api.multi
    @api.depends('share_product', 'share_qty')
    def _compute_subscription_amount(self):
        for sub_request in self:
            sub_request.subscription_amount = sub_request.share_product.list_price * sub_request.share_qty
            
    is_company = fields.Boolean(String="Is company?", default=_get_is_company)
    cooperator = fields.Many2one('res.partner', string="Cooperator", default=_get_partner)
    register_number = fields.Char(string="Register Number", required=True, default=_get_register_number)
    email = fields.Char(string="Email", required=True, default=_get_email)
    bank_account = fields.Char(string="Bank account", required=True, default=_get_bank_account)
    share_product = fields.Many2one('product.product', string='Share Type', domain=lambda self: self._get_possible_share(),\
        default=_default_product_id, required=True)
    share_qty = fields.Integer(string="Share Quantity", required=True)
    share_unit_price = fields.Float(related='share_product.list_price', string='Share price', readonly=True)
    subscription_amount = fields.Float(compute='_compute_subscription_amount', string='Subscription amount', digits=dp.get_precision('Account'), readonly=True)
    representative_name = fields.Char(string='Representative name', default=_get_representative_name)
    representative_number = fields.Char(string='Representative national register number', default=_get_representative_number)
    
    def check_belgian_ident_id(self, register_number):
        if self.env['subscription.request'].check_belgian_identification_id(register_number):
            return True
        else:
            raise UserError(_("The national register number is not valid."))

    @api.multi
    def create_subscription(self):
        sub_req = self.env['subscription.request']
        partner_obj = self.env['res.partner']
        
        cooperator = self.cooperator
        vals = {'partner_id': cooperator.id,
                'share_product_id': self.share_product.id,
                'ordered_parts': self.share_qty,
                'user_id': self.env.uid,
                'email': self.email,
                'source': 'crm'}
        
        if self.is_company:
            vals['company_name'] = cooperator.name
            vals['name'] = '/'
            vals['company_register_number'] = self.register_number
            vals['is_company'] = True
        else:
            vals['name'] = cooperator.name
            vals['no_registre'] = self.register_number
        
        coop_vals = {}
        if not self._get_email():
            coop_vals['email'] = self.email
        
        if not self._get_register_number():
            if self.is_company:
                coop_vals['company_register_number'] = self.register_number
            else:
                if self.check_belgian_ident_id(self.register_number):
                    coop_vals['national_register_number'] = self.register_number
        
        if self.is_company and not self._get_representative():
            representative = False
            if self.representative_number:
                representative_number = self.representative_number
                representative = partner_obj.search([('national_register_number','=',representative_number)])
            
            if representative:
                if len(representative) > 1:
                    raise UserError(_('There is two different persons with the same national register number. Please proceed to a merge before to continue'))
                if representative.parent_id:
                    raise UserError(_("A person can't be representative of two differents companies."))
                representative.parent_id = cooperator.id
            else:
                if self.check_belgian_ident_id(representative_number):
                    represent_vals = {'name':self.representative_name,'cooperator':True,
                                      'national_register_number':representative_number,
                                      'parent_id':cooperator.id,'representative':True}
                    partner_obj.create(represent_vals)
                
        if not self._get_bank_account():
            partner_bank = self.env['res.partner.bank']
            partner_bank.create({'partner_id':cooperator.id,
                                 'acc_number':self.bank_account})
        vals['iban'] = self.bank_account
            
        if coop_vals:
            cooperator.write(coop_vals)

        new_sub_req = sub_req.create(vals)
        
        return {
            'type': 'ir.actions.act_window',
            'view_type': 'form, tree',
            'view_mode': 'form',
            'res_model': 'subscription.request',
            'res_id': new_sub_req.id,
            'target': 'current',
        }