def insertednewcategory(request):
    if 'useradmin' in request.COOKIES and request.method == 'POST':
        category = tb_category()
        category.connectToDB('localhost', 'root', '')
        cat = request.POST['category']
        category.insertCategory(cat)
        category.closeConnection()
        return redirect(categoryhome)
    else:
        return redirect(touserhome)
def insertnewproduct(request):
    if 'useradmin' in request.COOKIES:
        category = tb_category()
        category.connectToDB('localhost', 'root', '')
        categories = category.getAllCategoryNames()
        category.closeConnection()
        return render(request, 'product/insert.html',
                      {'categories': categories})
    else:
        return redirect(touserhome)
def editcategory(request):
    if 'useradmin' in request.COOKIES:
        if request.method == 'POST' and request.POST['operation'] == 'edit':
            category = tb_category()
            category.connectToDB('localhost', 'root', '')
            cat_id = request.POST['editThis']
            cat = category.getCategory(cat_id)
            category.closeConnection()
            return render(request, 'category/edit.html', {'cat': cat})
        else:
            return redirect(touserhome)
    else:
        return redirect(touserhome)
def editedcategory(request):
    if 'useradmin' in request.COOKIES:
        if request.method == 'POST':
            category = tb_category()
            category.connectToDB('localhost', 'root', '')
            cat_id = request.POST['id']
            cat_name = request.POST['category']
            category.editCategory(cat_id, cat_name)
            category.closeConnection()
            return redirect(categoryhome)
        else:
            return redirect(touserhome)
    else:
        return redirect(touserhome)
def categoryhome(request):
    if 'useradmin' in request.COOKIES:
        category = tb_category()
        category.connectToDB('localhost', 'root', '')
        if request.method == 'POST':
            if request.POST['operation'] == 'delete':
                cat_id = request.POST['delThis']
                category.deleteCategory(cat_id)
                category.closeConnection()
                return redirect(categoryhome)
        categories = category.getAllCategories()
        total = len(categories)
        category.closeConnection()
        return render(request, 'category/home.html', {
            'categories': categories,
            'total': total
        })
    else:
        return redirect(touserhome)
def editordermaster(request):
    if 'useradmin' in request.COOKIES:
        if request.method == 'POST' and request.POST['operation'] == 'edit':
            user_id = request.POST['editThis']
            product = tb_product()
            product.connectToDB('localhost', 'root', '')
            products = product.getData(user_id)
            product.closeConnection()
            category = tb_category()
            category.connectToDB('localhost', 'root', '')
            categories = category.getAllCategoryNames()
            category.closeConnection()
            return render(request, 'product/edit.html', {
                'products': products,
                'categories': categories
            })
        else:
            return redirect(touserhome)
    else:
        return redirect(touserhome)