fix: company info create

2023-08-29 10:56:48 +08:00 · 2023-08-29 10:56:48 +08:00 · 6ce139bec3
parent 3f49586c9f
commit 6ce139bec3
1 changed files with 5 additions and 4 deletions
--- a/cmdb-api/api/views/common_setting/company_info.py
+++ b/cmdb-api/api/views/common_setting/company_info.py
@ -16,14 +16,15 @@ 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
            }
        }
+        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)