[IMP] reason code location set as a scrap location if set

This commit is contained in:
Bhavesh Odedra
2020-08-19 08:55:13 -07:00
committed by John Herholz
parent e541c26d66
commit da7876e647

View File

@@ -25,21 +25,23 @@ class StockScrap(models.Model):
def write(self, vals): def write(self, vals):
if "reason_code_id" in vals: if "reason_code_id" in vals:
vals.update( location_id = (
{ self.env["scrap.reason.code"]
"scrap_location_id": self.env["scrap.reason.code"] .browse(vals.get("reason_code_id"))
.browse(vals.get("reason_code_id")) .location_id
.location_id
}
) )
if location_id:
vals.update({"scrap_location_id": location_id})
return super(StockScrap, self).write(vals) return super(StockScrap, self).write(vals)
@api.model @api.model
def create(self, vals): def create(self, vals):
if "reason_code_id" in vals: if "reason_code_id" in vals:
vals["scrap_location_id"] = ( location_id = (
self.env["scrap.reason.code"] self.env["scrap.reason.code"]
.browse(vals.get("reason_code_id")) .browse(vals.get("reason_code_id"))
.location_id.id .location_id
) )
if location_id:
vals["scrap_location_id"] = location_id.id
return super(StockScrap, self).create(vals) return super(StockScrap, self).create(vals)