small tweak for balance adding

reValuate/users/views.py

@@ -74,9 +74,20 @@ def adminView(request):
                 imageObj = Image.objects.get(image=latestPicture, user_id=userId)
                 imageObj.tokenized = True
                 imageObj.save()
-                return redirect("/admin/")
-            else:
-                Image.objects.get(image=latestPicture, user_id=userId).delete()
+                addBal = Balance.objects.get(user_id=userId)
+                try:
+                    user_instance = Balance.objects.get(user=userId)
+                    sum_balance = 5 + user_instance.balanceValue
+                    balAdd = Balance(balanceValue=sum_balance, user_id=user_instance)
+                    balAdd.save()
+                    print(balAdd)
+                except Balance.DoesNotExist:
+                    balAdd = Balance(balanceValue=5, user_id=userId)
+                    balAdd.save()
+
+            return redirect("/admin/")
+        else:
+            Image.objects.get(image=latestPicture, user_id=userId).delete()
             return render(request, "adminView.html", {"image":latestPicture, "userId":userId})
     except IndexError:
         return render(request, "adminView.html")