views.py 文件源码

def deleteData(request, table):
    data = request.data

    if table == "user":
        username = data['username']
        if not User_data.objects.filter(username=username).exists():
            return HttpResponseBadRequest('User record with username ' + username + ' does not exist')
        else:
            user = User_data.objects.get(username=username)
            user.delete()
    elif table == "program":
        pid = data['pid']
        if not Program.objects.filter(pid=pid).exists():
            return HttpResponseBadRequest('Program record with pid ' + str(pid) + ' does not exist')
        else:
            prog = Program.objects.get(pid=pid)
            prog.delete()
    else:
        return HttpResponseBadRequest('Wrong table name')

    return HttpResponse('Delete is successful')