From 6ce139bec309e4c5052881c4848b0e0498652fd9 Mon Sep 17 00:00:00 2001 From: "hu.sima" Date: Tue, 29 Aug 2023 10:56:48 +0800 Subject: [PATCH] fix: company info create --- cmdb-api/api/views/common_setting/company_info.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/cmdb-api/api/views/common_setting/company_info.py b/cmdb-api/api/views/common_setting/company_info.py index d2aca2a..d027a9e 100644 --- a/cmdb-api/api/views/common_setting/company_info.py +++ b/cmdb-api/api/views/common_setting/company_info.py @@ -16,15 +16,16 @@ class CompanyInfoView(APIView): return self.jsonify(CompanyInfoCRUD.get()) def post(self): - info = CompanyInfoCRUD.get() - if info: - abort(400, ErrFormat.company_info_is_already_existed) data = { 'info': { **request.values } } - d = CompanyInfoCRUD.create(**data) + info = CompanyInfoCRUD.get() + if info: + d = CompanyInfoCRUD.update(info.get('id'), **data) + else: + d = CompanyInfoCRUD.create(**data) res = d.to_dict() return self.jsonify(res)