har0ke/tallybill
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Catch invalid json input

Browse Source
This commit is contained in:
har0ke 2021-06-12 18:51:06 +02:00
parent ea8bae6800
commit f6c45069b9
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main/views.py
View File

@ -2,6 +2,7 @@
# -*- coding: <utf-8> -*- # -*- coding: <utf-8> -*-
import io import io
import json import json
from json.decoder import JSONDecodeError
import math import math
import urllib.parse import urllib.parse
from datetime import date, datetime, timedelta from datetime import date, datetime, timedelta
@ -14,7 +15,7 @@ from django.db.models import Max, Min, F
from django.db.models.aggregates import Sum from django.db.models.aggregates import Sum
from django.db.models.query_utils import Q from django.db.models.query_utils import Q
from django.http import HttpResponseRedirect, HttpResponseNotFound, HttpResponseBadRequest from django.http import HttpResponseRedirect, HttpResponseNotFound, HttpResponseBadRequest
from django.http.response import HttpResponse from django.http.response import Http404, HttpResponse
from django.shortcuts import render from django.shortcuts import render
# Create your views here. # Create your views here.
@ -204,7 +205,11 @@ def logout(request):
@staff_member_required @staff_member_required
def select_product(request): def select_product(request):
if request.method == "POST": if request.method == "POST":
users = json.loads(request.POST["json_data"]) print(request.POST["json_data"])
try:
users = json.loads(request.POST["json_data"])
except JSONDecodeError as e:
return Http404("Invalid data input")
for user, products in users.items(): for user, products in users.items():
user = User.objects.get(pk=int(user)) user = User.objects.get(pk=int(user))
for product, count in products.items(): for product, count in products.items():