def remove(request):
me = get_my_account(request)
if is_authenticated(request) and is_staff(request):
supplier_id = request.POST.get('id')
try:
s = Supplier(id=supplier_id)
s.delete()
return STATUS_GOOD
except Exception:
return JsonResponse({'status': 'bad', 'message': 'Could not remove supplier'})
else:
return JsonResponse({'status': 'bad', 'message': 'User not authorized staff member'})
def add(request):
me = get_my_account(request)
if is_authenticated(request) and is_staff(request):
supplier_name = request.POST.get('name')
try:
s = Supplier(name=supplier_name)
s.save()
s = Supplier.objects.get(name=supplier_name)
return JsonResponse({'status': 'good', 'data': {
'id': s.id,
'name': s.name
}})
except Exception:
return JsonResponse({'status': 'bad', 'message': 'Could not add supplier'})
else:
return JsonResponse({'status': 'bad', 'message': 'User not authorized staff member'})