chore: Se remueve código duplicado usando asignador genérico

This commit is contained in:
Camilo Gonzalez 2024-01-28 17:10:27 -05:00
parent 5cc0277db9
commit ef8f45267f

View File

@ -1,10 +1,10 @@
# This file is part of Tryton. The COPYRIGHT file at the top level of # This file is part of Tryton. The COPYRIGHT file at the top level of
# this repository contains the full copyright notices and license terms. # this repository contains the full copyright notices and license terms.
from trytond.wizard import Wizard, StateView, Button, StateTransition from trytond.wizard import Wizard, StateView, Button, StateTransition
from trytond.model import ModelView, fields from trytond.model import ModelView, fields
from trytond.pool import Pool
from core.Prospect.wizards.assign_operator import GenericAssign
class ReassignProspectByProspectStart(ModelView): class ReassignProspectByProspectStart(ModelView):
@ -32,17 +32,9 @@ class ReasignProspectByProspect(Wizard):
reassign_by_prospect = StateTransition() reassign_by_prospect = StateTransition()
def transition_reassign_by_prospect(self): def transition_reassign_by_prospect(self):
pool = Pool() _prospect = self.start.prospect
ProspectTrace = pool.get('sale.prospect_trace') _operator = self.start.new_operator
self.start.prospect.assigned_operator = self.start.new_operator GenericAssign.assign_prospects_to_operator([_prospect], _operator)
if self.start.prospect.prospect_trace:
prospect_trace, = ProspectTrace.search(
[('prospect', '=', self.start.prospect)])
prospect_trace.prospect_assigned_operator =\
self.start.new_operator
prospect_trace.save()
self.start.prospect.save()
return 'end' return 'end'