mirror of https://github.com/veops/cmdb.git
43 lines
979 B
Python
43 lines
979 B
Python
# -*- coding:utf-8 -*-
|
|
from flask import request
|
|
|
|
from api.lib.common_setting.company_info import CompanyInfoCRUD
|
|
from api.resource import APIView
|
|
|
|
prefix = '/company'
|
|
|
|
|
|
class CompanyInfoView(APIView):
|
|
url_prefix = (f'{prefix}/info',)
|
|
|
|
def get(self):
|
|
return self.jsonify(CompanyInfoCRUD.get())
|
|
|
|
def post(self):
|
|
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)
|
|
|
|
|
|
class CompanyInfoViewWithId(APIView):
|
|
url_prefix = (f'{prefix}/info/<int:_id>',)
|
|
|
|
def put(self, _id):
|
|
data = {
|
|
'info': {
|
|
**request.values
|
|
}
|
|
}
|
|
d = CompanyInfoCRUD.update(_id, **data)
|
|
res = d.to_dict()
|
|
return self.jsonify(res)
|