def add_cat(self): name = str(self.add_cat_field.text()) if name: new_cat = Category(name=name) new_cat.save() self.table.update() self.add_cat_field.setText("")
def menuEdit(request): if request.method == 'POST': if 'delItem' in request.POST: print request.POST MenuItem.objects.filter(name=request.POST['delItem']).delete() return HttpResponseRedirect(".") elif 'addItem' in request.POST: try: priceVal = int(request.POST['price']) pPriceVal = int(request.POST['pPrice']) catId = int(request.POST['cat']) except ValueError: return HttpResponseRedirect(".") cat = Category.objects.filter(id=catId)[0] try: item = MenuItem(name=request.POST['name'], price=priceVal, pPrice=pPriceVal, category=cat) item.save() except IntegrityError: return HttpResponseRedirect(".") elif 'addCat' in request.POST: try: cat = Category(name=request.POST['cat']) cat.save() except IntegrityError: return HttpResponseRedirect(".") elif 'delCat' in request.POST: cat = Category.objects.filter(name=request.POST['delCat']) print cat print MenuItem.objects.filter(category=cat) MenuItem.objects.filter(category=cat).delete() cat.delete() return HttpResponseRedirect(".") itemDict = {} cats = Category.objects.all().order_by("name") for c in cats: cItems = MenuItem.objects.filter(category=c).order_by("name") itemDict[c] = cItems return render_to_response("pmenu_list_edit.html", { "itemDict":itemDict})