diff --git a/yuthon_property/wizard/property_statements_wizard.py b/yuthon_property/wizard/property_statements_wizard.py index 91e8504bf..6b454a52f 100644 --- a/yuthon_property/wizard/property_statements_wizard.py +++ b/yuthon_property/wizard/property_statements_wizard.py @@ -536,13 +536,12 @@ class PropertyStatementsWizard(models.Model): existing_record.write(bank_data) def _get_company_by_unit(self, unit_name): - if unit_name in ['天河投资', '天投城建', '兴源公司']: - return self.env['res.company'].sudo().search([('name', '=', '集团本部')], limit=1) - elif unit_name == '天河物管': + if unit_name == '天河物管': return self.env['res.company'].sudo().search([('name', '=', '天河物管')], limit=1) elif unit_name == '天河粮储': return self.env['res.company'].sudo().search([('name', '=', '天河粮储')], limit=1) - return False + else: + return self.env['res.company'].sudo().search([('name', '=', '集团本部')], limit=1) def classify_by_last_four(self, input_str): if not isinstance(input_str, str) or not input_str.strip():