# Author: Leonardo Pistone# Copyright 2015 Camptocamp SA## This program is free software: you can redistribute it and/or modify# it under the terms of the GNU Affero General Public License as# published by the Free Software Foundation, either version 3 of the# License, or (at your option) any later version.## This program is distributed in the hope that it will be useful,# but WITHOUT ANY WARRANTY; without even the implied warranty of# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the# GNU Affero General Public License for more details.## You should have received a copy of the GNU Affero General Public License# along with this program. If not, see <http://www.gnu.org/licenses/>.fromopenerpimportmodels,apiclassPicking(models.Model):_inherit="stock.picking"@api.modeldef_get_partner_to_invoice(self,picking):if'partner_to_invoice_id'inself.env.context:returnself.env.context['partner_to_invoice_id']else:returnsuper(Picking,self)._get_partner_to_invoice(picking)