Browse Source

[PEP8] fix pep8

pull/1/head
houssine 5 years ago
parent
commit
37f5ed6de3
  1. 144
      easy_my_coop/wizard/create_subscription_from_partner.py

144
easy_my_coop/wizard/create_subscription_from_partner.py

@ -1,11 +1,9 @@
# -*- 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"
@ -14,50 +12,51 @@ class PartnerCreateSubscription(models.TransientModel):
@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)]
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))
domain.append(('by_company', '=', True))
else:
domain.append(('by_individual','=',True))
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 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()
@ -70,46 +69,70 @@ class PartnerCreateSubscription(models.TransientModel):
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)]
domain = [('is_share', '=', True)]
partner = self._get_partner()
if partner.is_company:
domain.append(('by_company','=',True))
domain.append(('by_company', '=', True))
else:
domain.append(('by_individual','=',True))
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)
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=_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):
sub_req = self.env['subscription.request']
if sub_req.check_belgian_identification_id(register_number):
return True
else:
raise UserError(_("The national register number is not valid."))
@ -118,7 +141,7 @@ class PartnerCreateSubscription(models.TransientModel):
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,
@ -130,8 +153,8 @@ class PartnerCreateSubscription(models.TransientModel):
'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
@ -141,52 +164,58 @@ class PartnerCreateSubscription(models.TransientModel):
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)])
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'))
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."))
raise UserError(_("A person can't be representative of "
"two different 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}
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})
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
vals['no_registre'] = representative.national_register_number
if coop_vals:
cooperator.write(coop_vals)
new_sub_req = sub_req.create(vals)
return {
'type': 'ir.actions.act_window',
'view_type': 'form, tree',
@ -195,4 +224,3 @@ class PartnerCreateSubscription(models.TransientModel):
'res_id': new_sub_req.id,
'target': 'current',
}
Loading…
Cancel
Save