def get(self, request, company_id=None):
"""
Handling GET method.
:args
request: Request to View.
company_id: id of company to be returned.
:return: HttpResponse with company fields and values by id.
If company_id is 'None' returns all companies with their fields and values.
"""
if not company_id:
data = serializers.serialize("json", Company.get_company())
return HttpResponse(data)
company = Company.get_company(company_id)
company = model_to_dict(company)
return HttpResponse(json.dumps({"company": company}))
def delete(self, request, company_id):
"""
Handling DELETE method.
args
request: Request to View.
company_id: id of company to be deleted.
:return: HttpResponse with code 201 if company is deleted.
"""
company = Company()
company.delete_company(company_id)
data = serializers.serialize("json", Company.get_company())
return HttpResponse(data)
def put(self, request):
"""
Handling put method.
:param request: Request to View.
:return: HttpResponse with code 201 if company is updated or
HttpResponseBadRequest if request contain incorrect data.
"""
data = json.loads(request.body)
data["administrator"] = AdviserUser.objects.get(
id=data["administrator"])
company = Company.get_company(data["id"])
company_form = CompanyForm(data, company)
if not company_form.is_valid():
return HttpResponseBadRequest(
"Invalid input data. Please edit and try again.")
company.set_company(data)
return HttpResponse(status=201)
