vertical-cooperative/easy_my_coop/wizard/create_subscription_from_partner.py

# -*- coding: utf-8 -*-
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 partner.search([('parent_id', '=', partner.id),
                                   ('representative', '=', True)])
        return False

    @api.model
    def _get_representative_email(self):
        representative = self._get_representative()
        if representative:
            return representative.email
        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

    @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 Company Number",
                                  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=_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_email = fields.Char(string='Representative email',
                                       default=_get_representative_email)

    @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,
                'cooperator': True,
                'share_product_id': self.share_product.id,
                'ordered_parts': self.share_qty,
                'user_id': self.env.uid,
                'email': self.email,
                'source': 'crm',
                'address': self.cooperator.street,
                'zip_code': self.cooperator.zip,
                'city': self.cooperator.city,
                'country_id': self.cooperator.country_id.id
                }

        if self.is_company:
            vals['company_name'] = cooperator.name
            vals['company_email'] = cooperator.email
            vals['name'] = '/'
            vals['company_register_number'] = self.register_number
            vals['is_company'] = True
        else:
            vals['name'] = cooperator.name

        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

        if self.is_company and not self._get_representative():
            representative = False
            if self.representative_email:
                representative = partner_obj.search(
                    [('email', '=', self.representative_email)])

            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 different companies."))
                representative.parent_id = cooperator.id
            else:
                if self.representative_email:
                    represent_vals = {'name': self.representative_name,
                                      'cooperator': True,
                                      'email': self.representative_email,
                                      '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 self.is_company:
            representative = self._get_representative()
            vals['name'] = representative.name

        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',
        }